重思权重平均模型合并,揭示其有效性的内在机制。
Rethinking Weight-Averaged Model-merging
- 从权重结构分析出发,解释为何平均能提升性能。
- 对比权重空间与特征空间平均,发现适用场景差异。
- 发现权重平均具有正则化效果,增强预测稳定性。
模型合并,尤其是通过权重平均,已证明能在无需额外训练的情况下节省计算量并提升模型性能。然而,该技术的可解释性仍不清晰。本文从可解释性视角重新审视权重平均模型合并,提供实证洞察。从三个角度展开:(1) 分析学习到的权重结构,表明权重编码了结构化表征,有助于解释权重平均的兼容性;(2) 在多种模型架构(CNNs 和 ViTs)与数据集上,对比权重空间与特征空间平均的效果,揭示不同场景下哪种组合方式更优;(3) 研究参数缩放对预测稳定性的影响,说明权重平均作为一种正则化手段,有助于提升鲁棒性。通过将这些分析置于可解释性框架中,本工作为利益相关方提供了对未经训练模型组合方法安全性和可靠性的更透明、系统化理解。代码已公开于 https://github.com/billhhh/Rethink-Merge。
原文摘要 · Abstract (English)
Model merging, particularly through weight averaging, has shown surprising effectiveness in saving computations and improving model performance without any additional training. However, the interpretability of this technique works remains unclear. In this work, we reinterpret weight-averaged model merging through the lens of interpretability and provide empirical insights. We approach the problem from three perspectives: (1) we analyze the learned weight structures and demonstrate that model weights encode structured representations that help explain the compatibility of weight averaging; (2) we compare averaging in weight space and feature space across diverse model architectures (CNNs and ViTs) and datasets, aiming to expose under which circumstances what combination paradigm will work more effectively; (3) we study the effect of parameter scaling on prediction stability, highlighting how weight averaging acts as a form of regularization that contributes to robustness. By framing these analyses in an interpretability context, our work contributes to a more transparent and systematic understanding of model merging for stakeholders interested in the safety and reliability of untrained model combination methods. The code is available at https://github.com/billhhh/Rethink-Merge.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。