在费舍尔-罗曼流形上融合大模型,提升多专家合并的稳定性与性能
Functionality-Oriented LLM Merging on the Fisher--Rao Manifold
- 基于费舍尔-罗曼流形计算加权卡契均值,从功能角度合并模型
- 支持多模型融合且不降级,在异构模型下仍保持高准确率
- 适用于需要合并多个微调模型的实用场景,如智能系统集成
权重空间合并旨在不重新训练的情况下将多个微调的大语言模型合并为单一模型,但现有方法大多仍是参数空间的启发式操作。这带来三个实际限制:第一,线性平均、任务向量等方法在欧几里得坐标中进行,而目标应是合并预测行为;第二,当源检查点距离较远或异质性高时,欧几里得混合常引发表征坍塌,表现为激活方差收缩和有效秩下降,导致精度显著下降;第三,许多几何启发方法仅自然适用于两模型插值,难以推广到多于两个专家的有原则合并。本文提出在费舍尔-罗曼流形上计算加权卡契均值,该方法局部等价于最小化预测分布间的KL散度。我们推导出一种轻量级球面代理的固定点算法,保持范数并可直接扩展至多专家合并。在多个基准测试和坍塌诊断中,本方法在模型数量和异质性增加时仍保持稳定,持续优于先前基线。
原文摘要 · Abstract (English)
Weight-space merging aims to combine multiple fine-tuned LLMs into a single model without retraining, yet most existing approaches remain fundamentally parameter-space heuristics. This creates three practical limitations. First, linear averaging, task vectors, and related rules operate on Euclidean coordinates, even though the desired goal is to merge functionality, i.e., predictive behaviors across tasks. Second, when the source checkpoints are farther apart or more heterogeneous, Euclidean blends often trigger representation collapse, manifested as activation variance shrinkage and effective-rank degradation, which sharply degrades accuracy. Third, many geometry-inspired methods are most natural for two-model interpolation and do not extend cleanly to merging N>2 experts with a principled objective. We address these issues by formulating model merging as computing a weighted Karcher mean on the Fisher--Rao manifold, which is locally equivalent to minimizing a KL-based function distance between predictive distributions. We derive a practical fixed-point algorithm using a lightweight spherical proxy that preserves norms and generalizes directly to multi-expert merging. Across various benchmarks and collapse diagnostics, our method remains stable as the number and heterogeneity of merged models increase, consistently outperforming prior baselines.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。