SlowGuess/NAACL_2021_OCR huggingface dataset:
Xavier Garcia and Noah Constant and Ankur P. Parikh and Orhan First
{xgarcia,nconstant,aparikh,orhanf}  $@$  google.com
We propose a straightforward vocabulary adaptation scheme to extend the language capacity of multilingual machine translation models, paving the way towards efficient continual learning for multilingual machine translation. Our approach is suitable for large-scale datasets, applies to distant languages with unseen scripts, incurs only minor degradation on the translation performance for the original language pairs and provides competitive performance even in the case where we only possess monolingual data for the new languages.
The longstanding goal of multilingual machine translation (First et al., 2016; Johnson et al., 2017; Aharoni et al., 2019; Gu et al., 2018) has been to develop a universal translation model, capable of providing high-quality translations between any pair of languages. Due to limitations on the data available, however, current approaches rely on first selecting a set of languages for which we have data and training an initial translation model on this data jointly for all languages in a multi-task setup. In an ideal setting, one would continually update the model once data for new language pairs arrives. This setting, dubbed in the literature as continual learning (Ring, 1994; Rebuffi et al., 2017; Kirkpatrick et al., 2017; Lopez-Paz and Ranzato, 2017), introduces new challenges not found in the traditional multi-task setup, most famously catastrophic forgetting (McCloskey and Cohen, 1989), in which the model may lose its previously-learned knowledge as it learns new language pairs. This situation is further complicated by the training procedures of standard tokenizers, such as Byte-Pair Encoding (BPE) (Sennrich et al., 2015b) or Sentencepiece (Kudo and Richardson, 2018), which necessitate access to monolingual data for all the languages considered before producing the vocabulary. Failing to comply with these requirements, one risks
suboptimal segmentation rules which in the worst case could result in strings of entirely <UNK> tokens for text in a previously-unseen alphabet.
In this work, we investigate how vocabularies derived from BPE transform if they are rebuilt with the same settings but with additional data from a new language. We show in Section 3.1 that there is a large token overlap between the original and updated vocabularies. This large overlap allows us to retain the performance of a translation model after replacing its vocabulary with the updated vocabulary that additionally supports a new language.
Past works have explored adapting translation models to new languages, typically focusing on related languages which share similar scripts (Gu et al., 2018; Neubig and Hu, 2018; Lakew et al., 2019; Chronopoulou et al., 2020). These works usually focus solely on learning the new language pair, with no consideration for catastrophic forgetting. Moreover, these works only examine the setting where the new language pair comes with parallel data, despite the reality that for a variety of low-resource languages, we may only possess high-quality monolingual data with no access to parallel data. Finally, unlike our approach, these approaches do not recover the vocabulary one would have built if one had access to the data for the new language from the very beginning.
Having alleviated the vocabulary issues, we study whether we are able to learn the new language pair rapidly and accurately, matching the performance of a model which had access to this data at the beginning of training. We propose a simple adaptation scheme that allows our translation model to attain competitive performance with strong bilingual and multilingual baselines in a small amount of additional gradient steps. Moreover, our model retains most of the translation quality on the original language pairs it was trained on, exhibiting no signs of catastrophic forgetting.
Proceedings of the 2021 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies, pages 1184-1192 June 6-11, 2021. ©2021 Association for Computational Linguistics
2 Continual learning via vocabulary substitution
Related works Adapting translation models to new languages has been studied in the past. Neubig and Hu (2018) showed that a large multilingual translation model trained on a subset of languages of the TED dataset (Qi et al., 2018) could perform translation on the remaining (related) languages. Tang et al. (2020) was able to extend the multilingual translation model mBART (Liu et al., 2020) from 25 to 50 languages by exploiting the fact that mBART's vocabulary already supported those additional 25 languages. (Escolano et al., 2021) was able to add new languages to machine translation models by training language-specific encoders and decoders. Other works (Zoph et al., 2016; Lakew et al., 2018, 2019; Escolano et al., 2019) have studied repurposing translation models as initializations for bilingual models for a target low-resource language pair. Most recently (Chronopoulou et al., 2020) examined reusing language models for high-resource languages as initializations for unsupervised translation models for a related low-resource language through the following recipe: build vocabulary  $\mathcal{V}_X$  and a language model for high-resource language  $X$ ; once data for low-resource language  $Y$  arrives, build a joint vocabulary  $\mathcal{V}_{X,Y}$  and let  $\mathcal{V}_{Y|X}$  be the tokens from  $Y$  that appear in  $\mathcal{V}_{X,Y}$ ; substitute the vocabulary for the language model with the one given by  $\mathcal{V}_X \cup \mathcal{V}_{Y|X}$  and use the language model as the initialization for the translation model.
Our approach In this work, we are not only interested in the performance of our multilingual translation models on new language pairs, we also require that our models retain the performance on the multiple language pairs that they were initially trained on. We will also be interested in how the performance of these models compares with those obtained in the oracle setup where we have all the data available from the start. The approaches discussed above generate vocabularies that are likely different (both in selection and number of tokens) from the vocabulary one would obtain if one had a priori access to the missing data, due to the special attention given to the new language. This architectural divergence will only grow as we continually add new languages, which inhibits the comparisons to the oracle setup. We eliminate this mismatch by first building a vocabulary  $\mathcal{V}_N$  on the  $N$  languages
available, then once the new language arrives, build a new vocabulary  $\nu_{N + 1}$  as we would have if we had possessed the data from the beginning and replace  $\nu_{N}$  with  $\nu_{N + 1}$ . We then reuse the embeddings for tokens in the intersection<sup>1</sup> and continue training.
The success of our approach relies on the fact for large  $N$  (i.e. the multilingual setting),  $\nu_{N}$  and  $\nu_{N + 1}$  are mostly equivalent, which allows the model to retain its performance after we substitute vocabularies. We verify this in the following section.
In this section, we outline the set of experiments we conducted in this work. We first discuss the languages and data sources we use for our experiments. We then provide the training details for how we trained our initial translation models. Next, we compute the token overlap between various vocabularies derived from BPE before and after we include data for a new language and empirically verify that this overlap is large if the vocabulary already supports a large amount of languages. We then examine the amount of knowledge retained after vocabulary substitution by measuring the degradation of the translation performance on the original language pairs from replacing the original vocabulary with an updated one. Finally, we examine the speed and quality of the adaptation to new languages under various settings.
Languages considered Our initial model will have to access to data coming from 24 languages<sup>2</sup>. Our monolingual data comes primarily from the newscrawl datasets<sup>3</sup> and Wikipedia, while the parallel data comes WMT training sets and Paracrawl. We will adapt our model to the following four languages: Kazakh, which is not related linguistically to any of the original 24 languages, but does share scripts with Russian and Bulgarian; Bengali, which is related to the other Indo-Aryan languages but possesses a distinct script; Polish, which is related to (and shares scripts with) many of the Slavic languages in our original set; Pashto, which
${}^{1}$  Tokens shared between the two vocabularies are also forced to share the same indices. The remaining tokens are rewritten but we still reuse the outdated embeddings.
2In alphabetical order: Bulgarian, Czech, Danish, German, Greek, English, Spanish, Estonian, Finnish, French, Gujarati, Hindi, Croatian, Hungarian, Italian, Lithuanian, Latvian, Portuguese, Romanian, Russian, Slovak, Slovenian, Tamil.
Figure 1: The degradation in BLEU from substituting vocabularies at inference. The black dashed line represents the performance from the model trained with the modified vocabulary from the beginning, while the curves represent the BLEU scores from the original model using the new vocabulary at inference.
is not closely related<sup>4</sup> to any of the languages in our original set and has a distinct script. We provide an in-depth account of the data available for each language in the appendix.
Model configurations We perform our experiments in JAX (Bradbury et al., 2018), using the neural network library FLAX<sup>5</sup>. We use Transformers (Vaswani et al., 2017) as the basis of our translation models. We use the Transformer Big configuration and a shared BPE model of 64k tokens with byte-level fallback using the Sentencepiece<sup>6</sup> library. We used a maximum sequence length of 100, discarded all sequences longer than that during training.
Sampling scheme We train our models leveraging both monolingual and parallel datasets, following previous work (Siddhant et al., 2020; Garcia et al., 2020). We sample examples from monolingual and parallel sources with equal probability. Within each source, we use a temperature-based sampling scheme based on the numbers of samples of the relevant datasets with a temperature of 5 (Arivazhagan et al., 2019).
Training objectives We apply the MASS objective (Song et al., 2019) on the monolingual data and cross-entropy on the parallel data. We used the Adam(Kingma and Ba, 2015) optimizer, with an initial learning rate of 4e-4, coupled with a linear warmup followed by a linear decay to 0. The initial warmup took 1k steps, and the total training time was 500k steps. We also included weight decay with a hyperparameter of 0.2.
#langsin base	bn	pl	kk	ps	bn+pl+kk+ps
1	53.5%	47.0%	46.0%	47.8%	24.4%
5	84.0%	80.8%	81.8%	80.2%	57.7%
10	90.3%	87.4%	89.3%	87.2%	70.9%
15	93.1%	91.8%	90.7%	90.5%	76.9%
20	94.8%	90.1%	93.0%	93.1%	79.2%
24	95.4%	94.3%	95.2%	93.5%	82.7%
Table 1: Percentage of token overlap between vocabularies before & after the inclusion of a new language. We denote the case where we add all the unseen languages by the column ‘bn+pl+kk+ps’.
Evaluation We use beam search with a beam size of 4 and a length penalty of  $\alpha = 0.6$  for decoding. We evaluate the quality of our models using BLEU scores (Papineni et al., 2002). We exclusively use detokenized BLEU, computed through sacreBLEU (Post, 2018) for consistency with previous work and future reproducibility.
3.1 Transfer learning from vocabulary substitution
Measuring token overlap We now examine the impact on the vocabulary derived from a BPE model upon the inclusion on a new language. We first build corpora consisting of  $\text{text}^8$  from 1, 5, 10, 15, 20, and 24 of our original languages. For each corpus, we make copies and add additional data for either Bengali, Polish, Kazakh, Pashto, or their union, yielding a total of 30 corpora. We build BPE models using the same settings for each corpus, compute the token overlap between the vocabularies with and without the additional language, and
4Closest languages are in the Indic branch, but the Indic and Iranian branches split over 4000 years ago.
$^{5}$ https://github.com/google/flax
<sup>6</sup>We use 1.0 character coverage, split by whitespace, digits, and include a special token MASK for the MASS objective.
$^{7}$ BLEU + case.mixed + numrefs.1 + smooth.exp + tok.13a + version.1.4.14
We used 1 million lines of raw text per language.
Model	PMIndia bn←en	newsdev2020 pl←en	newstest2019 kk←en	FLoRes devset ps←en
Original Vocabulary	Unadapted	0.0	0.0	2.4	4.0	0.7	2.2	0.0	0.0
xx monolingual & parallel	5.7	13.6	20.2	26.2	3.9	17.2	2.8	10.3
4xx monolingual & parallel	5.3	15.1	18.3	25.0	2.7	15.8	2.3	8.4
Adapted Vocabulary	xx monolingual	0.0	1.7	13.9	24.3	0.9	19.0	0.0	6.5
xx monolingual (+BT)	-	-	21.3	24.1	4.7	19.5	-	-
xx monolingual & parallel	10.0	27.2	21.5	27.5	5.9	20.2	6.6	15.1
4xx monolingual & parallel	10.5	26.4	20.3	26.8	5.6	20.5	6.7	15.2
Oracle	xx monolingual & parallel	10.1	29.2	19.6	26.8	5.4	20.5	6.6	14.7
4xx monolingual & parallel	10.0	28.6	18.9	26.4	5.4	20.3	6.2	14.4
Table 2: BLEU scores on the new languages. The "monolingual" models have access to exclusively monolingual data for the new language(s), while "monolingual & parallel" models add parallel data as well. Models with "xx" add a single language, while "4xx" models add four languages together.
Figure 2: Measuring forgetting after adaptation. The difference in BLEU for the original language pairs between the oracle model and models adapted to Kazakh.
report the results in Table 1. In the multilingual setting, we attain large token overlap, more than  $90\%$ , even for languages with distinct scripts or when we add multiple languages at once. We extend this analysis to different vocabulary sizes and examine which tokens are "lost" in Appendix A.3.
3.2 Evaluating translation quality and catastrophic forgetting
Measuring the deterioration from swapping vocabularies at inference To measure the amount of knowledge transferred through the vocabulary substitution, we compute the translation performance of our initial translation model with the adapted vocabularies without any additional updates. For each new language, we compute the change in BLEU from the model with its original vocabulary and the one utilizing the adapted one and plot the results in Figure 1. Notably, we only incur minor degradation in performance from the vocabulary substitution.
We now study the effect of introducing a new language into our translation model. We require an adaptation recipe which enjoys the following prop
erties: fast, in terms of number of additional gradient steps; performant, in terms of BLEU scores on the new language pair; retentive, in terms of minimal regression in the translation performance of the model on the original language pairs.
Our solution: re-compute the probabilities for the temperature-based sampling scheme using the new data, upscale the probabilities of sampling new datasets by a factor then rescale the remaining probabilities so that their combined sum is one. We limit ourselves to either 15k or 30k additional steps (3% and 6% respectively of the training time for the original model) depending on the data available<sup>9</sup> to ensure fast adaptation. We reset the Adam optimizer's stored accumulators, reset the learning rate to 5e-5 and keep it fixed. We provide more details in Appendix A.2. Aside from these modifications, we continue training with the same objectives as before unless noted otherwise. We include the results for oracle models trained in the same way as the original model but with access to both the adapted
<sup>9</sup>We use 15k steps if we leverage both monolingual and parallel data for a single language pair. We use 30k steps if we only use monolingual data or if we are adapting to all four languages at once.
vocabulary and the missing data. We compute the BLEU scores and report them in Table 2.
Our models adapted with parallel data are competitive with the oracle models, even when we add all four languages at once and despite the restrictions we imposed on our adaption scheme. For languages that share scripts with the original ones (Kazakh and Polish), we can also attain strong performance leveraging monolingual data alone, albeit we need to introduce back-translation (Sennrich et al., 2015a) for optimal performance. We can also adapt the translation model using the original vocabulary, but the quality lags behind the models using the adapted vocabularies. This gap is larger for Bengali and Pashto, where the model is forced to rely on byte-level fallback, further reaffirming the value of using the adapted vocabularies.
To examine whether catastrophic forgetting has occurred, we proceed as in Section 3.1 and examine the performance on the original language pairs after adaptation on the new data against the oracle model which had access to this data in the beginning of training. We present the results for the models adapted to Kazakh in Figure 2. All the models' performance on the original language pairs deviate only slightly from the oracle model, mitigating some of the degradation from the vocabulary substitution i.e. compare the kk and bn+pl+kk+ps curves in Figure 1 to the curves in Figure 2.
Lastly, we compare our models with external baselines for Kazakh. We consider the multilingual model mBART (Liu et al., 2020) as well as all the WMT submissions that reported results on English  $\leftrightarrow$  Kazakh. Of these baselines, only mBART and (Kocmi et al., 2018) use sacreBLEU which inhibits proper comparison with the rest of the models. We include them for completeness. We report the scores in Table 3. Our adapted models are able to outperform mBART in both directions, and as well some of the weaker WMT submissions, despite those models specifically optimizing for that language pair and task.
We present an approach for adding new languages to multilingual translation models. Our approach allows for rapid adaptation to new languages with distinct scripts with only a minor degradation in performance on the original language pairs.
Model	newtest2019
kk←en
Without
en ↔ kk	xx monolingual	0.9	19.0
xx monolingual (+BT)	4.7	19.5
With
en ↔ kk	Kocmi and Bojar (2019)	8.7	18.5
Li et al. (2019)	11.1	30.5
Casas et al. (2019)	15.5	21.0
Dabre et al. (2019)	6.4	26.4
Briakou and Carpuat (2019)	-	9.94
Littell et al. (2019)	-	25.0
mBART (Liu et al., 2020)	2.5	7.4
xx monolingual & parallel	5.9	20.2
4xx monolingual & parallel	5.6	20.5
Table 3: BLEU scores on the new languages against external baselines. The models in italics are ours.
Roee Aharoni, Melvin Johnson, and Orhan First. 2019. Massively multilingual neural machine translation. In *NAACL*.
Naveen Arivazhagan, Ankur Bapna, Orhan First, Dmitry Lepikhin, Melvin Johnson, Maxim Krikun, Mia Xu Chen, Yuan Cao, George Foster, Colin Cherry, et al. 2019. Massively multilingual neural machine translation in the wild: Findings and challenges. arXiv preprint arXiv:1907.05019.
James Bradbury, Roy Frostig, Peter Hawkins, Matthew James Johnson, Chris Leary, Dougal Maclaurin, and Skye Wanderman-Milne. 2018. Jax: composable transformations of python+numpy programs.
Eleftheria Briakou and Marine Carpuat. 2019. The university of maryland's kazakh-english neural machine translation system at wmt19. In Proceedings of the Fourth Conference on Machine Translation (Volume 2: Shared Task Papers, Day 1), pages 134-140.
Noe Casas, José AR Fonollosa, Carlos Escolano, Christine Basta, and Marta R Costa-jussa. 2019. The talp-upc machine translation systems for wmt19 news translation task: pivoting techniques for low resource mt. In Proceedings of the Fourth Conference on Machine Translation (Volume 2: Shared Task Papers, Day 1), pages 155-162.
Alexandra Chronopoulou, Dario Stojanovski, and Alexander Fraser. 2020. Reusing a pretrained language model on languages with limited corpora for unsupervised nmt. arXiv preprint arXiv:2009.07610.
Raj Dabre, Kehai Chen, Benjamin Marie, Rui Wang, Atsushi Fujita, Masao Utiyama, and Eiichiro Sumita. 2019. Nict's supervised neural machine translation systems for the wmt19 news translation task. In Proceedings of the Fourth Conference on Machine Translation (Volume 2: Shared Task Papers, Day 1), pages 168-174.
Carlos Escolano, Marta R. Costa-jussà, and José A. R. Fonollosa. 2019. From bilingual to multilingual neural machine translation by incremental training. In
Proceedings of the 57th Annual Meeting of the Association for Computational Linguistics: Student Research Workshop, pages 236-242, Florence, Italy. Association for Computational Linguistics.
Carlos Escolano, Marta R Costa-Jussà, and José AR Fonollosa. 2021. From bilingual to multilingual neural-based machine translation by incremental training. Journal of the Association for Information Science and Technology, 72(2):190-203.
Orhan First, Kyunghyun Cho, and Yoshua Bengio. 2016. Multi-way, multilingual neural machine translation with a shared attention mechanism. In NAACL.
Xavier Garcia, Pierre Foret, Thibault Sellam, and Ankur P Parikh. 2020. A multilingual view of unsupervised machine translation. arXiv preprint arXiv:2002.02955.
Jiatao Gu, Hany Hassan, Jacob Devlin, and Victor OK Li. 2018. Universal neural machine translation for extremely low resource languages. arXiv preprint arXiv:1802.05368.
Melvin Johnson, Mike Schuster, Quoc V Le, Maxim Krikun, Yonghui Wu, Zhifeng Chen, Nikhil Thorat, Fernanda Viégas, Martin Wattenberg, Greg Corrado, et al. 2017. Google's multilingual neural machine translation system: Enabling zero-shot translation. Transactions of the Association for Computational Linguistics, 5:339-351.
Diederik P Kingma and Jimmy Ba. 2015. Adam: A method for stochastic optimization. In ICLR.
James Kirkpatrick, Razvan Pascanu, Neil Rabinowitz, Joel Veness, Guillaume Desjardins, Andrei A Rusu, Kieran Milan, John Quan, Tiago Ramalho, Agnieszka Grabska-Barwinska, et al. 2017. Overcoming catastrophic forgetting in neural networks. Proceedings of the national academy of sciences, 114(13):3521-3526.
Tom Kocmi and Ondrej Bojar. 2019. CUNI submission for low-resource languages in WMT news 2019. In Proceedings of the Fourth Conference on Machine Translation (Volume 2: Shared Task Papers, Day 1), pages 234-240, Florence, Italy. Association for Computational Linguistics.
Tom Kocmi, Roman Sudarikov, and Ondrej Bojar. 2018. CUNI submissions in WMT18. In Proceedings of the Third Conference on Machine Translation: Shared Task Papers, pages 431-437, Belgium, Brussels. Association for Computational Linguistics.
Taku Kudo and John Richardson. 2018. Sentencepiece: A simple and language independent subword tokenizer and tokenizer for neural text processing. In Proceedings of the 2018 Conference on Empirical Methods in Natural Language Processing: System Demonstrations, pages 66-71.