U NSUPERVISED M ACHINE T RANSLATION U SING M ONOLINGUAL C ORPORA O NLY
Anonymous authors
Paper under double-blind review
A BSTRACT
Machine translation has recently achieved impressive performance thanks to recent advances in deep learning and the availability of large-scale parallel corpora.
There have been numerous attempts to extend these successes to low-resource language pairs, yet requiring tens of thousands of parallel sentences.
In this work, we take this research direction to the extreme and investigate whether it is possible to learn to translate even without any parallel data.
We propose a model that takes sentences from monolingual corpora in two different languages and maps them into the same latent space.
By learning to reconstruct in both languages from this shared feature space, the model effectively learns to translate without using any labeled data.
We demonstrate our model on two widely used datasets and two language pairs, reporting BLEU scores of 32.8 and 15.1 on the Multi30k and WMT
English-French datasets, without using even a single parallel sentence at training time.
I NTRODUCTION
Thanks to recent advances in deep learning (Sutskever et al., 2014; Bahdanau et al., 2015) and the availability of large-scale parallel corpora, machine translation has now reached impressive performance on several language pairs (Wu et al., 2016).
However, these models work very well only when provided with massive amounts of parallel data, in the order of millions of parallel sentences.
Unfortunately, parallel corpora are costly to build as they require specialized expertise, and are often nonexistent for low-resource languages.
Conversely, monolingual data is much easier to ﬁnd, and many languages with limited parallel data still possess signiﬁcant amounts of monolingual data.
There have been several attempts at leveraging monolingual data to improve the quality of machine translation systems in a semi-supervised setting (Munteanu et al., 2004; Irvine, 2013; Irvine & Callison-Burch, 2015; Zheng et al., 2017).
Most notably, Sennrich et al.
(2015a) proposed a very effective data-augmentation scheme, dubbed “back-translation”, whereby an auxiliary translation system from the target language to the source language is ﬁrst trained on the available parallel data, and then used to produce translations from a large monolingual corpus on the target side.
The pairs composed of these translations with their corresponding ground truth targets are then used as additional training data for the original translation system.
Another way to leverage monolingual data on the target side is to augment the decoder with a language model (Gulcehre et al., 2015).
And ﬁnally, Cheng et al.
; He et al.
(2016) have proposed to add an auxiliary auto-encoding task on monolingual data, which ensures that a translated sentence can be translated back to the original one.
All these works still rely on several tens of thousands parallel sentences, however.
Previous work on zero-resource machine translation has also relied on labeled information, not from the language pair of interest but from other related language pairs (Firat et al., 2016;
Johnson et al., 2016; Chen et al., 2017) or from other modalities (Nakayama & Nishida, 2017; Lee et al., 2017).
The only exception is the work by Ravi & Knight (2011); Pourdamghani & Knight (2017), where the machine translation problem is reduced to a deciphering problem.
Unfortunately, their method is limited to rather short sentences and it has only been demonstrated on a very simplistic setting comprising of the most frequent short sentences, or very closely related languages.
In this paper, we investigate whether it is possible to train a general machine translation system without any form of supervision whatsoever.
The only assumption we make is that there exists a monolingual corpus on each language.
This set up is interesting for a twofold reason.
First, this is applicable whenever we encounter a new language pair for which we have no annotation.
Second, it provides a strong lower bound performance on what any good semi-supervised approach is expected to yield.
The key idea is to build a common latent space between the two languages (or domains) and to learn to translate by reconstructing in both domains according to two principles: (i) the model has to be able to reconstruct a sentence in a given language from a noisy version of it, as in standard denoising auto-encoders (Vincent et al., 2008).
(ii)
The model also learns to reconstruct any source sentence given a noisy translation of the same sentence in the target domain, and vice versa.
For (ii), the translated sentence is obtained by using a back-translation procedure (Sennrich et al., 2015a), i.e. by using the learned model to translate the source sentence to the target domain.
In addition to these reconstruction objectives, we constrain the source and target sentence latent representations to have the same distribution using an adversarial regularization term, whereby the model tries to fool a discriminator which is simultaneously trained to identify the language of a given latent sentence representation (Ganin et al., 2016).
This procedure is then iteratively repeated, giving rise to translation models of increasing quality.
To keep our approach fully unsupervised, we initialize our algorithm by using a na ¨ ıve unsupervised translation model based on a word by word translation of sentences with a bilingual lexicon derived from the same monolingual data (Conneau et al., 2017).
As a result, and by only using monolingual data, we can encode sentences of both languages into the same feature space, and from there, we can also decode/translate in any of these languages; see
Figure 1 for an illustration.
While not being able to compete with supervised approaches using lots of parallel resources, we show in section 4 that our model is able to achieve remarkable performance.
For instance, on the
WMT dataset we can achieve the same translation quality of a similar machine translation system trained with full supervision on 100,000 sentence pairs.
On the Multi30K-Task1 dataset we achieve a BLEU above 22 on all the language pairs, with up to 32.76 on English-French.
Next, in section 2, we describe the model and the training algorithm.
We then present experimental results in section 4.
Finally, we further discuss related work in section 5 and summarize our ﬁndings in section 6.
U NSUPERVISED N EURAL M ACHINE T RANSLATION
In this section, we ﬁrst describe the architecture of the translation system, and then we explain how we train it.
2. N EURAL M ACHINE T RANSLATION M ODEL
The translation model we propose is composed of an encoder and a decoder, respectively responsible for encoding source and target sentences to a latent space, and to decode from that latent space to the source or the target domain.
We use a single encoder and a single decoder for both domains (Johnson et al., 2016).
The only difference when applying these modules to different languages is the choice of lookup tables.
Let us denote by W S the set of words in the source domain associated with the (learned) words embeddings Z S
=
( z s 1 , ...., z s |W S | ) , and by W T the set of words in the target domain associated with the embeddings Z T =
( z t 1 , ...., z t |W T | ) , Z being the set of all the embeddings.
Given an input sentence of m words x = ( x 1 , x 2 , ..., x m ) in a particular language (cid:96) , (cid:96) ∈ { src, tgt } , an encoder e θ enc , Z ( x , (cid:96) ) computes a sequence of m hidden states z
= ( z 1 , z 2 , ..., z m ) by using the corresponding word embeddings, i.e. Z S if (cid:96) =
src and Z T if (cid:96) =
tgt ; the other parameters θ enc are instead shared between the source and target languages.
For the sake of simplicity, the encoder will be denoted as e ( x , (cid:96) ) in the following.
These hidden states are vectors in R n , n being the dimension of the latent space.
A decoder d θ dec , Z ( z , (cid:96) ) takes as input z and a language (cid:96) , and generates an output sequence
y =
( y 1 , y 2 , ..., y k ) , where each word y i is in the corresponding vocabulary W (cid:96) .
This decoder makes use of the corresponding word embeddings, and it is otherwise parameterized by a vector θ dec that does not depend on the output language.
It will thus be denoted d
( z , (cid:96) ) in the following.
To generate an output word y i , the decoder iteratively takes as input the previously generated word y i − 1 ( y 0 being a start symbol which is language dependent), updates its internal state, and returns the word that has the highest probability of being the next one.
The process is repeated until the decoder generates a stop symbol indicating the end of the sequence.
In this article, we use a sequence-to-sequence model with attention (Bahdanau et al., 2015), without input-feeding.
The encoder is a bidirectional-LSTM which returns a sequence of hidden states z
= ( z 1 , z 2 , ..., z m ) .
At each step, the decoder, which is also an LSTM, takes as input the previous hidden state, the current word and a context vector given by a weighted sum over the encoder states.
In all the experiments we consider, both encoder and decoder have 3 layers.
The LSTM layers are shared between the source and target encoder, as well as between the source and target decoder.
We also share the attention weights between the source and target decoder.
The embedding and LSTM hidden state dimensions are all set to 300.
Sentences are generated using greedy decoding.
2. O VERVIEW OF THE M ETHOD
We consider a dataset of sentences in the source domain, denoted by D src , and another dataset in the target domain, denoted by D tgt .
These datasets do not correspond to each other, in general.
We train the encoder and decoder by reconstructing a sentence in a particular domain, given a noisy version of the same sentence in the same or in the other domain.
At a high level, the model starts with an unsupervised na¨ıve translation model obtained by making word-by-word translation of sentences using a parallel dictionary learned in an unsupervised way (Conneau et al., 2017).
Then, at each iteration, the encoder and decoder are trained by minimizing an objective function that measures their ability to both reconstruct and translate from a noisy version of an input training sentence.
This noisy input is obtained by dropping and swapping words in the case of the auto-encoding task, while it is the result of a translation with the model at the previous iteration in the case of the translation task.
In order to promote alignment of the latent distribution of sentences in the source and the target domains, our approach also simultaneously learns a discriminator in an adversarial setting.
The newly learned encoder/decoder are then used at the next iteration to generate new translations, until convergence of the algorithm.
At test time and despite the lack of parallel data at training time, the encoder and decoder can be composed into a standard machine translation system.
2. D ENOISING AUTO - ENCODING
Training an autoencoder of sentences is a trivial task, if the sequence-to-sequence model is provided with an attention mechanism like in our work 1 .
Without any constraint, the auto-encoder very quickly learns to merely copy every input word one by one.
Such a model would also perfectly copy sequences of random words, suggesting that the model does not learn any useful structure in the data.
To address this issue, we adopt the same strategy of Denoising Auto-encoders (DAE) (Vincent et al., 2008)), and add noise to the input sentences (see Figure 1-left), similarly to Hill et al.
Considering a domain (cid:96) =
src or (cid:96) =
tgt , and a stochastic noise model denoted by C which operates on sentences, we deﬁne the following objective function:
[Equation]
where ˆ x ∼ d ( e ( C ( x ) , (cid:96) ) , (cid:96) ) means that ˆ x is a reconstruction of the corrupted version of x , with x sampled from the monolingual dataset D (cid:96) .
In this equation, ∆ is a measure of discrepancy between the two sequences, the sum of token-level cross-entropy losses in our case.
Noise model C ( x ) is a randomly sampled noisy version of sentence x .
In particular, we add two different types of noise to the input sentence.
First, we drop every word in the input sentence with a probability p wd .
Second, we slightly shufﬂe the input sentence.
To do so, we apply a random permutation σ to the input sentence, verifying the condition ∀ i ∈ { 1 , n } , | σ ( i ) − i | ≤ k where n is the length of the input sentence, and k is a tunable parameter.
To generate a random permutation verifying the above condition for a sentence of size n , we generate a random vector q of size n , where q i = i + U 3(0 , α ) , and U is a draw from the uniform distribution in the speciﬁed range.
Then, we deﬁne σ to be the permutation that sorts the array q .
In particular, α < 1 will return the identity, α = + ∞ can return any permutation, and α = k + 1 will return permutations σ verifying ∀ i ∈ { 1 , n } , | σ ( i ) − i | ≤ k .
Although biased, this method generates permutations similar to the noise observed with word-by-word translation.
In our experiments, both the word dropout and the input shufﬂing strategies turned out to have a critical impact on the results, see also section 4.5, and using both strategies at the same time gave us the best performance.
In practice, we found p wd = 0 .
1 and k = 3 to be good parameters.
2. C ROSS D OMAIN T RAINING
The second objective of our approach is to constrain the model to be able to map an input sentence from a the source/target domain (cid:96) 1 to the target/source domain (cid:96) 2 , which is what we are ultimately interested in at test time.
The principle here is to sample a sentence x ∈ D (cid:96) 1 , and to generate a corrupted translation of this sentence in (cid:96) 2 .
This corrupted version is generated by applying the current translation model denoted M to x such that y
= M ( x ) .
Then a corrupted version C ( y ) is sampled (see Figure 1-right).
The objective is thus to learn the encoder and the decoder such that they can reconstruct x from C ( y ) .
The cross-domain loss can be written as:
[Equation]
where ∆ is again the sum of token-level cross-entropy losses.
2. A DVERSARIAL TRAINING
Intuitively, the decoder of a neural machine translation system works well only when its input is produced by the encoder it was trained with, or at the very least, when that input comes from a distribution very close to the one induced by its encoder.
Therefore, we would like our encoder to output features in the same space regardless of the actual language of the input sentence.
If such condition is satisﬁed, our decoder may be able to decode in a certain language regardless of the language of the encoder input sentence.
Note however that the decoder could still produce a bad translation while yielding a valid sentence in the target domain, as constraining the encoder to map two languages in the same feature space does not imply a strict correspondence between sentences.
Fortunately, the previously introduced loss for cross-domain training in equation 2 mitigates this concern.
Also, recent work on bilingual lexical induction has shown that such a constraint is very effective at the word level, suggesting that it may also work at the sentence level, as long as the two latent representations exhibit strong structure in feature space.
In order to add such a constraint, we train a neural network, which we will refer to as the discriminator , to classify between the encoding of source sentences and the encoding of target sentences (Ganin et al., 2016).
The discriminator operates on the output of the encoder, which is a sequence of latent vectors ( z 1 , ..., z m ) , with z i ∈ R n , and produces a binary prediction about the language of the encoder input sentence: p D ( l | z 1 , ..., z m )
∝ m (cid:81)
j =1 p D ( (cid:96) | z j ) , with p D : R n → [0; 1] , where 0 corresponds to the source domain, and 1 to the target domain.
The discriminator is trained to predict the language by minimizing the following cross-entropy loss:
L
[Equation]
e
[Equation]
x i , (cid:96) i ))] , where ( x i , (cid:96) i ) corresponds to sentence and language id pairs uniformly sampled from the two monolingual datasets, θ D are the parameters of the discriminator, θ enc are the parameters of the encoder, and Z are the encoder word embeddings.
The encoder is trained instead to fool the discriminator:
[Equation]
(3) with (cid:96) j = (cid:96) 1 if (cid:96) i = (cid:96) 2 , and vice versa.
Final Objective function The ﬁnal objective function at one iteration of our learning algorithm is thus:
[Equation]
where λ auto , λ cd , and λ adv are hyper-parameters weighting the importance of the auto-encoding, cross-domain and adversarial loss.
In parallel, the discriminator loss L D is minimized to update the discriminator.
T RAINING
In this section we describe the overall training algorithm and the unsupervised criterion we used to select hyper-parameters.
3. I TERATIVE T RAINING
The ﬁnal learning algorithm is described in Algorithm 1 and the general architecture of the model is shown in Figure 2.
As explained previously, our model relies on an iterative algorithm which starts from an initial translation model M (1) (line 3).
This is used to translate the available monolingual data, as needed by the cross-domain loss function of Equation 2.
At each iteration, a new encoder and decoder are trained by minimizing the loss of Equation 4 – line 7 of the algorithm.
Then, a new translation model M ( t +1) is created by composing the resulting encoder and decoder, and the process repeats.
To jump start the process, M (1) simply makes a word-by-word translation of each sentence using a parallel dictionary learned using the unsupervised method proposed by Conneau et al.
(2017), which only leverages monolingual data.
The intuition behind our algorithm is that as long as the initial translation model M (1) retains at least some information of the input sentence, the encoder will map such translation into a representation in feature space that also corresponds to a cleaner version of the input, since the encoder is trained to denoise.
At the same time, the decoder is trained to predict noiseless outputs, conditioned on noisy features.
Putting these two pieces together will produce less noisy translations, which will enable better back-translations at the next iteration, and so on so forth.
1: procedure T RAINING ( D src , D tgt , T ) 2:
Infer bilingual dictionary using monolingual data (Conneau et al., 2017) 3:
M (1) ← unsupervised word-by-word translation model using the inferred dictionary 4: for t = 1 , T do 5: using M ( t ) , translate each monolingual dataset 6: // discriminator training & model training as in eq.7: θ discr ← arg min L D , θ enc , θ dec , Z ← arg min L 8:
M ( t +1) ← e ( t ) ◦
d ( t ) // update MT model 9: end for 10: return M ( T +1) 11: end procedure
3. U NSUPERVISED M ODEL S ELECTION C RITERION
In order to select hyper-parameters, we wish to have a criterion correlated with the translation quality.
However, we do not have access to parallel sentences to judge how well our model translates, not even at validation time.
Therefore, we propose the surrogate criterion which we show correlates well with BLEU (Papineni et al., 2002), the metric we care about at test time.
For all sentences x in a domain (cid:96) 1 , we translate these sentences to the other domain (cid:96) 2 , and then translate the resulting sentences back to (cid:96) 1 .
The quality of the model is then evaluated by computing the BLEU score over the original inputs and their reconstructions via this two-step translation process.
The performance is then averaged over the two directions, and the selected model is the one with the highest average score.
Given an encoder e , a decoder d and two non-parallel datasets D src and D tgt , we denote
M src → tgt ( x ) =
d ( e ( x, src ) , tgt ) the translation model from src to tgt , and M tgt → src the model in the opposite direction.
Our model selection criterion MS ( e, d, D src , D tgt ) is:
[Equation]
Figure 3 shows a typical example of the correlation between this measure and the ﬁnal translation model performance (evaluated here using a parallel dataset).
The unsupervised model selection criterion is used both to a) determine when to stop training and b) to select the best hyper-parameter setting across different experiments.
In the former case, the
BLEU score of tar to source
English-French dataset as a function of the number of passes through the dataset at iteration ( t )
= 1 of the algorithm (training M (2) given M (1) ).
BLEU correlates very well with the proposed model selection criterion, see Equation 5.
Spearman correlation coefﬁcient between the proposed criterion and BLEU on the test set is 0.95 in average.
In the latter case, the coefﬁcient is in average 0.75, which is ﬁne but not nearly as good.
For instance, the BLEU score on the test set of models selected with the unsupervised criterion are sometimes up to 1 or 2 BLEU points below the score of models selected using a small validation set of 500 parallel sentences.
E XPERIMENTS
In this section, we ﬁrst describe the datasets and the pre-processing we used, then we introduce the baselines we considered, and ﬁnally we report the extensive empirical validation proving the effectiveness of our method.
We will release the code to the public once the revision process is over.
4. D ATASETS
In our experiments, we consider the English-French and English-German language pairs, on three different datasets.
WMT’14 English-French
We use the full training set of 36 million pairs, we lower-case them and remove sentences longer than 50 words, as well as pairs with a source/target length ratio above 1.5, resulting in a parallel corpus of about 30 million sentences.
Next, we build monolingual corpora by selecting the English sentences from 15 million random pairs, and selecting the French sentences from the complementary set.
The former set constitutes our English monolingual dataset.
The latter set is our French monolingual dataset.
The lack of overlap between the two sets ensures that there is not exact correspondence between examples in the two datasets.
The validation set is comprised of 3,000 English and French sentences extracted from our monolingual training corpora described above.
These sentences are not the translation of each other, and they will be used by our unsupervised model selection criterion, as explained in 3.2.
Finally, we report results on the full newstest2014 dataset.
WMT’16 English-German
We follow the same procedure as above to create monolingual training and validation corpora in English and German, which results in two monolingual training corpora of 1.8 million sentences each.
We test our model on the newstest2016 dataset.
Multi30k-Task
The task 1 of the Multi30k dataset (Elliott et al., 2016) has 30,000 images, with annotations in English, French and German, that are translations of each other.
We consider the
English-French and English-German pairs.
We disregard the images and only consider the parallel annotations, with the provided training, validation and test sets, composed of 29,000, 1,000 and 1,000 pairs of sentences respectively.
For both pairs of languages and similarly to the WMT datasets [Table] above, we split the training and validation sets into monolingual corpora, resulting in 14,500 monolingual source and target sentences in the training set, and 500 sentences in the validation set.
Table 1 summarizes the number of monolingual sentences in each dataset, along with the vocabulary size.
To limit the vocabulary size on the WMT en-fr and WMT de-en datasets, we only considered words with more than 100 and 25 occurrences, respectively.
4. B ASELINES
Word-by-word translation (WBW)
The ﬁrst baseline is a system that performs word-by-word translations of the input sentences using the inferred bilingual dictionary (Conneau et al., 2017).
This baseline provides surprisingly good results for related language pairs, like English-French, where the word order is similar, but performs rather poorly on more distant pairs like English-German, as can be seen in Table 2.
Word reordering (WR) After translating word-by-word as in WBW, here we reorder words using an LSTM-based language model trained on the target side.
Since we cannot exhaustively score every possible word permutation (some sentences have about 100 words), we consider all pairwise swaps of neighboring words, we select the best swap, and iterate ten times.
We use this baseline only on the WMT dataset that has a large enough monolingual data to train a language model.
Oracle Word Reordering (OWR)
Using the reference, we produce the best possible generation using only the words given by WBW.
The performance of this method is an upper-bound of what any model could do without replacing words.
Supervised Learning We ﬁnally consider exactly the same model as ours, but trained with supervision, using the standard cross-entropy loss on the original parallel sentences.
4. U NSUPERVISED DICTIONARY LEARNING
To implement our baseline and also to initialize the embeddings Z of our model, we ﬁrst train word embeddings on the source and target monolingual corpora using fastText (Bojanowski et al., 2017), and then we apply the unsupervised method proposed by Conneau et al.
(2017) to infer a bilingual dictionary which can be use for word-by-word translation.
Since WMT yields a very large-scale monolingual dataset, we obtain very high-quality embeddings and dictionaries, with an accuracy of 84 .
48% and 77 .
29% on French-English and GermanEnglish, which is on par with what could be obtained using a state-of-the-art supervised alignment method (Conneau et al., 2017).
On the Multi30k datasets instead, the monolingual training corpora are too small to train good word embeddings (more than two order of magnitude smaller than WMT).
We therefore learn word vectors on Wikipedia using fastText 2 . [Table]
E XPERIMENTAL D ETAILS
Discriminator Architecture The discriminator is a multilayer perceptron with three hidden layers of size 1024, Leaky-ReLU activation functions and an output logistic unit.
Following Goodfellow (2016), we include a smoothing coefﬁcient s = 0 .
1 in the discriminator predictions.
Training Details
The encoder and the decoder are trained using Adam (Kingma & Ba, 2014), with a learning rate of 0 . 0003 , β 1 = 0 . 5 , and a mini-batch size of 32 .
The discriminator is trained using
RMSProp (Tieleman & Hinton, 2012) with a learning rate of 0 . 0005 .
We evenly alternate between one encoder-decoder and one discriminator update.
We set λ auto = λ cd = λ
adv = 1 .
4. E XPERIMENTAL R ESULTS
Table 2 shows the BLEU scores achieved by our model and the baselines we considered.
First, we observe that word-by-word translation is surprisingly effective when translating into English, obtaining a BLEU score of 16.77 and 10.09 for fr-en on respectively Multi30k-Task1 and WMT datasets.
Word-reordering only slightly improves upon word-by-word translation.
Our model instead, clearly outperforms these baselines, even on the WMT dataset which has more diversity of topics and sentences with much more complicated structure.
After just one iteration, we obtain a BLEU score of 27.48 and 12.10 for the en-fr task.
Interestingly, we do even better than oracle reordering, suggesting that our model not only reorders but also correctly substitutes some words.
After a few iterations, our model obtains BLEU of 32.76 and 15.05 on Multi30k-Task1 and WMT datasets for the English to French task, which is rather remarkable.
Comparison with supervised approaches Here, we assess how much labeled data are worth our two large monolingual corpora.
On WMT, we trained the very same NMT architecture on both language pairs, but with supervision using various amounts of parallel data.
Figure 4-right shows the resulting performance.
Our unsupervised approach obtains the same performance than a supervised NMT model trained on about 100,000 parallel sentences, which is impressive.
Of course, adding more parallel examples allows the supervised approach to outperform our method, but the good performance of our unsupervised method suggests that it could be very effective for low-resources languages where no parallel data are available.
Moreover, these results open the door to the development of semi-supervised translation models, which will be the focus of future investigation.
With a phrase-based machine translation system, we obtain 21.6 and 22.4 BLEU on WMT en-fr and fr-en , which is better than the supervised NMT baseline we report for that same amount of parallel sentences, which is 16.8 and 16.4 respectively.
However, if we train the same supervised NMT model with BPE (Sennrich et al., 2015b), we obtain 22.6 BLEU for en-fr , suggesting that our results on unsupervised machine translation could also be improved by using
BPE, as this removes unknown words (about 9% of the words in de-en are replaced by the unknown token otherwise).
Iterative Learning Figure 4-left illustrates the quality of the learned model after each iteration of the learning process in the language pairs of Multi30k-Task1 dataset, other results being provided in Table 2.
One can see that the quality of the obtained model is high just after the ﬁrst iteration [Table] of the process.
Subsequent iterations yield signiﬁcant gains although with diminishing returns.
At iteration 3, the performance gains are marginal, showing that our approach quickly converges.
Table 3 shows examples of translations of three sentences on the Multi30k dataset, as we iterate.
Iteration 0 corresponds to the word-by-word translation obtained with our cross-lingual dictionary, which clearly suffers from word order issues.
We can observe that the quality of the translations increases at every iteration.
Ablation Study We perform an ablation study to understand the importance of the different components of our system.
To this end, we have trained multiple versions of our model with some missing components: the discriminator, the cross-domain loss, the auto-encoding loss, etc.
Tableshows that the best performance is obtained with the simultaneous use of all the described elements. [Table]
The most critical component is the unsupervised word alignment technique, either in the form of a back-translation dataset generated using word-by-word translation, or in the form of pretrained embeddings which enable to map sentences of different languages in the same latent space.
On the English-French pair of Multi30k-Task1, with a back-translation dataset but without pretrained embeddings, our model obtains a BLEU score of 25.29 and 26.10, which is only a few points below the model using all components.
Similarly, when the model uses pretrained embeddings but no back-translation dataset (when λ cd = 0), it obtains 25.44 and 27.14.
On the other hand, a model that does not use any of these components only reaches 8.78 and 9.15 BLEU.
The adversarial component also signiﬁcantly improves the performance of our system, with a difference of up to 5.33 BLEU in the French-English pair of Multi30k-Task1.
This conﬁrms our intuition that, to really beneﬁt from the cross-domain loss, one has to ensure that the distribution of latent sentence representations is similar across the two languages.
Without the auto-encoding loss (when λ auto = 0 ), the model only obtains 20.02, which is 8.05 BLEU points below the method using all components.
Finally, performance is greatly degraded also when the corruption process of the input sentences is removed, as the model has much harder time learning useful regularities and merely learns to copy input data.
R ELATED WORK
A similar work to ours is the style transfer method with non-parallel text by Shen et al.
The authors consider a sequence-to-sequence model, where the latent state given to the decoder is also fed to a discriminator.
The encoder is trained with the decoder to reconstruct the input, but also to fool the discriminator.
The authors also found it beneﬁcial to train two discriminators, one for the source and one for the target domain.
Then, they trained the decoder so that the recurrent hidden states during the decoding process of a sentence in a particular domain are not distinguishable according to the respective discriminator.
This algorithm, called Professor forcing, was initially introduced by Lamb et al.
(2016) to encourage the dynamics of the decoder observed during inference to be similar to the ones observed at training time.
Similarly, Xie et al.
(2017) also propose to use an adversarial training approach to learn representations invariant to speciﬁc attributes.
In particular, they train an encoder to map the observed data to a latent feature space, and a model to make predictions based on the encoder output.
To remove bias existing in the data from the latent codes, a discriminator is also trained on the encoder outputs to predict speciﬁc attributes, while the encoder is jointly trained to fool the discriminator.
They show that the obtained invariant representations lead to better generalization on classiﬁcation and generation tasks.
Before that, Hu et al.
(2017) trained a variational autoencoder (Kingma & Welling, 2013) where the decoder input is the concatenation of an unstructured latent vector, and a structured code representing the attribute of the sentence to generate.
A discriminator is trained on top of the decoder to classify the labels of generated sentences, while the decoder is trained to satisfy this discriminator.
Because of the non-differentiability of the decoding process, at each step, their decoder takes as input the probability vector predicted at the previous step.
Perhaps, the most relevant prior work is by He et al.
(2016), who essentially optimizes directly for the model selection metric we propose in section 3.2.
One drawback of their approach, which has not been applied to the fully unsupervised setting, is that it requires to back-propagate through the sequence of discrete predictions using reinforcement learning-based approaches which are notoriously inefﬁcient.
In this work, we instead propose to a) use a symmetric architecture, and b) freeze the translator from source to target when training the translator from target to source, and vice versa.
By alternating this process we operate with a fully differentiable model and we efﬁciently converge.
In the vision domain, several studies tackle the unsupervised image translation problem, where the task consists in mapping two image domains A and B, without paired supervision.
For instance, in the CoGAN architecture (Liu & Tuzel, 2016), two generators are trained to learn a common representation space between two domains, by sharing some of their convolutional layers.
This is similar to our strategy of sharing the LSTM weights across the source and target encoders and decoders.
Liu et al.
(2017) propose a similar approach, based on variational autoencoders, and generative adversarial networks (Goodfellow et al., 2014).
Taigman et al.
(2016) use similar approaches for emoji generation, and apply a regularization term to the generator so that it behaves like an identity mapping when provided with input images from the target domain.
Zhu et al.
(2017) introduced a cycle consistency loss, to capture the intuition that if an image is mapped from A to B, then from B to A, then the resulting image should be identical to the input one.
Our approach is also reminiscent of the Fader Networks architecture (Lample et al., 2017), where a discriminator is used to remove the information related to speciﬁc attributes from the latent states of an autoencoder of images.
The attribute values are then given as input to the decoder.
The decoder is trained with real attributes, but at inference, it can be fed with any attribute values to generate variations of the input images.
The model presented in this paper can be seen as an extension to the text domain of the Fader Networks, where the attribute is the language itself.
C ONCLUSION
We presented a new approach to neural machine translation where a translation model is learned using monolingual datasets only, without any alignment between sentences or documents.
The principle of our approach is to start from a simple unsupervised word-by-word translation model, and to iteratively improve this model based on a reconstruction loss, and using a discriminator to align latent distributions of both the source and the target languages.
Our experiments demonstrate that our approach is able to learn effective translation models without any supervision of any sort