arXiv:2505.23117cs.LGcs.AI2025-05被引 2

通过空间对齐提升模型融合效果,实现高效多任务学习

Decom-Renorm-Merge: Model Merging on the Right Space Improves Multitasking

  • 用奇异值分解将权重矩阵映射到统一空间,实现精准融合
  • 在ViT、DeBERTa、T5和Llama3.1-8B上均超越现有方法
  • 重归一化是构建稳定融合空间的关键,适合多模型集成场景

大规模训练时代,模型融合已成为高效构建多任务模型的工具,可避免传统多任务学习所需的高计算开销。现有方法通常假设权重矩阵相同位置的元素具有相同功能,从而进行直接逐项比较与融合。然而,微调后的神经网络中神经元可能形成不同的特征组合,导致直接逐项融合存在问题。本文提出Decom-Renorm-Merge(DRM)方法,利用奇异值分解将权重矩阵分解并映射至对齐的联合空间,使逐项融合成为可能。我们在多种架构上验证了DRM的有效性,包括小型编码器类模型如ViT和DeBERTa,编码器-解码器类模型如T5,以及大型解码器类模型如Llama3.1-8B。实验结果表明,无论全量微调还是低秩适应设置下,DRM均优于多个先进融合技术。此外,分析显示重归一化是构建鲁棒联合空间的核心组件,显著提升融合性能。

原文摘要 · Abstract (English)

In the era of large-scale training, model merging has evolved into a tool for creating multitasking models efficiently. It enables the knowledge of models to be fused, without the need for heavy computation as required in traditional multitask learning. Existing merging methods often assume that entries at identical positions in weight matrices serve the same function, enabling straightforward entry-wise comparison and merging. However, this assumption overlooks the complexity of finetuned neural networks, where neurons may develop distinct feature compositions, making direct entry-wise merging problematic. We present Decom-Renorm-Merge (DRM), a simple yet effective approach that leverages Singular Value Decomposition to decompose and coordinate weight matrices into an aligned joint space, where entry-wise merging becomes possible. We showcase the effectiveness of DRM across various settings ranging from smaller encoder-based such as ViT and DeBERTa, encoder-decoder-based such as T5, and larger decoder-based such as Llama3.1-8B. Our experimental results show that DRM outperforms several state-of-the-art merging techniques across full finetuning and low-rank adaptation settings. Moreover, our analysis reveals renormalization as the crucial component for creating a robust and even joint space for merging, significantly contributing to the method's performance.

模型融合多任务学习权重对齐Llama3

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