arXiv:2502.18821cs.LG2025-02ICLR被引 8

用自然梯度融合专家,让大模型训练更贴合参数空间曲率。

CAMEx: Curvature-aware Merging of Experts

  • 基于自然梯度的非欧几何融合,贴合参数空间真实曲率。
  • 在多个NLP任务上超越传统方法,提升预训练与微调性能。
  • 无需额外计算资源,适合高效扩展大模型训练。

现有专家融合方法多依赖欧几里得几何,假设参数空间平坦,限制了模型泛化能力,尤其在预训练阶段。曲率感知方法通常需额外信息和计算资源来近似费舍尔信息矩阵,带来显著内存开销。本文提出CAMEx(Curvature-Aware Merging of Experts),引入自然梯度以捕捉参数流形的非欧几何特性。通过利用自然梯度,CAMEx能更有效地适应参数空间结构,使模型更新与流形几何对齐,从而提升预训练与微调效果,且无需显著内存开销。贡献有三:(1) CAMEx在多种自然语言处理任务中显著优于传统欧氏融合方法,改善预训练与微调性能;(2) 提出动态融合架构,优化资源利用,实现高性能的同时降低计算成本,支持大语言模型高效扩展;(3) 提供理论与实证证据验证方法效率。代码已开源:https://github.com/kpup1710/CAMEx。

原文摘要 · Abstract (English)

Existing methods for merging experts during model training and fine-tuning predominantly rely on Euclidean geometry, which assumes a flat parameter space. This assumption can limit the model's generalization ability, especially during the pre-training phase, where the parameter manifold might exhibit more complex curvature. Curvature-aware merging methods typically require additional information and computational resources to approximate the Fisher Information Matrix, adding memory overhead. In this paper, we introduce CAMEx (Curvature-Aware Merging of Experts), a novel expert merging protocol that incorporates natural gradients to account for the non-Euclidean curvature of the parameter manifold. By leveraging natural gradients, CAMEx adapts more effectively to the structure of the parameter space, improving alignment between model updates and the manifold's geometry. This approach enhances both pre-training and fine-tuning, resulting in better optimization trajectories and improved generalization without the substantial memory overhead typically associated with curvature-aware methods. Our contributions are threefold: (1) CAMEx significantly outperforms traditional Euclidean-based expert merging techniques across various natural language processing tasks, leading to enhanced performance during pre-training and fine-tuning; (2) we introduce a dynamic merging architecture that optimizes resource utilization, achieving high performance while reducing computational costs, facilitating efficient scaling of large language models; and (3) we provide both theoretical and empirical evidence to demonstrate the efficiency of our proposed method. The code is publicly available at: https://github.com/kpup1710/CAMEx.

专家融合自然梯度大模型训练非欧几何

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