$M^3$ Scaling Law: Optimizing Multi-Epoch, Multi-Lingual, and Multi-Stage Training for Low-Resource Language Models
About
In this paper, we study a fundamental design problem in pretraining Large Language Models (LLMs) for low-resource language regimes. Existing works adopt multi-epoch, multi-lingual, and multi-stage training to utilize the limited target-language corpus efficiently, but no prior scaling law can compare recipes spanning these approaches under the same compute budget $C$ and target-language corpus size $D_T$, leaving the optimal training setup unclear. To address this gap, we propose the $M^3$ Scaling Law, a unified predictive model parameterized by the model scale, the number of target-corpus epochs $k$, the average target-language ratio $r$, and the final-stage target-language ratio $r_f$, which places monolingual single-stage, multi-lingual single-stage, and multi-lingual multi-stage recipes on a single target-language loss surface. Across three language pairs, it extrapolates to unseen hyperparameter regions more accurately than existing scaling laws. Using $M^3$ as a surrogate objective, we derive two practical guidelines for low-resource LLM pretraining: (i) as $D_T$ decreases, the optimal recipe shifts directly from monolingual single-stage to multi-lingual two-stage training at a compute-budget-dependent threshold, with multi-lingual single-stage never optimal in our experimental grid; and (ii) the optimal number of epochs collapses onto a single curve in the scarcity variable $D_T/D^*(C)$, where $D^*(C) \propto C^{\alpha/(\alpha+\beta)}$ is the monolingual compute-optimal corpus size.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Scaling-law extrapolation | 2-stage multi-lingual (test) | R^2 (C)0.84 | 10 | |
| Extrapolation Accuracy Prediction | Japanese, Indonesian, and Swahili All 1-stage + 2-stage data (test) | R^2 (C)0.79 | 8 | |
| Scaling-law extrapolation | Japanese, Indonesian, and Swahili 1-stage data only | C (R^2)0.79 | 6 | |
| Scaling-law extrapolation | Japanese, Indonesian, and Swahili Monolingual Multi-epoch 1-stage (test) | R^2 (C)0.88 | 5 | |
| Scaling-law extrapolation | Japanese, Indonesian, and Swahili Multi-lingual single-epoch both stages (test) | C Score0.9 | 5 | |
| Scaling Law Extrapolation Accuracy | Japanese-Indonesian-Swahili 1-stage (test) | R² (C)0.95 | 5 |