arXiv:2510.04686cs.LGcs.AI2025-10被引 2

优化器噪声尺度决定模型融合成败,可调参数提升融合效果

How does the optimizer implicitly bias the model merging loss landscape?

  • 用有效噪声尺度统一解释优化器对融合的影响
  • 噪声尺度适中时融合成功率最高,呈非单调关系
  • 学习率、权重衰减等均可调节该尺度,适合模型融合研究者

模型融合将多个独立训练的模型合并为单一模型,保持推理成本不变。常用方法包括线性插值(平均权重)和任务算术(结合微调与基础模型的差值向量)。尽管实用,但其有效性机制仍不清晰。本文探讨优化动态如何影响损失景观几何结构及其对融合成功的影响。我们发现,一个单一量——有效噪声尺度——可统一解释不同优化器组件对模型融合的影响。在多种架构和数据集上,融合成功率是有效噪声尺度的非单调函数,存在明确最优值。分解该量可知,较大的学习率、更强的权重衰减、较小的批量大小和数据增强均独立调节有效噪声尺度,并呈现相同定性趋势。不同于以往将优化器噪声与单个极小值的平坦性或泛化能力关联的研究,我们证明其也影响全局损失景观,从而预测独立训练解能否成功融合。研究揭示了优化如何塑造损失景观并影响模型融合,提示可通过操控训练动态进一步优化融合效果。

原文摘要 · Abstract (English)

Model merging combines independent solutions with different capabilities into a single one while maintaining the same inference cost. Two popular approaches are linear interpolation, which simply averages multiple model weights, and task arithmetic, which combines task vectors obtained by the difference between finetuned and base models. While useful in practice, what properties make merging effective are poorly understood. This paper explores how the optimization dynamics affect the loss landscape geometry and its impact on merging success. We show that a single quantity -- the effective noise scale -- unifies the impact of different optimizer components on model merging. Across architectures and datasets, merging success is a non-monotonic function of the effective noise scale, with a distinct optimum. Decomposing this quantity, we find that larger learning rates, stronger weight decay, smaller batch sizes, and data augmentation all independently modulate the effective noise scale and exhibit the same qualitative trend. Unlike prior work connecting optimizer noise to the flatness or generalization of individual minima, we show that it also affects the global loss landscape, predicting when independently trained solutions can be successfully merged. Our findings broaden the understanding of how optimization shapes the loss landscape geometry and its consequences for model merging, suggesting that training dynamics could be further manipulated to improve model merging.

模型融合优化器损失景观训练动态

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。