通过模块化合并提升低资源语言的跨语言迁移能力
The Unreasonable Effectiveness of Model Merging for Cross-Lingual Transfer in LLMs
- 分离数学与语言任务参数,分别训练后通过层替换合并
- 在无本地数学数据时,三种语言下性能超越基线
- 层替换合并效果最优,且后期回滚比全程冻结更有效
大语言模型在高资源语言外的任务上仍表现不佳。本文研究低资源语言中的跨语言迁移,尤其在特定任务微调数据稀缺的情况下。我们验证了数学推理与多语言能力相关参数子集基本不重叠。为利用这一隐含可分性,提出多种模块化框架,在微调中将数学与语言优化分配至模型不同部分。方法包括冻结参数或事后模型合并。在无本语言数学数据条件下,该策略在三种语言、四种模型及两种微调范式(全量与LoRA)中均优于基线。最有效的方法是分别训练语言与数学专家,再通过层替换合并。我们通过任务向量线性性理论解释此现象,并实证表明:训练后回滚无效更新,比从一开始就冻结更优。
原文摘要 · Abstract (English)
Large language models (LLMs) still struggle across tasks outside of high-resource languages. In this work, we investigate cross-lingual transfer to lower-resource languages where task-specific post-training data is scarce. Building on prior work, we first validate that the subsets of model parameters that matter most for mathematical reasoning and multilingual capabilities are distinctly non-overlapping. To exploit this implicit separability between task and target language parameterization, we develop and analyze numerous modular frameworks to improve the composition of the two during fine-tuning. These methods generally employ freezing parameters or post hoc model merging to assign math and language improvement to different key parts of the LLM. In the absence of in-language math data, we demonstrate that the modular approaches successfully improve upon baselines across three languages, four models, and two fine-tuning paradigms (full and LoRA). Furthermore, we identify the most consistently successful modular method to be fine-tuning separate language and math experts and model merging via Layer-Swapping, somewhat surprisingly. We offer possible explanations for this result via recent works on the linearity of task vectors. We further explain this by empirically showing that reverting less useful fine-tuning updates after training often outperforms freezing them from the start.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。