arXiv:2602.05943cs.LG2026-02被引 2

用几何方法合并大模型,避免遗忘旧知识。

Orthogonal Model Merging

  • 在正交群流形上操作,保持权重的几何结构。
  • 相比传统方法,减少灾难性遗忘,多任务表现更稳。
  • 适合需要融合多个微调模型的场景,如多能力集成。

微调后的大型语言模型(LLMs)融合日益重要,用于将多种能力整合到单一模型中。然而,现有方法依赖欧氏空间中的线性运算,常破坏预训练权重的内在几何特性(如超球面能量)。为此,我们提出正交模型融合(OrthoMerge),在由正交群构成的黎曼流形上执行融合操作,以保留权重的几何结构。通过将正交微调(OFT)学习的任务特定正交矩阵映射至李代数,OrthoMerge 实现了兼顾方向与强度的合理高效融合。除直接使用OFT获得的正交矩阵外,我们进一步将该方法扩展至非OFT类微调模型(如低秩微调、全量微调),采用正交残差解耦策略:通过求解正交普罗克鲁斯特斯问题提取专家模型的正交成分,在正交群流形上融合,其余线性残差则通过标准加法方式处理。大量实验证明,OrthoMerge 能有效缓解灾难性遗忘,维持多样任务下的模型性能。

原文摘要 · Abstract (English)

Merging finetuned Large Language Models (LLMs) has become increasingly important for integrating diverse capabilities into a single unified model. However, prevailing model merging methods rely on linear arithmetic in Euclidean space, which often destroys the intrinsic geometric properties of pretrained weights, such as hyperspherical energy. To address this, we propose Orthogonal Model Merging (OrthoMerge), a method that performs merging operations on the Riemannian manifold formed by the orthogonal group to preserve the geometric structure of the model's weights. By mapping task-specific orthogonal matrices learned by Orthogonal Finetuning (OFT) to the Lie algebra, OrthoMerge enables a principled yet efficient integration that takes into account both the direction and intensity of adaptations. In addition to directly leveraging orthogonal matrices obtained by OFT, we further extend this approach to general models finetuned with non-OFT methods (i.e., low-rank finetuning, full finetuning) via an Orthogonal-Residual Decoupling strategy. This technique extracts the orthogonal components of expert models by solving the orthogonal Procrustes problem, which are then merged on the manifold of the orthogonal group, while the remaining linear residuals are processed through standard additive merging. Extensive empirical results demonstrate the effectiveness of OrthoMerge in mitigating catastrophic forgetting and maintaining model performance across diverse tasks.

模型融合几何方法大模型

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