用高资源语言数据混合,比调参更能提升低资源语言模型性能。
Mix, Don't Tune: Bilingual Pre-Training Outperforms Hyperparameter Search in Data-Constrained Settings

- 用英语数据混合训练阿拉伯语模型,提升效果优于调参。
- 混合可使下游任务准确率提升2到13倍,模型越大越明显。
- 适合低资源语言训练,尤其对大模型有效。
世界上多数语言的预训练处于数据受限状态,模型需重复训练数据多次,导致泛化能力下降。现有两种解决方案:激进的超参数调优(如高权重衰减)和引入高资源辅助语言数据进行混合训练。前者通过缩小权重限制网络容量,后者通过可调节的混合比例扩展训练分布,引入新知识。我们在150M至1.43B参数的四个模型规模上,以阿拉伯语为低资源目标、英语为辅助语言,进行了约1000次预训练实验。结果表明:混合带来的性能提升显著高于超参数调优,且随模型增大而扩大;混合相当于为验证损失带来2–3倍目标数据量的增益,下游任务准确率提升2–13倍,且收益随模型规模陡增;目标语言验证损失系统性低估了混合的价值——混合通过多样化训练信号实现正则化,并提供重复目标数据无法提供的知识,而验证损失仅捕捉前者效应。建议:优先使用高资源语言数据混合,关注混合比例而非调参,并通过μP从小型代理模型迁移超参数。
原文摘要 · Abstract (English)
For most languages of the world, language model pre-training operates in a data-constrained regime where models must repeat their training data many times, degrading generalization. Two remedies exist: aggressive hyperparameter tuning such as high weight decay, and mixing in data from a high-resource auxiliary language to directly aid the low-resource target. While hyperparameter tuning regularizes the model by shrinking weights to restrict network capacity, auxiliary data mixing uses a tunable mixing ratio to expand the training distribution and diversify the training signal with new knowledge. Both offer a principled way to improve training in a data-constrained domain. We compare these levers systematically across four model scales from 150M to 1.43B parameters, using Arabic as the low-resource target and English as the auxiliary, over approximately 1000 pre-training runs. Three findings emerge. First, mixing yields larger improvements than hyperparameter tuning on both validation loss and downstream task accuracy, and the gap grows with model size. Second, we quantify how much mixing helps: it boosts performance by an amount equivalent to 2--3$\times$ the unique target data on validation loss and 2--13$\times$ on downstream task accuracy, with the gain scaling steeply with model size. Third, this divergence reveals that target-language validation loss systematically underestimates mixing's value. Mixing regularizes by diversifying the training signal and contributes knowledge the repeated target corpus cannot supply; validation loss captures only the first effect. Our practical recommendations are: mix in a high-resource language, prioritize the mixing ratio over hyperparameter tuning, and transfer hyperparameters from a small proxy model via $μ$P.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。