The present invention relates to a sentence generation device, a sentence generation method, a sentence generation learning device, a sentence generation learning method, and a program.
Sentence summarization technology based on neural networks is progressing. The summarization technology is a technology by which input of text is received and a sentence for a predetermined purpose (e.g. a summary sentence) is generated based on a pre-trained neural network model.
For example, it has been reported that the summarization accuracy is increased by estimating a I/O model regarding whether or not source words are included in a summary at word level and weighting copy probability of the source words (e.g. NPL 1).
However, the conventional technology has a problem in that, when the importance of each sentence or each word is evaluated in the case where an input sentence (source text) is summarized, the length of a sentence to be generated as output (i.e. an output length) cannot be considered.
The present invention has been made in view of the foregoing, and an object of the present invention is to make it possible to evaluate the importance of a constituent element of an input sentence in correspondence with a designated output length.
To solve the above-described problem, a sentence generation device has: an estimation unit for receiving input of a first sentence and an output length, and estimating importance of each word constituting the first sentence using a pre-trained model; and a generation unit for generating a second sentence based on the importance.
The importance of a constituent element of an input sentence can be evaluated in correspondence with a designated output length.
Hereinafter, embodiments of the present invention will be described with reference to the drawings. First, problems to be addressed in the present embodiment are defined as follows.
Source text XC and an output length L are received, and a summary YL with consideration given to the output length L is generated.
A query XQ, source text XC, and an output length L are received, and a query-dependent summary YL for which the output length L is considered is generated. Note that the query refers to a character string indicating a point of interest (focus point) regarding generation of target text (an output sentence). In the present embodiment, the focus point of the summary is the query.
Note that each of XC, XQ, YL, and Y is a word string, and it is assumed that XC=xC1, . . . , xCn, XQ=xQ1, . . . , xQm, YL=yL1, . . . , yLk, and Y=y1, . . . , yk′. Note that n denotes the number of words (the length of the word string) included in the source text XC. m denotes the number of words (the length of the word string) included in the query XQ. k denotes the number of words (the length of the word string) included in the summary YL. k′ denotes the number of words (the length of the word string) included in the summary Y when the output length is not considered).
A model disclosed in the present embodiment roughly consists of two constituent elements (a content selection model and a generation model). The content selection model is a model for predicting words that are to be included in a summary, and the generation model generates words while considering a predicted value of the content selection model. The framework of the present embodiment is a configuration in which both query-dependent and independent summaries can be dealt with only by replacing the content selection model.
The content selection model predicts probabilities pext1, . . . , pextn∈[0, 1] regarding whether each of the words XC1, . . . , xCn, of the source text is included in the summary sentence. pextt closer to 1 indicates that a t-th word xCt is more likely to be included in the summary sentence. The same problem settings have also been proposed in existing studies. In the present embodiment, NPL 1 is referenced in the case of a query-independent summary, and “Y. Wang, K. Liu, J. Liu, W. He, Y. Lyu, H. Wu, S. Li, and H. Wang. Multi-passage machine reading comprehension with cross-passage answer verication. In ACL (1), pages 1918-1927, 2018.” (hereinafter, “Reference Literature 1”) is referenced in the case of a query-dependent summary. However, known technologies disclosed in literature other than NPL 1 or Reference Literature 1 may also be referenced.
Initially, word embedding sequences eC1, . . . , eCn, and eQ1, . . . , eQm, of a source text word string XC and a query word string XQ are encoded using a BiGRU in a d-dimensional hidden state.
u
t
Q=BiGRU(ut-1Q,etQ) (1)
u
t
C=BiGRU(ut-1C,etC) (2)
Next, matching between the query and the source text is performed using the above information. Here, the model in “M. Hu, Y. Peng, Z. Huang, X. Qiu, F. Wei, and M. Zhou. Reinforced mnemonic reader for machine reading comprehension. In IJCAI, pages 4099-4106, 2018.” (hereinafter, “Reference Literature 2”) is used in the matching. The model in Reference Literature 2 expresses matching mainly using three modules, which are the Interactive alignment, the Self alignment, and the Evidence collection.
<Interactive Alignment>
In this module,
(uC∈2d×n,uQ∈2d×m)
is received as input (n and m denote the sentence lengths (i.e. the number of words arranged) of the source text and the query, respectively), and
o∈
2d×n
is output.
To simplify the description, hereinafter, the input is denoted as (A, B). This module is constituted by two sub-modules, which are an attention match and a semantic fusion. Initially, an attention match (A, B) obtains, using the attention of the respective vectors, a weighted vector
h
t∈2d
at each position t.
Next, a semantic fusion (At, ht) calculates a vector
h
t∈2d
considering interaction between At and ht.
Ã
t
=f(Wr[At;ht;At∘ht;At−ht]) (5)
g
t=σ(Wg[At;ht;At∘ht;At−ht]) (6)
o
t
=g∘Ã
t+(1−g)∘At (7)
σ denotes a sigmoid function. f denotes ReLU.
W
r∈2d×8d,Wg∈2d×8d
is a learning parameter. [;] denotes an operator for vector concatenation.
<Self Alignment>
The model structure is shared with the Interactive alignment. That is to say, the Interactive alignment performs the same processing as that of the Interactive alignment using (o, o), instead of (uC, uQ), as input (A′, B′), and obtains output o′.
<Evidence Collection>
This module obtains
v∈
2d×n
with respect to the output o′ of the Self alignment, using a BiGRU.
v
t=BiGRU(vt-1,ot) (8)
Lastly, a two-layer feedforward network is applied, and a one-dimensional word selection score (importance) is output for each word of the source text.
p
t
ext=σ(w2τf(W1vt+b1)+b2) (9)
Here,
W
1∈d×2d,b1∈d,w2∈2d,b2∈
is a learning parameter.
In the case of a query-independent summary, no query exists, but important words need to be selected for the entire source text, and therefore label prediction for each word is performed while considering information regarding the entire source text. Initially, the source text is encoded using a BiLSTM.
u
t
C=BiLSTM(ut-1C,etC) (10)
Instead of attention match in the case of a query-dependent summary, a document-level vector uCdt is obtained using uCt.
w
d∈2d
is a learning parameter. Next, input (A, B)=(uCt, uCdt) is applied to the semantic fusion. Thereafter, the formula (8) is not applied, and the same processing as that in the case of a query-dependent summary is performed except that vt=ot.
In the case of a query-dependent summary, modeling is performed while considering the correlation between the query and the source text, whereas, in the case of a query-independent summary, modeling is performed while considering both document-level vectors and word-level vectors.
When considering the control of the output length, it is desirable that the output itself of content selection changes depending on the length. For example, it is desired to change the output distribution itself of the content selection model between when 10 words are to be output and when 30 words are to be output, and therefore modeling depending on the number of output words is performed. Specifically, as embedding corresponding to the output length L, eL is concatenated with the input vector eC in the formula (2) as follows:
u
t
C=BiGRU(ut-1C,[etC;eL]) (13)
As output length information regarding a correct answer to be given at this time, the length of a reference summary is given. eL is a learning parameter and is obtained by converting the output length designated as a scalar to a vector in a predetermined number of dimensions (in general, 100 dimensions, for example, are used). Parameters at the time of the conversion are obtained by learning.
That is to say, in the model that considers the output length, in the case of a query-dependent summary, the formula (2) is replaced with the formula (13). In the case of a query-independent summary, the formula (10) is replaced with the formula (13).
As for the content selection model, no correct-answer data exists, and therefore pseudo correct-answer data is automatically generated using a method similar to those of existing studies (NPL 1 and “W.-T. Hsu, C.-K. Lin, M.-Y. Lee, K. Min, J. Tang, and M. Sun. A unified model for extractive and abstractive summarization using inconsistency loss. In ACL (1), pages 132-141, 2018.” (hereinafter, “Reference Literature 3”).
Specifically, first, a pseudo sentence correct-answer label is given to each sentence of the source text using a Recall score of Rouge-L at the sentence level, similarly to Reference Literature 3. Then, with a set of sentences with a sentence label 1 as pseudo source text, word alignment is calculated using DP matching between the reference summary and the pseudo source text. A word selection label 1 is given for matched words, and a word selection label 0 is given for other words. When training the content selection model, training may be performed with a correct-answer label r as the word selection label.
The present embodiment uses a combined attention (Reference Literature 3) in which the attention probability of the base model of “A. See, P. J. Liu, and C. D. Manning. Get to the point: Summarization with pointer-generator networks. In ACL (1), pages 1073-1083, 2017.” is weighted with the output pextt of the content selection model. Furthermore, the output length is also controlled in the generation model.
As a base model for the generation model, a pointer-generator model (Reference Literature 3), which is used in many summarization models, is used. However, any of known models other than the pointer-generator model may be used as the base model.
In the pointer-generator model, the generation probability of a summary Y=y1, . . . , yk is defined as follows:
Here, p(z) denotes the probability of whether yj is copied from the source text. Also, the probability p (yj|zj=1, y1:j-1, x) of the case where yj with z=1 is copied is defined as:
based on an attention distribution aj at a time j. When the decoder state at the time j is denoted as sj, the following formulas hold:
a
jt=exp(gjt)/(Σt′exp(gjt′)) (15)
g
jt
=w
a
τtanh(Wvvt+Wssj+ba) (16)
w
a∈d,Wv∈d×2d,Ws∈d×2dba∈d
is a learning parameter.
The Combined attention replaces the formula (15) with the following formula (17) and calculates an attention distribution to be substituted into the formula (14).
A model that considers the output length on the decoder side is formalized using length embedding, as is dealt with in the existing studies. The present embodiment uses a method of concatenating remaining length embedding to be output with each input of the decoder. Specifically, a length embedding vector
{tilde over (e)}
l
is combined with the input of LSTM of the decoder, as follows:
h
t=LSTM(ht-1,[etx;{tilde over (e)}l]) (18)
Here, l=max((L−t), 0). Note that
{tilde over (e)}
l
is a learning parameter other than eL that is used in the content selection.
The content selection model and the base model (generation model) are pre-trained in advance using respective objective functions. Thereafter, the content selection model is fixed, and a generation model in which the formula (15) is replaced with the formula (17) is trained. The source text XC, the correct-answer label r of each word included in the source text XC (a label indicating whether or not the word is included in the target text (output sentence)), the target text Y or YL (in the case of considering the output length L), the output length L in the case of considering the output length, and the query XQ in the case of considering the query are input as training data to the models during the training.
The objective functions of the content selection model and the generation model are denoted as Lext and Lgen, respectively. The content selection model, which is a binary identification problem for each word, is a binary cross-entropy function, and the generation model is defined similarly to a general language model.
Next, a sentence generation device 10 that realizes the above-described content selection model and generation unit will be described.
A program that realizes processing performed in the sentence generation device 10 is provided by a recording medium 101, which is a CD-ROM or the like. Upon the recording medium 101 in which the program is stored being set to the drive device 100, the program is installed in the auxiliary storage device 102 from the recording medium 101 via the drive device 100. However, the program need not necessarily be installed from the recording medium 101, and may alternatively be downloaded from other computers via a network. The auxiliary storage device 102 stores the installed program, and also stores necessary files, data, or the like.
If an instruction to start the program is given, the memory device 103 loads the program from the auxiliary storage device 102 and stores the loaded program. The CPU 104 performs functions of the sentence generation device 10 in accordance with the program stored in the memory device 103. The interface device 105 is used as an interface for connecting to the network.
In the first embodiment, the input to the sentence generation device 10 is the source text XC and the output length L, and the output from the sentence generation device 10 is the summary YL, with consideration given to the output length L. That is to say, the sentence generation device 10 of the first embodiment realizes a query-independent summary with the output length L designated.
The content selection unit 11a estimates the importance of each constituent element (each sentence or each word) of an input sentence (source text XC) using the content selection model (neural network) that has been trained using the method described in [1.4 Training].
The generation unit 12 generates a target text (the summary sentence YL) based on the results of estimation performed by the content selection unit 11a, using the generation model (neural network) that has been trained using the method described in [1.4 Training]
In step S101a, the content selection unit 11a estimates the importance of each sentence or each word that constitutes the source text XC, in correspondence with the input output length L.
Subsequently, the generation unit 12 performs processing described in [1.3 Combining of Content Selection Model and Generation Model] based on the importance of each sentence or each word that is the output from the content selection unit 11a and the vector sequence (v in the formula (8)) of the source text XC, and generates text (the summary YL) corresponding to the output length L (S102).
Subsequently, the training will be described.
During the training, the sentence generation device 10 further has a parameter learning unit 13. The parameter learning unit 13 learns parameters for the machine learning models (the content selection model and the generation model) using an error between the importance of each word estimated by the content selection unit 11a and the label (0 or 1) of the training data indicating whether or not the word is included in an output sentence that is a correct answer, and an error between an output sentence generated by the generation unit 12 and target text (of the training data) that is the correct answer. Note that the details of the training are as described in [1.4 Training].
As described above, according to the first embodiment, the importance of a constituent element of an input sentence can be evaluated in correspondence with a designated output length.
Next, the second embodiment will be described. In the second embodiment, differences from the first embodiment will be described. The points not particularly mentioned in the second embodiment may be the same as those in the first embodiment.
In
In the second embodiment, the input to the sentence generation device 10 is the source text XC and the query XQ, and the output from the sentence generation device 10 is the summary YL with consideration given to the query XQ. That is to say, the sentence generation device 10 of the second embodiment realizes a query-dependent summary for which the output length L is not designated.
In step S101b, the content selection unit 11b estimates the importance of each sentence or each word that constitutes the source text XC in correspondence with the input query XQ, by performing processing described in [1.2.1 Query-Dependent Summary].
Note that in the second embodiment, the functional configuration during the training may be the same as that shown in
As described above, according to the second embodiment, a sentence can be generated with respect to an input sentence, in correspondence with a designated query.
Next, the third embodiment will be described. In the third embodiment, differences from the first embodiment will be described. The points not particularly mentioned in the third embodiment may be the same as those in the first embodiment.
In
In the third embodiment, the input to the sentence generation device 10 is the source text XC, the output length L, and the query XQ, and the output from the sentence generation device 10 is the summary YL with consideration given to the query XQ. That is to say, the sentence generation device 10 of the third embodiment realizes a query-dependent summary for which the output length L is designated.
In step S101c, the content selection unit 11c estimates the importance of each sentence or each word that constitutes the source text XC in correspondence with the input output length L and query XQ by performing processing described in [1.2.3 Model that Considers Output Length] and [1.2.1 Query-Dependent Summary].
Note that in the third embodiment, the functional configuration during the training may be the same as that shown in
As described above, according to the third embodiment, a sentence can be generated with respect to an input sentence, in correspondence with a designated output length and query.
CNN/DM was used that is generally used as query-independent data in summary tasks. Since no large-scale query-dependent data exists, Q&A task data in MS-MARCO3.1 (“P. Bajaj, D. Campos, N. Craswell, L. Deng, J. Gao, X. Liu, R. Majumder, A. McNamara, B. Mitra, T. Nguyen, M. Rosenberg, X. Song, A. Stoica, S. Tiwary, and T. Wang. MS MARCO: A human generated machine reading comprehension dataset. CoRR, abs/1611.09268v3, 2018.” (hereinafter, “Reference Literature 4”)) was used. The Q&A task data is data obtained by manually creating an answer for a query based on target source text (10 Bing search results). Since many words in the answer are extracted from the source text, it is considered as a setting of a query-dependent summary. In this experiment, a combination of the aforementioned 10 pieces of text are to be summarized. Since the original data of MS-MARCO is large-scale, in this experiment, 300,000 pieces and 10,000 pieces of the original data were randomly sampled and used as training data and evaluation data, respectively. The overview of each data is shown in Table 1.
Table 1: The number of pieces of data used in the experiment
Regarding query-independent data, training was performed with the source text length and the summary text length being truncated to 400 words and 100 words, respectively, during the training, similarly to the conventional summarizing method. The upper limit of the number of words at the time of the test was 120. Regarding query-dependent data, the maximum length of the source text during the training was 800 words. Adagrad was used as an optimization method, and the initial learning rate during pre-training and the initial learning rate during fine tuning were set respectively to 0.15 and 0.05. The initial accumulator value was 0.1, similarly to the existing studies. For word embedding, a 100-dimensional vector of Glove was used and fixed. The number of hidden layers in the GRU and the LSTM was set to 256. The length embedding eL,
{tilde over (e)}
l
was 100-dimensional. The dropout rate was 0.3.
The present embodiment is characterized in that query dependency and the output length in the content selection model are controlled. First, in section 3.4.1, the accuracy when the content selection and the generation model in the case of performing these controls are combined is evaluated using Rouge-L. Next, section 3.4.2 shows the results of analyzing the output length control focusing on the query-dependent summaries for convenience.
[3.4 Results and Considerations]
[3.4.1 Evaluation Regarding Summarization Accuracy with Controlled Output Length]
“Did content selection+generation model summarize important information in correspondence with the designated output length?” Table 2 shows Rouge-L scores in the case of controlling the output length of the entire proposed model (content selection+generation)
First, in both cases of query-dependent and independent summaries, the shorter the output length, the higher the identification rate, and the longer the output length, the higher the reproducibility. Based on this, it can be understood that more important information is included in the summaries. It can be understood from these results that the control of the content selection and the length is also effectively functioning in the query-dependent setting. In the case of CNN/DM, the longer the output length, the higher the F value of Rouge. On the other hand, in the case of MS-MARCO, the F value is highest when the output length constraint is 20. This is due to the length of summary sentences of the original data. Such behavior is exhibited since the average length of summaries is long in the case of CNN/DM, and the average length of summaries is short in the case of MS-MARCO.
“Is the output length control in the content selection model effective?” Table 3 shows the results in the case where, in query-independent summaries (CNN/DM), the control (gold condition) of the output length is considered in the content selection model in addition to the generation model and in the case where it is not considered.
It can be understood that the Rouge scores is improved by controlling the output length in the content selection model.
“Did the sentence generation device 10 output summaries in accordance with the designated output length?” Table 4 shows designated output lengths and actual output lengths.
It was understood that the sentence generation device 10 can output lengths close to the designated lengths, and understood that the output length can also be appropriately controlled in the query-dependent setting. In particular, since in MS-MARCO many of the outputs are short, namely about one sentence, it is conceivable there are many short-length variations and output length information has been successfully learned.
Note that although the above embodiments define the content selection model for the query, the query-dependent content selection model of the present embodiment can be extended to a content selection model with various conditions regarding a focus point for the target text (output sentence), such as a keyword other than the query and interactive context.
Note that, in the present embodiment, the content selection unit 11a and the content selection unit 11c are examples of an estimation unit. The source text is an example of a first sentence. The target text is an example of a second sentence. The parameter learning unit 13 is an example of a learning unit. The sentence generation device 10 is an example of a sentence generation device and a sentence generation learning device.
Although the embodiments of the present invention have been described in detail, the present invention is not limited to those specific embodiments, and various modifications and changes may be made within the scope of the gist of the present invention described in the claims.
Number | Date | Country | Kind |
---|---|---|---|
2019-037616 | Mar 2019 | JP | national |
Filing Document | Filing Date | Country | Kind |
---|---|---|---|
PCT/JP2020/007343 | 2/25/2020 | WO | 00 |