Language Exposure Matters: Distribution and Order in Multilingual BabyLM Pre-training
Organizations: School of Computation, Information and Technology, TU Munich · Munich Center for Machine Learning · Munich Data Science Institute
Abstract
We investigate two aspects of multilingual pre-training with limited data, namely how language exposure is distributed across English, Dutch, and Chinese, and how that exposure is ordered during training. In particular, we compare balanced multilingual exposure to language-dominant exposure, where one language is exposed with more training data than the others. We also compare mixed training, where multilingual training data is used throughout training, with curriculum training, where monolingual exposure of the dominant language is followed by balanced multilingual exposure. Our results show that multilingual BabyLM performance varies with both the amount and order of language exposure. Mixed training outperforms curriculum learning across all 15 matched configurations, indicating that simultaneous multilingual exposure is more effective than staged exposure in our setting. Several language-dominant mixed training configurations achieve higher scores than the balanced configuration, suggesting that a balanced distribution is not necessarily the highest-performing configuration in our setting. Interestingly, the Dutch-primary configurations achieve the best overall performance.
Figures & tables
| Configuration | Primary | Other 1 | Other 2 |
|---|---|---|---|
| Balanced | 33% | 33% | 33% |
| 50/25/25 | 50% | 25% | 25% |
| 60/20/20 | 60% | 20% | 20% |
| 70/15/15 | 70% | 15% | 15% |
| 80/10/10 | 80% | 10% | 10% |
| 90/5/5 | 90% | 5% | 5% |
Appendix figures & tables2 assets
Supplementary material from the paper’s appendix.
Appendix
| Setting | Value |
|---|---|
| Model size | 110M parameters |
| Tokenizer | Byte-level BPE |
| Vocabulary size | 32,000 |
| Context length | 1024 |
| Optimizer | AdamW |
| Learning rate |
| Model | English | Dutch | Chinese | Multi | |||||||
|---|---|---|---|---|---|---|---|---|---|---|---|
| Config | Strategy | ZS | FT | Avg | ZS | FT | Avg | ZS | FT | Avg | Avg |
| Balanced baseline | |||||||||||
| 33/33/33 | Mixed | 55.48 | 45.52 | 47.06 | 54.99 | 42.70 | 46.61 | 48.73 | 43.78 | 45.03 | 46.23 |
| English-primary | |||||||||||
| 50/25/25 | Mixed | 56.50 | 45.94 | 48.15 | 54.77 | 41.80 | 45.89 | 48.96 | 43.25 | 44.50 | 46.18 |
| Curr | 54.39 | 42.79 | 45.91 | 53.05 | 43.47 | 46.39 | 48.13 | 44.97 | 45.87 | 46.06 | |