The embodiments are directed to summarizing text using summarization models, and more specifically to a decoder neural network in a summarization model that generates a diverse set of summaries.
Generating an abstract summary of a text involves different decisions. These decisions include whether to copy content directly from the input document, whether to paraphrase content, the level of specificity and/or generality of the summary, the length of the summary, readability of the summary, etc. Current summarization systems implicitly encode these decisions as parameters, but do not provide a mechanism for users to control the parameters to obtain a diverse set of summaries for a given input. Commonly used sampling methods such as beam search, top-k decoding or diverse decoding tend to output stylistically similar summaries. These sampling methods also cannot be queried to generate multiple diverse summaries satisfying a target set of features or styles.
In one or more implementations, not all of the depicted components in each figure may be required, and one or more implementations may include additional components not shown in a figure. Variations in the arrangement and type of the components may be made without departing from the scope of the subject disclosure. Additional components, different components, or fewer components may be utilized within the scope of the subject disclosure.
The embodiments are directed to a summarization model that receives input, such as a text or document, and generates a diverse set of summaries of the input. The summarization model includes an encoder, such as a single transformer-based encoder to encode the input and a mixture-of-experts framework with multiple decoders for summary generation.
The summarization model receives input and generates a summary over a series of time steps. The summary may include multiple tokens, with each token being a word. At each time step of the summary generation phase, each decoder computes a probability of the next token being included in the summary. The summarization model computes next token's probability distribution by combining the probability distributions generated by each decoder. Multiple decoders allow the summarization model to distribute the diverse stylistic and lexical features encountered in the training data, even those within the same reference summary, and across parameters of the separate decoders. For example, a summarization framework may include two decoders. The first decoder may learn to copy phrases or words from an input document. The second decoder may learn to paraphrase and make syntactic transformations. While the individual decoders cannot cover the range of stylistic variations in the dataset, a weighted combination or mixture of the two decoders can be used to model the summarization dataset.
In some embodiments, a summarization model may be trained using unguided and guided settings. In the unguided setting, the summarization model does not explicitly control the partitioning of the summary features. In the guided setting, different decoders are trained to learn contrasting summary styles along one specific feature, such as low abstractiveness or high abstractiveness. The experiments on three summarization datasets called the CNN, NEWSROOM, and XSUM datasets, illustrate that the summarization model generates summaries that have significantly better stylistic diversity and improvement in the top-K quality as compared to baseline models, such as a Bidirectional and Auto-Regressive Transformers (BART) model. Moreover, the multiple decoders in the summarization model allow for sampling from any combination of available decoders which translates into flexile summaries.
Memory 120 may be used to store software executed by computing device 100 and/or one or more data structures used during operation of computing device 100. Memory 120 may include one or more types of machine readable media. Some common forms of machine readable media may include floppy disk, flexible disk, hard disk, magnetic tape, any other magnetic medium, CD-ROM, any other optical medium, punch cards, paper tape, any other physical medium with patterns of holes, RAM, PROM, EPROM, FLASH-EPROM, any other memory chip or cartridge, and/or any other medium from which a processor or computer is adapted to read.
Processor 110 and/or memory 120 may be arranged in any suitable physical arrangement. In some embodiments, processor 110 and/or memory 120 may be implemented on a same board, in a same package (e.g., system-in-package), on a same chip (e.g., system-on-chip), and/or the like. In some embodiments, processor 110 and/or memory 120 may include distributed, virtualized, and/or containerized computing resources. Consistent with such embodiments, processor 110 and/or memory 120 may be located in one or more data centers and/or cloud computing facilities.
In some examples, memory 120 may include a non-transitory, tangible, machine readable media that includes executable code that when run by one or more processors (e.g., processor 110) may cause the one or more processors to perform the methods described in further detail herein. Memory 120 stores a summarization model 130. Summarization model 130 receives text input 140, such as an article, a document, etc., and generates a summary 150 of the text input 140. Summary 150 may be, for example, an abstract of the article or a document or another type of text input 140.
Encoder 202 and decoder network 204 may be neural networks. Decoder network 204 may include network multiple decoders φ1, φ2, . . . φk. There may be a k number of decoders, where k is an integer.
Summarization model 130 may generate words over a time period comprising multiple time steps. At each time step, each decoder 206A-C may output a probability distribution Pφ
Each one of decoders 206 may be divided into M decoder blocks or layers that include layers specific to decoders 206 and shared layers. M may be an integer, such as M=12. As illustrated in
During the training phase, the text input 140, which may be article x and reference summary y are tokenized and fed into the encoder 202 and decoder network 204 respectively. The encodings from article x are fed into the decoder network 204.
The gating mechanism 208 may be included in decoder network 204. The gating mechanism 208 includes a feed forward layer 218 and a softmax layer 220. In the gating mechanism, let him be the hidden state output of the mth decoder layer (layer 216m) at time step i. The mth layer 216 is the last shared layer between the decoders 206A-C. The hidden state output him may be fed through a feed forward layer W (218) and the softmax layer 220. The size of the feed forward layer 218 may be size=[|him|, k]. The softmax layer 220 receives the output of the feed forward layer 228 and outputs the probability distribution of selecting specific decoders 206A-C for word generation. The probability distribution is used to compute the overall next-token output probability over a vocabulary and is computed as follows:
P(|x,<1)=Σj=1:kgij*Pφ
where k corresponds to the number of decoders, gij corresponds to the probability of selecting the jth decoder at time step i. The word or token that corresponds to the probability of the jth decoder 206 is included in summary 150.
During the training stage, the summarization model 130 may be trained to minimize the cross entropy loss of the reference summaries conditioned on the input document, such as text input 140. The loss may be determined as follows:
loss=−Σi log P(yi|x,<i) (2)
The gating mechanism 208 may be trained using unguided training and guided training.
loss=−Σi log[(1−g)*Pφ
In this case, if g=0, the first decoder may determine the summary 150, and if g=1, the second decoder may determine the summary 150. By setting g∈[0,1], summarization model 130 maybe trained using a mixture of decoders to denote mid-level specificity.
Going back to
In an inference strategy with individual decoders 206, summarization model 130 generates summaries using selected jth decoder, such as decoder 206A, 206B or 206C in
In an inference strategy with multiple decoders 206A-B, the gating mechanism 208 generates vector 226 that includes a mixture of weights. The weights indicate a contribution of probability distributions 222A-C generated by decoders 206A-C to the probability 224 generated by the decoder network 204. The weights in vector 226 may have values between zero and one. The weights may be generated by passing the hidden state him that is the output of the last shared layer 216m through feed forward layer 218 and the softmax layer 220. The vector 226 for decoder φj may be gij=(WT him)j at time step t.
In an inference strategy that uses a manually specified vector 226, summarization model 130 may receive an input from a user identifying values for vector 226, which may control sampling from decoders 206A-C. For example, suppose decoder 206A is trained to learn abstractive features and decoder 206B is trained to learn extractive features. Summarization model 130 may control the degree of abstraction in the generated summaries 150 by sampling the probability distributions 222A and 222B using different values for vector 226, such as [1−g,g]. For example, given a user specified distribution vector 226 of [0.3, 0.7], the output probabilities for decoders 206A and 206B may be as follows:
P(|·)=0.3*Pφ
At process 602, a text input 140, such as an article, is received at summarization model 130 that includes encoder 202 and decoder network 204 with multiple decoders 206. As discussed above, decoders 206 may share a pre-defined number of bottom layers, such as layers 216a-216m, and have non-overlapping and distinct top layers, such as respective layers 210A-M, 212A-M, and 214A-M in the corresponding decoders 206A-C. Additionally, decoder network 204 may include a gating mechanism 208 that shares shared layers 216a-216m with decoders 206.
At process 604, encodings are generated from the text input. For example, text input 140 is tokenized and fed into an encoder 202 of summarization model 130 to generate encodings.
At process 606, encodings are fed into a decoder network 204 of the summarization model 130 to generate probability distributions. For time steps other than the first time step, the output token or tokens of the decoder network 104 is also fed as input into decoder network 104 for the subsequent iteration together with the encodings. The encodings and the output from the previous time step are passed through the shared layers 216a-m and individual layers of decoders 206, such as layers 210A-M of decoder 206A, layers 212A-M of decoder 206B, and layers 214A-M of decoder 206C to generate probability distributions 222A-222C.
At process 608, a probability of a word or token is generated from the probability distributions using a gating mechanism. For example, using an unguided strategy, gating mechanism 208 may determine gating vector 226 which identifies weights with indicate a contribution of a probability distributions 222A-C of each decoder 206A-C to probability 224 of a next-token in summary 150. The contributions of each probability distribution 222A-C are then added into probability 224 from which a word or a token is determined. In another example, using a guided strategy, gating mechanism 208 may determine whether to determine a word or token for inclusion into summary from one of decoders 206A-C. In yet example, using a guided strategy, gating vector 226 may include user input which specifies a contribution of a probability distribution of each decoder 206A-C to the next-token or word in summary 150.
At process 610, a word in a vocabulary is selected based on the probability distribution as the word or token for the summary 150.
Processes 604-612 may repeat over multiple time steps until summary 150 is generated.
Some examples of computing devices, such as computing device 100 may include non-transitory, tangible, machine readable media that include executable code that when run by one or more processors (e.g., processor 110) may cause the one or more processors to perform the processes of method 600. Some common forms of machine readable media that may include the processes of method 600 are, for example, floppy disk, flexible disk, hard disk, magnetic tape, any other magnetic medium, CD-ROM, any other optical medium, punch cards, paper tape, any other physical medium with patterns of holes, RAM, PROM, EPROM, FLASH-EPROM, any other memory chip or cartridge, and/or any other medium from which a processor or computer is adapted to read.
This description and the accompanying drawings that illustrate inventive aspects, embodiments, implementations, or applications should not be taken as limiting. Various mechanical, compositional, structural, electrical, and operational changes may be made without departing from the spirit and scope of this description and the claims. In some instances, well-known circuits, structures, or techniques have not been shown or described in detail in order not to obscure the embodiments of this disclosure. Like numbers in two or more figures represent the same or similar elements.
In this description, specific details are set forth describing some embodiments consistent with the present disclosure. Numerous specific details are set forth in order to provide a thorough understanding of the embodiments. It will be apparent, however, to one skilled in the art that some embodiments may be practiced without some or all of these specific details. The specific embodiments disclosed herein are meant to be illustrative but not limiting. One skilled in the art may realize other elements that, although not specifically described here, are within the scope and the spirit of this disclosure. In addition, to avoid unnecessary repetition, one or more features shown and described in association with one embodiment may be incorporated into other embodiments unless specifically described otherwise or if the one or more features would make an embodiment non-functional.
Although illustrative embodiments have been shown and described, a wide range of modification, change and substitution is contemplated in the foregoing disclosure and in some instances, some features of the embodiments may be employed without a corresponding use of other features. One of ordinary skill in the art would recognize many variations, alternatives, and modifications. Thus, the scope of the invention should be limited only by the following claims, and it is appropriate that the claims be construed broadly and in a manner consistent with the scope of the embodiments disclosed herein.
This application claims priority under 35 U.S.C. § 119 to U.S. Provisional Application No. 63/235,544, filed Aug. 20, 2021, which is hereby expressly incorporated by reference herein in its entirety.
Number | Date | Country | |
---|---|---|---|
63235544 | Aug 2021 | US |