解决LoRA模型合并时参数差异导致性能下降的问题。
Decouple and Orthogonalize: A Data-Free Framework for LoRA Merging

- 将参数拆分为大小和方向两部分,分别独立合并。
- 在不依赖数据的情况下,实现跨任务更高性能的合并效果。
- 适合需要高效整合多个LoRA模型的研究者使用。
随着开源模型日益丰富,模型合并成为降低训练、存储与推理成本的重要手段。现有方法多针对全量微调模型设计,忽视了广泛使用的LoRA。我们实证发现:(a) 针对全量微调的合并方法在LoRA上表现差;(b) LoRA模块参数幅度方差远大于全微调权重;(c) 幅度方差越大,合并性能越差。由于大方差导致合并参数分布偏移,引发信息丢失与性能下降。为此,我们提出解耦正交合并(DO-Merging):将参数分解为幅度与方向,分步合并,减少幅度差异对方向对齐的影响,保留任务信息。同时引入无数据、逐层带正交约束的梯度下降法,缓解方向合并中的干扰。提供解耦与正交成分的理论保障。大量实验验证,在视觉、语言与多模态领域,该方法显著优于现有方法,且开销极小。各组件可灵活接入现有方法,实现近乎免费的性能提升。
原文摘要 · Abstract (English)
With more open-source models available for diverse tasks, model merging has gained attention by combining models into one, reducing training, storage, and inference costs. Current research mainly focuses on model merging for full fine-tuning, overlooking the popular LoRA. However, our empirical analysis reveals that: a) existing merging methods designed for full fine-tuning perform poorly on LoRA; b) LoRA modules show much larger parameter magnitude variance than full fine-tuned weights; c) greater parameter magnitude variance correlates with worse merging performance. Considering that large magnitude variances cause deviations in the distribution of the merged parameters, resulting in information loss and performance degradation, we propose a Decoupled and Orthogonal merging approach(DO-Merging). By separating parameters into magnitude and direction components and merging them independently, we reduce the impact of magnitude differences on the directional alignment of the merged models, thereby preserving task information. Furthermore, we introduce a data-free, layer-wise gradient descent method with orthogonal constraints to mitigate interference during the merging of direction components. We provide theoretical guarantees for both the decoupling and orthogonal components. And we validate through extensive experiments across vision, language, and multi-modal domains that our proposed DO-Merging can achieve significantly higher performance than existing merging methods at a minimal cost. Notably, each component can be flexibly integrated with existing methods, offering near free-lunch improvements across tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。