arXiv:2508.03121cs.LG2025-08中稿 · Transactions on Ma…被引 10

改进模型合并的回归均值法,提升通用性和效果

RegMean++: Enhancing Effectiveness and Generalization of Regression Mean for Model Merging

  • 引入层内与跨层依赖关系优化合并目标
  • 在多种场景下均优于原方法,尤其在分布外泛化表现更佳
  • 简单有效,适合需要稳定合并性能的研究者

回归均值(RegMean)将模型合并建模为线性回归问题,通过最小化合并模型与候选模型间预测差异,求解各线性层的最优权重,具备解析解和计算高效性。然而,该方法独立合并每一层,忽略了早期层特征在深层传播中对最终输出的影响。为此,本文提出RegMean++,在原目标中显式引入层内与跨层依赖关系,更好捕捉合并模型的行为。大量实验表明,RegMean++在多种场景下持续优于RegMean,包括域内(ID)、域外(OOD)泛化、序列合并、大规模任务及多种分布偏移下的鲁棒性。此外,其性能在多个先进模型合并方法中也具有竞争力。

原文摘要 · Abstract (English)

Regression Mean (RegMean), an approach that formulates model merging as a linear regression problem, aims to find the optimal weights for each linear layer in the merged model by minimizing the discrepancy in predictions between the merged and candidate models. RegMean provides a precise closed-form solution for the merging problem; therefore, it offers explainability and computational efficiency. However, RegMean merges each linear layer independently, overlooking how the features and information in earlier layers propagate through deeper layers and influence the final predictions of the merged model. Here, we introduce RegMean++, a simple yet effective alternative to RegMean, that explicitly incorporates both intra-layer and cross-layer dependencies between merged models' layers into RegMean's objective. By accounting for these dependencies, RegMean++ better captures the behaviors of the merged model. Extensive experiments demonstrate that RegMean++ consistently outperforms RegMean across diverse settings, including in-domain (ID) and out-of-domain (OOD) generalization, sequential merging, large-scale tasks, and robustness under several types of distribution shifts. Furthermore, RegMean++ achieves competitive performance across diverse settings compared to various advanced model merging methods.

模型合并回归均值泛化能力

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