Query and Output: Generating Words by Querying Distributed Word Representations for Paraphrase Generation

Most recent approaches use the sequence-to-sequence model for paraphrase generation. The existing sequence-to-sequence model tends to memorize the words and the patterns in the training dataset instead of learning the meaning of the words. Therefore, the generated sentences are often grammatically correct but semantically improper. In this work, we introduce a novel model based on the encoder-decoder framework, called Word Embedding Attention Network (WEAN). Our proposed model generates the words by querying distributed word representations (i.e. neural word embeddings), hoping to capturing the meaning of the according words. Following previous work, we evaluate our model on two paraphrase-oriented tasks, namely text simplification and short text abstractive summarization. Experimental results show that our model outperforms the sequence-to-sequence baseline by the BLEU score of 6.3 and 5.5 on two English text simplification datasets, and the ROUGE-2 F1 score of 5.7 on a Chinese summarization dataset. Moreover, our model achieves state-of-the-art performances on these three benchmark datasets.


Introduction
Paraphrase is a restatement of the meaning of a text using other words.Many natural language generation tasks are paraphrase-orientated, such as text simplification and short text summarization.Text simplification is to make the text easier to read and understand, especially for poor readers, while short text summarization is to generate a brief sentence to describe the short texts (e.g.posts on the social media).Most recent approaches use sequence-to-sequence model for paraphrase generation (Prakash et al., 2016;Cao et al., 2017).It compresses the source text information into dense vectors with the neural encoder, and the neural decoder generates the target text using the compressed vectors.
Although neural network models achieve success in paraphrase generation, there are still two major problems.One of the problem is that the existing sequence-to-sequence model tends to memorize the words and the patterns in the training dataset instead of the meaning of the words.The main reason is that the word generator (i.e. the output layer of the decoder) does not model the semantic information.The word generator, which consists of a linear transformation and a softmax operation, converts the Recurrent Neural Network (RNN) output from a small dimension (e.g.500) to a much larger dimension (e.g.50,000 words in the vocabulary), where each dimension represents the score of each word.The latent assumption of the word generator is that each word is independent and the score is irrelevant to each other.Therefore, the scores of a word and its synonyms may be of great difference, which means the word generator learns the word itself rather than the relationship between words.
The other problem is that the word generator has a huge number of parameters.Suppose we have a sequence-to-sequence model with a hidden size of 500 and a vocabulary size of 50,000.The word generator has up to 25 million parameters, which is even larger than other parts of the encoder-decoder model in total.The huge size of parameters will result in slow convergence, because there are a lot of parameters to be learned.Moreover, under the distributed framework, the more parameters a model has, the more bandwidth and memory it consumes.
To tackle both of the problems, we propose a novel model called Word Embedding Attention Network (WEAN).The word generator of WEAN is attention based, instead of the simple linear softmax operation.In our attention based word generator, the RNN output is a query, the candidate words are the values, and the corresponding word representations are the keys.In order to predict the word, the attention mechanism is used to select the value matching the query most, by means of querying the keys.In this way, our model generates the words according to the distributed word representations (i.e.neural word embeddings) in a retrieval style rather than the traditional generative style.Our model is able to capture the semantic meaning of a word by referring to its embedding.Besides, the attention mechanism has a much smaller number of parameters compared with the linear transformation directly from the RNN output space to the vocabulary space.The reduction of the parameters can increase the convergence rate and speed up the training process.Moreover, the word embedding is updated from three sources: the input of the encoder, the input of the decoder, and the query of the output layer.
Following previous work (Cao et al., 2017), we evaluate our model on two paraphrase-oriented tasks, namely text simplification and short text abstractive summarization.Experimental results show that our model outperforms the sequence-tosequence baseline by the BLEU score of 6.3 and 5.5 on two English text simplification datasets, and the ROUGE-2 F1 score of 5.7 on a Chinese summarization dataset.Moreover, our model achieves state-of-the-art performances on all of the benchmark datasets.

Proposed Model
We propose a novel model based on the encoderdecoder framework, which generates the words by querying distributed word representations with the attention mechanism.In this section, we first present the overview of the model architecture.Then, we explain the details of the word generation, especially the way to query word embeddings.

Overview
Word Embedding Attention Network is based on the encoder-decoder framework, which consists of two components: a source text encoder, and a target text decoder.Figure 1 is an illustration of our model.Given the source texts, the encoder compresses the source texts into dense representation vectors, and the decoder generates the paraphrased texts.To predict a word, the decoder uses the hidden output to query the word embeddings.The word embeddings assess all the candidate words, and return the word whose embedding matches the query most.The selected word is emitted as the predicted token, and its embedding is then used as the input of the LSTM at the next time step.After the back propagation, the word embedding is updated from three sources: the input of the encoder, the input of the decoder, and the query of the output layer.We show the details of our WEAN in the following subsection.

Encoder and Decoder
The goal of the source text encoder is to provide a series of dense representation of complex source texts for the decoder.In our model, the source text encoder is a Long Short-term Memory Network (LSTM), which produces the dense representation {h 1 , h 2 , ..., h N } from the source text {x 1 , x 2 , ..., x N }: The goal of the target text decoder is to generate a series of paraphrased words from the dense representation of source texts.Fisrt, the LSTM of the decoder compute the dense representation of generated words s t .Then, the dense representations are fed into an attention layer (Bahdanau et al., 2014) to generate the context vector c t , which captures context information of source texts.Attention vector c t is calculated by the weighted sum of encoder hidden states: where g(s t , h i ) is an attentive score between the decoder hidden state s t and the encoder hidden state h i .In this way, c t and s t respectively represent the context information of source texts and the target texts at the t th time step.

Word Generation by Querying Word Embedding
For the current sequence-to-sequence model, the word generator computes the distribution of output words y t in a generative style: where W ∈ R k×V is a trainable parameter matrix, k is hidden size, and V is the number of words in the vocabulary.When the vocabulary is large, the number of parameters will be huge.
Our model generates the words in a retrieval style rather than the traditional generative style, by querying the word embeddings.We denote the combination of the source context vector c t and the target context vector s t as the query q t : The candidate words w i and their corresponding embeddings e i are paired as the key-value pairs {w i , e i }(i = 1, 2, ..., n), where n is the number of candidate words.We give the details of how to determine the set of candidate words in Section 2.4.Our model uses q t to query the key-value pairs {w i , e i }(i = 1, 2, ..., n) by evaluating the relevance between the query q t and each word vector e i with a score function f (q t , e i ).The query process can be regarded as the attentive selection of the word embeddings.We borrow the attention energy functions (Luong et al., 2015) as the relevance score function f (q t , e i ): where W q and W e are two trainable parameter matrices, and v T is a trainable parameter vector.
In implementation, we select the general attention function as the relevance score function, based on the performance on the validation sets.The keyvalue pair with the highest score {w t , e t } is selected.At the test stage, the decoder generates the key w t as the t th predicted word, and inputs the value e t to the LSTM unit at the t + 1 th time step.At the training stage, the scores are normalized as the word probability distribution:

Selection of Candidate Key-value Pairs
As described in Section 2.3, the model generates the words in a retrieval style, which selects a word according to its embedding from a set of candidate key-value pairs.We now give the details of how to obtain the set of candidate key-value pairs.We extract the vocabulary from the source text in the training set, and select the n most frequent words as the candidate words.We reuse the embeddings of the decoder inputs as the values of the candidate words, which means that the decoder input and the predicted output share the same vocabulary and word embeddings.Besides, we do not use any pretrained word embeddings in our model, so that all of the parameters are learned from scratch.

Training
Although our generator is a retrieval style, WEAN is as differentiable as the sequence-to-sequence model.The objective of training is to minimize the cross entropy between the predicted word probability distribution and the golden one-hot distribution: We use Adam optimization method to train the model, with the default hyper-parameters: the learning rate α = 0.001, and β 1 = 0.9, β 2 = 0.999, ǫ = 1e − 8.

Experiments
Following the previous work (Cao et al., 2017), we test our model on the following two paraphrase orientated tasks: text simplification and short text abstractive summarization.

Datasets
The datasets are both from the alignments between English Wikipedia website2 and Simple English Wikipedia website. 3The Simple English Wikipedia is built for "the children and adults who are learning the English language", and the articles are composed with "easy words and short sentences".Therefore, Simple English Wikipedia is a natural public simplified text corpus.
• Parallel Wikipedia Simplification Corpus (PWKP).PWKP (Zhu et al., 2010) is a widely used benchmark for evaluating text simplification systems.It consists of aligned complex text from English WikiPedia (as of Aug. 22nd, 2009) and simple text from Simple Wikipedia (as of Aug. 17th, 2009).The dataset contains 108,016 sentence pairs, with 25.01 words on average per complex sentence and 20.87 words per simple sentence.Following the previous work (Zhang and Lapata, 2017), we remove the duplicate sentence pairs, and split the corpus with 89,042 pairs for training, 205 pairs for validation and 100 pairs for test.
• English Wikipedia and Simple English Wikipedia (EW-SEW).EW-SEW is a publicly available dataset provided by Hwang et al. (2015).To build the corpus, they first align the complex-simple sentence pairs, score the semantic similarity between the complex sentence and the simple sentence, and classify each sentence pair as a good, good partial, partial, or bad match.Following the previous work (Nisioi et al., 2017), we discard the unclassified matches, and use the good matches and partial matches with a scaled threshold greater than 0.45.The corpus contains about 150K good matches and 130K good partial matches.We use this corpus as the training set, and the dataset provided by Xu et al. (Xu et al., 2016) as the validation set and the test set.The validation set consists of 2,000 sentence pairs, and the test set contains 359 sentence pairs.Besides, each complex sentence is paired with 8 reference simplified sentences provided by Amazon Mechanical Turk workers.

Evaluation Metrics
Following the previous work (Nisioi et al., 2017;Hu et al., 2015), we evaluate our model with different metrics on two tasks.
• Automatic evaluation.We use the BLEU score (Papineni et al., 2002) as the automatic evaluation metric.BLEU is a widely used metric for machine translation and text simplification, which measures the agreement between the model outputs and the gold references.The references can be either single or multiple.In our experiments, the references are single on PWKP, and multiple on EW-SEW.
• Human evaluation.Human evaluation is essential to evaluate the quality of the model outputs.Following Nisioi et al. (2017) and Zhang et al. (2017), we ask the human raters to rate the simplified text in three dimensions: Fluency, Adequacy and Simplicity.Fluency assesses whether the outputs are grammatically right and well formed.Adequacy represents the meaning preservation of the simplified text.Both the scores of fluency and adequacy range from 1 to 5 (1 is very bad and 5 is very good).Simplicity shows how simpler the model outputs are than the source text, which ranges from 1 to 5.

Baselines
We compare our model with several neural text simplification systems.
• Seq2seq is our implementation of the sequence-to-sequence model with attention mechanism, which is the most popular neural model for text generation.
• EncDecA is a model based on the encoderdecoder with attention, implemented by Zhang and Lapata (2017).
• PBMT-R (Wubben et al., 2012) is a phrase based machine translation model which reranks the outputs.
• Hybrid (Narayan and Gardent, 2014) is a hybrid approach which combines deep semantics and mono-lingual machine translation.

Results
We compare WEAN with state-of-the-art models for text simplification.Table 1 and Table 2 summarize the results of the automatic evaluation.On PWKP dataset, we compare WEAN with PBMT, Hybrid, EncDecA, DRESS and DRESS-LS.WEAN achieves a BLEU score of 54.54, outperforming all of the previous systems.On EW-SEW dataset, we compare WEAN with PBMT-R, Hybrid, SBMT-SARI, and the neural models described above.We do not find any public release code of PBMT-R and SBMT-SARI.Fortunately, Xu et al. (2016) provides the predictions of PBMT-R and SBMT-SARI on EW-SEW test set, so that we can compare our model with these systems.
LCSTS R-1 R-2 R-L RNN-W (Hu et al., 2015) 17.7 8.5 15.8 RNN (Hu et al., 2015) 21.5 8.9 18.6 RNN-cont-W (Hu et al., 2015) 26.8 16.1 24.1 RNN-cont (Hu et al., 2015) 29.9 17.4 27.2 SRB (Ma et al., 2017) 33.3 20.0 30.1 CopyNet-W (Gu et al., 2016) 35.0 22.3 32.0 CopyNet (Gu et al., 2016) 34.4 21.6 31.3RNN-dist (Chen et al., 2016)  It shows that the neural models have better performance in BLEU, and WEAN achieves the best BLEU score with 94.45.We perform the human evaluation of WEAN and other related systems, and the results are shown in Table 3. DRESS-LS is based on the reinforcement learning, and it encourages the fluency, simplicity and relevance of the outputs.Therefore, it achieves a high score in our human evaluation.WEAN gains a even better score than DRESS-LS.Besides, WEAN generates more adequate and simpler outputs than the reference on PWKP.The predictions of SBMT-SARI are the most adequate among the compared systems on EW-SEW.In general, WEAN outperforms all of the other systems, considering the balance of fluency, adequate and simplicity.We conduct significance tests based on t-test.The significance tests suggest that WEAN has a very significant improvement over baseline, with p ≤ 0.001 over DRESS-LS in all of the dimension on PWKP, p ≤ 0.05 over DRESS-LS in the dimension of fluency, p ≤ 0.005 over NTS-w2v in the dimension of simplicity and p ≤ 0.005 over DRESS-LS in the dimension of all.

Dataset
Large Scale Chinese Social Media Short Text Summarization Dataset (LCSTS): LCSTS is constructed by Hu et al. (2015).The dataset consists of more than 2,400,000 text-summary pairs, constructed from a famous Chinese social media website called Sina Weibo. 4 It is split into three parts, with 2,400,591 pairs in PART I, 10,666 pairs in PART II and 1,106 pairs in PART III.All the text-summary pairs in PART II and PART III are manually annotated with relevant scores ranged from 1 to 5. We only reserve pairs with scores no less than 3, leaving 8,685 pairs in PART II and 725 pairs in PART III.Following the previous work (Hu et al., 2015), we use PART I as training set, PART II as validation set, and PART III as test set.

Evaluation Metrics
Our evaluation metric is ROUGE score (Lin and Hovy, 2003), which is popular for summarization evaluation.The metrics compare an automatically produced summary against the reference summaries, by computing overlapping lexical units, including unigram, bigram, trigram, and longest common subsequence (LCS).Following previous work (Rush et al., 2015;Hu et al., 2015), we use ROUGE-1 (unigram), ROUGE-2 (bi-gram) and ROUGE-L (LCS) as the evaluation metrics in the reported experimental results.

Settings
The vocabularies are extracted from the training sets, and the source contents and the summaries share the same vocabularies.We tune the hyperparameters based on the ROUGE scores on the validation sets.In order to alleviate the risk of word segmentation mistakes, we split the Chinese sentences into characters.We prune the vocabulary size to 4,000, which covers most of the common characters.We set the word embedding size and the hidden size to 512, the number of LSTM layers of the encoder is 2, and the number of LSTM layers of the decoder is 1.The batch size is 64, and we do not use dropout (Srivastava et al., 2014) on this dataset.Following the previous work (Li et al., 2017), we implement a beam search optimization, and set the beam size to 5.

Baselines
We compare our model with the state-of-the-art baselines.
• RNN and RNN-cont are two sequence-tosequence baseline with GRU encoder and decoder, provided by Hu et al. (2015).
• CopyNet (Gu et al., 2016) incorporates a copy mechanism to allow part of the generated summary is copied from the source content.
• SRB (Ma et al., 2017) is a sequence-tosequence based neural model with improving the semantic relevance between the input text and the output summary.
• DRGD (Li et al., 2017) is a deep recurrent generative decoder model, combining the decoder with a variational autoencoder.
• Seq2seq is our implementation of the sequence-to-sequence model with the attention mechanism.

Results
We report the ROUGE F1 score of our model and the baseline models on the test sets.Table 4 summarizes the comparison between our model and the baselines.Our model achieves the score of 37.8 ROUGE-1, 25.6 ROUGE-2, and 35.2 ROUGE-L, outperforming all of the previous models.First, we compare our model with the sequence-to-sequence model.It shows that our model significant outperforms the sequenceto-sequence baseline with a large margin of 5.7 ROUGE-1, 5.7 ROUGE-2, and 6.0 ROUGE-L.Then, we compare our model with other related models.The state-of-the-art model is DRGD (Li et al., 2017), which obtains the score of 37.0 ROUGE-1, 24.2 ROUGE-2, and 34.2 ROUGE-L.Our model has a relative gain of 0.8 ROUGE-1, 1.4 ROUGE-2 and 1.0 ROUGE-L over the state-of-the-art models.

Reducing Parameters
Our WEAN reduces a large number of the parameters in the output layer.To analyze the parameter reduction, we compare our WEAN model with the sequence-to-sequence model.Table 5 lists the number of the parameters in the output layers of two models.Both PWKP and EWSEWhave the vocabulary size of 50000 words and the hidden size of 256, resulting 50000 × 256 = 12, 800, 000 parameters.LCSTS has a vocabulary size of 4000 and the hidden size of 512, so the seq2seq has 4000 × 512 = 2, 048, 000 parameters in the output layers.WEAN only has two parameter matrices and one parameter vector at most in Equation 5, without regard to the vocabulary size.It has 256 × 256 × 2 + 256 = 131, 328 parameters on PWKP and EWSEW, and 512×512×2+512 = 524, 800 parameters on LCSTS.Besides, WEAN does not have any extra parameters in the other part of the model.

Speeding up Convergence
Figure 2 shows the training curve of WEAN and Seq2seq on the PWKP validation set.WEAN achieve near the optimal score in only 2-3 epochs, while Seq2seq takes more than 15 epochs to achieve the optimal score.Therefore, WEAN has much faster convergence rate, compared with Seq2seq.With the much faster training speed, WEAN does not suffer loss in BLEU, and even improve the BLEU score.

Source
Yoghurt or yogurt is a dairy product produced by bacterial fermentation of milk .Reference Yoghurt or yogurt is a dairy product made by bacterial fermentation of milk .

NTS
. or yoghurt is a dairy product produced by bacterial fermentation of milk .

NTS-w2v
It is made by bacterial fermentation of milk .PBMT-R Yoghurt or yogurt is a dairy product produced by bacterial fermentation of .SBMT-SARI Yogurt or yogurt is a dairy product drawn up by bacterial fermentation of milk .WEAN Yoghurt or yogurt is a dairy product made by bacterial fermentation of milk .Source Depending on the context, another closely-related meaning of constituent is that of a citizen residing in the area governed, represented, or otherwise served by a politician; sometimes this is restricted to citizens who elected the politician.

Reference
The word constituent can also be used to refer to a citizen who lives in the area that is governed, represented, or otherwise served by a politician; sometimes the word is restricted to citizens who elected the politician.NTS Depending on the context, another closely-related meaning of constituent is that of a citizen living in the area governed, represented, or otherwise served by a politician; sometimes this is restricted to citizens who elected the politician.This is restricted to citizens who elected the politician.PBMT-R Depending on the context and meaning of closely-related siemens-martin -rrb-is a citizen living in the area, or otherwise, was governed by a 1924-1930 shurba; this is restricted to people who elected it.SBMT-SARI In terms of the context, another closely-related sense of the component is that of a citizen living in the area covered, make up, or if not, served by a policy; sometimes this is limited to the people who elected the policy.WEAN Depending on the context, another closely-related meaning of constituent is that of a citizen who lives in the area governed, represented, or otherwise served by a politician; sometimes the word is restricted to citizens who elected the politician.

Case Study
Table 6 shows two examples of different text simplification system outputs on EW-SEW.For the first example, NTS, NTS-w2v and PBMT-R miss some essential constituents, so that the sentences are incomplete and not fluent.SBMT-SARI generates a fluent sentence, but the output does not preserve the original meaning.The predicted sentence of WEAN is fluent, simple, and the same as the reference.For the second example, NTS-w2v omits so many words that it lacks a lot of information.PBMT-R generates some irrelevant words, like 'siemens-martin', '-rrb-', and 'shurba', which hurts the fluency and adequacy of the generated sentence.SBMT-SARI is able to generate a fluent sentence, but the meaning is different from the source text, and even more difficult to understand.Compared with the statistic model, WEAN generates a more fluent sentence.Besides, WEAN can capture the semantic meaning of the word by querying the word embeddings, so the generated sentence is semantically correct, and very close to the original meaning.
(2012) presents a method for text simplification using phrase based machine translation with reranking the outputs.Kauchak (2013) proposes a text simplification corpus, and evaluates language modeling for text simplification on the proposed corpus.Narayan and Gardent (2014) propose a hybrid approach to sentence simplification which combines deep semantics and monolingual machine translation.Hwang et al. (2015) introduces a parallel simplification corpus by evaluating the similarity between the source text and the simplified text based on WordNet.Glavaš and Štajner (2015) propose an unsupervised approach to lexical simplification that makes use of word vectors and require only regular corpora.Xu et al. (2016) design automatic metrics for text simplification.Recently, most works focus on the neural sequence-to-sequence model.Nisioi et al. (2017) present a sequence-to-sequence model, and re-ranks the predictions with BLEU and SARI.Zhang and Lapata (2017) propose a deep reinforcement learning model to improve the simplicity, fluency and adequacy of the simplified texts.Cao et al. (2017) introduce a novel sequence-tosequence model to join copying and restricted generation for text simplification.Rush et al. (2015) first used an attention-based encoder to compress texts and a neural network language decoder to generate summaries.Following this work, recurrent encoder was introduced to text summarization, and gained better performance (Lopyrev, 2015;Chopra et al., 2016).Towards Chinese texts, Hu et al. (2015) built a large corpus of Chinese short text summarization.To deal with unknown word problem, Nallapati et al. (2016) proposed a generator-pointer model so that the decoder is able to generate words in source texts.Gu et al. (2016) also solved this issue by incorporating copying mechanism.

Conclusion
We propose a novel model based on the encoderdecoder framework, which generates the words by querying distributed word representations.Experimental results show that our model outperforms the sequence-to-sequence baseline by the BLEU score of 6.3 and 5.5 on two English text simplification datasets, and the ROUGE-2 F1 score of 5.7 on a Chinese summarization dataset.Moreover, our model achieves state-of-the-art performances on these three benchmark datasets.

Figure 2 :
Figure 2: The training curve of WEAN and Seq2seq on the PWKP validation set.

Table 1 :
Automatic evaluation of our model and other related systems on PWKP datasets.The results are reported on the test sets.

Table 2 :
Automatic evaluation of our model and other related systems on EW-SEW datasets.The results are reported on the test sets.
tune our hyper-parameter on the development set.The model has two LSTM layers.The hidden size of LSTM is 256, and the embedding size is 256.

Table 3 :
Human evaluation of our model and other related systems on PWKP and EW-SEW datasets.The results are reported on the test sets.

Table 5 :
The number of the parameters in the output layer.The numbers of rest parameters between Seq2seq and WEAN are the same.

Table 6 :
Two examples of different text simplification system outputs in EW-SEW dataset.Differences from the source texts are shown in bold.