Attention-based NMT with Coverage and Context Gate
In neural machine translation (NMT), generation of a target word depends on both source and target contexts. We find that source contexts have a direct impact on the adequacy of a translation while target contexts affect the fluency. Intuitively, generation of a content word should rely more on the source context and generation of a functional word should rely more on the target context. Due to the lack of effective control over the influence from source and target contexts, conventional NMT tends to yield fluent but inadequate translations. To address this problem, we propose context gates which dynamically control the ratios at which source and target contexts contribute to the generation of target words. In this way, we can enhance both the adequacy and fluency of NMT with more careful control of the information flow from contexts. Experiments show that our approach significantly improves upon a standard attention-based NMT system by +2.3 BLEU points.READ FULL TEXT VIEW PDF
Attention-based NMT with Coverage and Context Gate
Attention-based NMT with Coverage and Context Gate
|input||jīnnián qián liǎng yuè guǎngdōng gāoxīn jìshù chǎnpǐn chūkǒu 37.6yì měiyuán|
|NMT||in the first two months of this year , the export of new high level technology product was UNK - billion us dollars|
|china ’s guangdong hi - tech exports hit 58 billion dollars|
|china ’s export of high and new hi - tech exports of the export of the export of the export of the export of the export of the export of the export of the export of|
has made significant progress in the past several years. Its goal is to construct and utilize a single large neural network to accomplish the entire translation task. One great advantage of NMT is that the translation system can be completely constructed by learning from data without human involvement (cf., feature engineering in statistical machine translation (SMT)). The encoder-decoder architecture is widely employed [Cho et al.2014, Sutskever et al.2014]
, in which the encoder summarizes the source sentence into a vector representation, and the decoder generates the target sentence word-by-word from the vector representation. The representation of the source sentence and the representation of the partially generated target sentence (translation) at each position are referred to as source context and target context, respectively. The generation of a target word is determined jointly by the source context and target context.
Several techniques in NMT have proven to be very effective, including gating [Hochreiter and Schmidhuber1997, Cho et al.2014] and attention [Bahdanau et al.2015] which can model long-distance dependencies and complicated alignment relations in the translation process. Using an encoder-decoder framework that incorporates gating and attention techniques, it has been reported that the performance of NMT can surpass the performance of traditional SMT as measured by BLEU score [Luong et al.2015].
Despite this success, we observe that NMT usually yields fluent but inadequate translations.111Fluency measures whether the translation is fluent, while adequacy measures whether the translation is faithful to the original sentence [Snover et al.2009]. We attribute this to a stronger influence of target context on generation, which results from a stronger language model than that used in SMT. One question naturally arises: what will happen if we change the ratio of influences from the source or target contexts?
Table 1 shows an example in which an attention-based NMT system [Bahdanau et al.2015] generates a fluent yet inadequate translation (e.g., missing the translation of “guǎngdōng”). When we halve the contribution from the source context, the result further loses its adequacy by missing the partial translation “in the first two months of this year”. One possible explanation is that the target context takes a higher weight and thus the system favors a shorter translation. In contrast, when we halve the contribution from the target context, the result completely loses its fluency by repeatedly generating the translation of “chūkǒu” (i.e., “the export of”) until the generated translation reaches the maximum length. Therefore, this example indicates that source and target contexts in NMT are highly correlated to translation adequacy and fluency, respectively.
In fact, conventional NMT lacks effective control on the influence of source and target contexts. At each decoding step, NMT treats the source and target contexts equally, and thus ignores the different needs of the contexts. For example, content words in the target sentence are more related to the translation adequacy, and thus should depend more on the source context. In contrast, function words in the target sentence are often more related to the translation fluency (e.g., “of” after “is fond”), and thus should depend more on the target context.
In this work, we propose to use context gates to control the contributions of source and target contexts on the generation of target words (decoding) in NMT. Context gates are non-linear gating units which can dynamically select the amount of context information in the decoding process. Specifically, at each decoding step, the context gate examines both the source and target contexts, and outputs a ratio between zero and one to determine the percentages of information to utilize from the two contexts. In this way, the system can balance the adequacy and fluency of the translation with regard to the generation of a word at each position.
Experimental results show that introducing context gates leads to an average improvement of +2.3 BLEU points over a standard attention-based NMT system [Bahdanau et al.2015]. An interesting finding is that we can replace the GRU units in the decoder with conventional RNN units and in the meantime utilize context gates. The translation performance is comparable with the standard NMT system with GRU, but the system enjoys a simpler structure (i.e., uses only a single gate and half of the parameters) and a faster decoding (i.e., requires only half the matrix computations for decoding).222Our code is publicly available at https://github.com/tuzhaopeng/NMT.
Suppose that represents a source sentence and
a target sentence. NMT directly models the probability of translation from the source sentence to the target sentence word by word:
where . As shown in Figure 1, the probability of generating the i-th word
is computed by using a recurrent neural network (RNN) in the decoder:
first linearly transforms its input then applies a softmax function,is the previously generated word, is the -th decoding hidden state, and is the -th source representation. The state is computed as follows:
is an -dimensional embedding of the previously generated word .
is a vector representation extracted from the source sentence by the encoder. The encoder usually uses an RNN to encode the source sentence into a sequence of hidden states , in which is the hidden state of the -th source word . can be either a static vector that summarizes the whole sentence (e.g., ) [Cho et al.2014, Sutskever et al.2014], or a dynamic vector that selectively summarizes certain parts of the source sentence at each decoding step (e.g., in which
is alignment probability calculated by an attention model)[Bahdanau et al.2015].
, , are matrices with and being the numbers of units of decoder hidden state and source representation, respectively.
The inputs to the decoder (i.e., , , and ) represent the contexts. Specifically, the source representation stands for source context, which embeds the information from the source sentence. The previous decoding state and the previously generated word constitute the target context.333In a recent implementation of NMT (https://github.com/nyu-dl/dl4mt-tutorial), and are combined together with a GRU before being fed into the decoder, which can boost translation performance. We follow the practice and treat both of them as target context.
We first empirically investigate our hypothesis: whether source and target contexts correlate to translation adequacy and fluency. Figure 2(a) shows the translation lengths with various scaling ratios for source and target contexts:
For example, the pair (1.0, 0.5) means fully leveraging the effect of source context while halving the effect of target context. Reducing the effect of target context (i.e., the lines (1.0, 0.8) and (1.0, 0.5)) results in longer translations, while reducing the effect of source context (i.e., the lines (0.8, 1.0) and (0.5, 1.0)) leads to shorter translations. When halving the effect of the target context, most of the generated translations reach the maximum length, which is three times the length of source sentence in this work.
Figure 2(b) shows the results of manual evaluation on 200 source sentences randomly sampled from the test sets. Reducing the effect of source context (i.e., (0.8, 1.0) and (0.5, 1.0)) leads to more fluent yet less adequate translations. On the other hand, reducing the effect of target context (i.e., (1.0, 0.5) and (1.0, 0.8)) is expected to yield more adequate but less fluent translations. In this setting, the source words are translated (i.e., higher adequacy) while the translations are in wrong order (i.e., lower fluency). In practice, however, we observe the side effect that some source words are translated repeatedly until the translation reaches the maximum length (i.e., lower fluency), while others are left untranslated (i.e., lower adequacy). The reason is two fold:
NMT lacks a mechanism that guarantees that each source word is translated.444The recently proposed coverage based technique can alleviate this problem [Tu et al.2016]. In this work, we consider another approach, which is complementary to the coverage mechanism. The decoding state implicitly models the notion of “coverage” by recurrently reading the time-dependent source context . Lowering its contribution weakens the “coverage” effect and encourages the decoder to regenerate phrases multiple times to achieve the desired translation length.
The translation is incomplete. As shown in Table 1, NMT can get stuck in an infinite loop repeatedly generating a phrase due to the overwhelming influence of the source context. As a result, generation terminates early because the translation reaches the maximum length allowed by the implementation, even though the decoding procedure is not finished.
The quantitative (Figure 2) and qualitative (Table 1) results confirm our hypothesis, i.e., source and target contexts are highly correlated to translation adequacy and fluency. We believe that a mechanism that can dynamically select information from source context and target context would be useful for NMT models, and this is exactly the approach we propose.
Inspired by the success of gated units in RNN [Hochreiter and Schmidhuber1997, Cho et al.2014], we propose using context gates to dynamically control the amount of information flowing from the source and target contexts and thus balance the fluency and adequacy of NMT at each decoding step.
Intuitively, at each decoding step , the context gate looks at input signals from both the source (i.e., ) and target (i.e., and ) sides, and outputs a number between and for each element in the input vectors, where denotes “completely transferring this” while denotes “completely ignoring this”. The corresponding input signals are then processed with an element-wise multiplication before being fed to the activation layer to update the decoding state.
Formally, a context gate consists of a sigmoid neural network layer and an element-wise multiplication operation, as illustrated in Figure 3. The context gate assigns an element-wise weight to the input signals, computed by
is a logistic sigmoid function, and, , are the weight matrices. Again, , and are the dimensions of word embedding, decoding state, and source representation, respectively. Note that has the same dimensionality as the transferred input signals (e.g., ), and thus each element in the input vectors has its own weight.
Next, we consider how to integrate context gates into an NMT model.
The context gate can decide the amount of context information used in generating the next target word at each step of decoding. For example, after obtaining the partial translation “…new high level technology product”, the gate looks at the translation contexts and decides to depend more heavily on the source context. Accordingly, the gate assigns higher weights to the source context and lower weights to the target context and then feeds them into the decoding activation layer. This could correct inadequate translations, such as the missing translation of “guǎngdōng”, due to greater influence from the target context.
We have three strategies for integrating context gates into NMT that either affect one of the translation contexts or both contexts, as illustrated in Figure 4. The first two strategies are inspired by output gates in LSTMs [Hochreiter and Schmidhuber1997], which control the amount of memory content utilized. In these kinds of models, only affects either source context (i.e., ) or target context (i.e., and ):
Context Gate (source)
Context Gate (target)
where is an element-wise multiplication, and is the context gate calculated by Equation 4. This is also essentially similar to the reset gate in the GRU, which decides what information to forget from the previous decoding state before transferring that information to the decoding activation layer. The difference is that here the “reset” gate resets the context vector rather than the previous decoding state.
The last strategy is inspired by the concept of update gate from GRU, which takes a linear sum between the previous state and the candidate new state
. In our case, we take a linear interpolation between source and target contexts:
Context Gate (both)
Context gates are inspired by the gating scalar model proposed by Xu:2015:ICML for the image caption generation task. The essential difference lies in the task requirement:
In image caption generation, the source side (i.e., image) contains more information than the target side (i.e., caption). Therefore, they employ a gating scalar to scale only the source context.
In machine translation, both languages should contain equivalent information. Our model jointly controls the contributions from the source and target contexts. A direct interaction between input signals from both sides is useful for balancing adequacy and fluency of NMT.
Other differences in the architecture include:
Xu:2015:ICML uses a scalar that is shared by all elements in the source context, while we employ a gate with a distinct weight for each element. The latter offers the gate a more precise control of the context vector, since different elements retain different information.
We add peephole connections to the architecture, by which the source context controls the gate. It has been shown that peephole connections make precise timings easier to learn [Gers and Schmidhuber2000].
Our context gate also considers the previously generated word
as input. The most recently generated word can help the gate to better estimate the importance of target context, especially for the generation of function words in translations that may not have a corresponding word in the source sentence (e.g., “of” after “is fond”).
Experimental results (Section 5.4) show that these modifications consistently improve translation quality.
GRU or LSTM) as the activation function in the decoder. One might suspect that the context gate proposed in this work is somewhat redundant, given the existing gates that control the amount of information carried over from the previous decoding state(e.g., reset gate in GRU). We argue that they are in fact complementary: the context gate regulates the contextual information flowing into the decoding state, while the gated unit captures long-term dependencies between decoding states. Our experiments confirm the correctness of our hypothesis: the context gate not only improves translation quality when compared to a conventional RNN unit (e.g., an element-wise ), but also when compared to a gated unit of GRU, as shown in Section 5.2.
Recently, Tu:2016:ACL propose adding a coverage mechanism into NMT to alleviate over-translation and under-translation problems, which directly affect translation adequacy. They maintain a coverage vector to keep track of which source words have been translated. The coverage vector is fed to the attention model to help adjust future attention. This guides NMT to focus on the un-translated source words while avoiding repetition of source content. Our approach is complementary: the coverage mechanism produces a better source context representation, while our context gate controls the effect of the source context based on its relative importance. Experiments in Section 5.2 show that combining the two methods can further improve translation performance. There is another difference as well: the coverage mechanism is only applicable to attention-based NMT models, while the context gate is applicable to all NMT models.
A thread of work in language modeling (LM) attempts to exploit auxiliary sentence-level or document-level context in an RNN LM [Mikolov and Zweig2012, Ji et al.2015, Wang and Cho2016]. Independent of our work, Wang:2016:ACL propose “early fusion” models of RNNs where additional information from an inter-sentence context is “fused” with the input to the RNN. Closely related to Wang:2016:ACL, our approach aims to dynamically control the contributions of required source and target contexts for machine translation, while theirs focuses on integrating auxiliary corpus-level contexts for language modelling to better approximate the corpus-level probability. In addition, we employ a gating mechanism to produce a dynamic weight at different decoding steps to combine source and target contexts, while they do a linear combination of intra-sentence and inter-sentence contexts with static weights. Experiments in Section 5.2 show that our gating mechanism significantly outperforms linear interpolation when combining contexts.
In machine translation, there are certain syntactic elements of the target language that are missing in the source (i.e., null-generated words). In fact this was the preliminary motivation for our approach: current attention models lack a mechanism to control the generation of words that do not have a strong correspondence on the source side. The model structure of NMT is quite similar to the traditional word-based SMT [Brown et al.1993]. Therefore, techniques that have proven effective in SMT may also be applicable to NMT. Toutanova:2002:EMNLP extend the calculation of translation probabilities to include null-generated target words in word-based SMT. These words are generated based on both the special source token null and the neighbouring word in the target language by a mixture model. We have simplified and generalized their approach: we use context gates to dynamically control the contribution of source context. When producing null-generated words, the context gate can assign lower weights to the source context, by which the source-side information have less influence. In a sense, the context gate relieves the need for a null state in attention.
|3||2 + Context Gate ()||80.7M||30.86||30.85||24.71||28.81|
|5||4 + Context Gate ()||87.9M||31.96||32.29||24.97||29.74|
|6||4 + Context Gate ()||87.9M||32.38||32.11||23.78||29.42|
|7||4 + Context Gate ()||87.9M||33.52||33.46||24.85||30.61|
|9||8 + Context Gate ()||88.0M||34.13||34.83||26.22||31.73|
We carried out experiments on Chinese-English translation. The training dataset consisted of 1.25M sentence pairs extracted from LDC corpora555The corpora include LDC2002E18, LDC2003E07, LDC2003E14, Hansards portion of LDC2004T07, LDC2004T08 and LDC2005T06., with 27.9M Chinese words and 34.5M English words respectively. We chose the NIST 2002 (MT02) dataset as the development set, and the NIST 2005 (MT05), 2006 (MT06) and 2008 (MT08) datasets as the test sets. We used the case-insensitive 4-gram NIST BLEU score [Papineni et al.2002]
as the evaluation metric, andsign-test [Collins et al.2005] for the statistical significance test.
For efficient training of the neural networks, we limited the source and target vocabularies to the most frequent 30K words in Chinese and English, covering approximately 97.7% and 99.3% of the data in the two languages respectively. All out-of-vocabulary words were mapped to a special token UNK. We trained each model on sentences of length up to 80 words in the training data. The word embedding dimension was 620 and the size of a hidden layer was 1000. We trained our models until the BLEU score on the development set stops improving.
We compared our method with representative SMT and NMT666There is some recent progress on aggregating multiple models or enlarging the vocabulary(e.g.,, in [Jean et al.2015]), but here we focus on the generic models. models:
Moses [Koehn et al.2007]: an open source phrase-based translation system with default configuration and a 4-gram language model trained on the target portion of training data;
GroundHog [Bahdanau et al.2015]: an open source attention-based NMT model with default setting. We have two variants that differ in the activation function used in the decoder RNN: 1) GroundHog () uses a simple function as the activation function, and 2) GroundHog () uses a sophisticated gate function ;
|GroundHog vs. GroundHog+Context Gate|
Table 2 shows the translation performances in terms of BLEU scores. We carried out experiments on multiple NMT variants. For example, “2 + Context Gate ()” in Row 3 denotes integrating “Context Gate ()” into the baseline in Row 2 (i.e., GroundHog ()). For baselines, we found that the gated unit (i.e., , Row 4) indeed surpasses its vanilla counterpart (i.e., , Row 2), which is consistent with the results in other work [Chung et al.2014]. Clearly the proposed context gates significantly improve the translation quality in all cases, although there are still considerable differences among the variants:
Context gates introduce a few new parameters. The newly introduced parameters include , , in Equation 4. In this work, the dimensionality of the decoding state is , the dimensionality of the word embedding is , and the dimensionality of context representation is . The context gates only introduce 3.6M additional parameters, which is quite small compared to the number of parameters in the existing models (e.g., 84.3M in the “GroundHog ()”).
We first carried out experiments on a simple decoder without gating function (Rows 2 and 3), to better estimate the impact of context gates. As shown in Table 2, the proposed context gate significantly improved translation performance by 4.2 BLEU points on average. It is worth emphasizing that context gate even outperforms a more sophisticated gating function (i.e., GRU in Row 4). This is very encouraging, since our model only has a single gate with half of the parameters (i.e., 3.6M versus 7.2M) and less computations (i.e., half the matrix computations to update the decoding state888We only need to calculate the context gate once via Equation 4 and then apply it when updating the decoding state. In contrast, GRU requires the calculation of an update gate, a reset gate, a proposed updated decoding state and an interpolation between the previous state and the proposed state. Please refer to [Cho et al.2014] for more details.).
We then investigated the effect of the context gates on a standard NMT with GRU as the decoding activation function (Rows 4-7). Several observations can be made. First, context gates also boost performance beyond the GRU in all cases, demonstrating our claim that context gates are complementary to the reset and update gates in GRU. Second, jointly controlling the information from both translation contexts consistently outperforms its single-side counterparts, indicating that a direct interaction between input signals from the source and target contexts is useful for NMT models.
We finally tested on a stronger baseline, which employs a coverage mechanism to indicate whether or not a source word has already been translated [Tu et al.2016]. Our context gate still achieves a significant improvement of 1.6 BLEU points on average, reconfirming our claim that the context gate is complementary to the improved attention model that produces a better source context representation. Finally, our best model (Row 7) outperforms the SMT baseline system using the same data (Row 1) by 3.3 BLEU points.
From here on, we refer to “GroundHog” for “GroundHog ()”, and “Context Gate” for “Context Gate ()” if not otherwise stated.
We also conducted a subjective evaluation of the benefit of incorporating context gates. Two human evaluators were asked to compare the translations of 200 source sentences randomly sampled from the test sets without knowing which system produced each translation. Table 3 shows the results of subjective evaluation. The two human evaluators made similar judgments: in adequacy, around 30% of GroundHog translations are worse, 52% are equal, and 18% are better; while in fluency, around 29% are worse, 52% are equal, and 19% are better.
|+ Context Gate||67.43||55.52|
|+ Context Gate||63.80||49.40|
where is a candidate alignment, and and are the sets of sure and possible links in the reference alignment respectively (). denotes the alignment matrix, and for both and we assign the elements that correspond to the existing links in and probability and the other elements probability . In this way, we are able to better evaluate the quality of the soft alignments produced by attention-based NMT.
We find that context gates do not improve alignment quality when used alone. When combined with coverage mechanism, however, it produces better alignments, especially one-to-one alignments by selecting the source word with the highest alignment probability per target word (i.e., AER score). One possible reason is that better estimated decoding states (from the context gate) and coverage information help to produce more concentrated alignments, as shown in Figure 6.
|2||1 + Gating Scalar||31.62||31.48||23.85||28.98|
|3||1 + Context Gate ()||31.69||31.63||24.25||29.19|
|4||1 + Context Gate ()||32.15||32.05||24.39||29.53|
Table 5 shows a detailed analysis of architecture components measured in BLEU scores. Several observations can be made:
Operation Granularity (Rows 2 and 3): Element-wise multiplication (i.e., Context Gate ()) outperforms the vector-level scalar (i.e., Gating Scalar), indicating that precise control of each element in the context vector boosts translation performance.
Gate Strategy (Rows 3 and 4): When only fed with the previous decoding state , Context Gate () consistently outperforms Context Gate (), showing that jointly controlling information from both source and target sides is important for judging the importance of the contexts.
Peephole connections (Rows 4 and 5): Peepholes, by which the source context controls the gate, play an important role in the context gate, which improves the performance by 0.57 in BLEU score.
Previously generated word (Rows 5 and 6): Previously generated word provides a more explicit signal for the gate to judge the importance of contexts, leading to a further improvement on translation performance.
We follow Bahdanau:2015:ICLR and group sentences of similar lengths together. Figure 7 shows the BLEU score and the averaged length of translations for each group. GroundHog performs very well on short source sentences, but degrades on long source sentences (i.e., ), which may be due to the fact that source context is not fully interpreted. Context gates can alleviate this problem by balancing the source and target contexts, and thus improve decoder performance on long sentences. In fact, incorporating context gates boost translation performance on all source sentence groups.
We confirm that context gate weight correlates well with translation performance. In other words, translations that contain higher (i.e., source context contributes more than target context) at many time steps are better in translation performance. We used the mean of the sequence as the gate weight of each sentence. We calculated the Pearson Correlation between the sentence-level gate weight and the corresponding improvement on translation performance (i.e., BLEU, adequacy, and fluency scores),999We use the average of correlations on subjective evaluation metrics (i.e., adequacy and fluency) by two evaluators. as shown in Table 6. We observed that context gate weight is positively correlated with translation performance improvement and that the correlation is higher on long sentences.
As an example, consider this source sentence from the test set:
zhōuliù zhèngshì yīngguó mínzhòng dào chāoshì cǎigòu de gāofēng shíkè, dāngshí 14 jiā chāoshì de guānbì lìng yīngguó zhè jiā zuì dà de liánsuǒ chāoshì sǔnshī shùbǎiwàn yīngbàng de xiāoshòu shōurù .
GroundHog translates it into:
twenty - six london supermarkets were closed at a peak hour of the british population in the same period of time .
which almost misses all the information of the source sentence. Integrating context gates improves the translation adequacy:
this is exactly the peak days British people buying the supermarket . the closure of the 14 supermarkets of the 14 supermarkets that the largest chain supermarket in england lost several million pounds of sales income .
Coverage mechanisms further improve the translation by rectifying over-translation (e.g., “of the 14 supermarkets”) and under-translation (e.g., “saturday” and “at that time”):
saturday is the peak season of british people ’s purchases of the supermarket . at that time , the closure of 14 supermarkets made the biggest supermarket of britain lose millions of pounds of sales income .
We find that source and target contexts in NMT are highly correlated to translation adequacy and fluency, respectively. Based on this observation, we propose using context gates in NMT to dynamically control the contributions from the source and target contexts in the generation of a target sentence, to enhance the adequacy of NMT. By providing NMT the ability to choose the appropriate amount of information from the source and target contexts, one can alleviate many translation problems from which NMT suffers. Experimental results show that NMT with context gates achieves consistent and significant improvements in translation quality over different NMT models.
Context gates are in principle applicable to all sequence-to-sequence learning tasks in which information from the source sequence is transformed to the target sequence (corresponding to adequacy) and the target sequence is generated (corresponding to fluency). In the future, we will investigate the effectiveness of context gates to other tasks, such as dialogue and summarization. It is also necessary to validate the effectiveness of our approach on more language pairs and other NMT architectures (e.g., using LSTM as well as GRU, or multiple layers).
This work is supported by China National 973 project 2014CB340301. Yang Liu is supported by the National Natural Science Foundation of China (No. 61522204) and the 863 Program (2015AA015407). We thank action editor Chris Quirk and three anonymous reviewers for their insightful comments.