arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2502.06876cs.CLcs.AIcs.LG

混合数据还是融合模型?通过模型融合平衡大型语言模型的有用性、诚实性和无害性

Mix Data or Merge Models? Balancing the Helpfulness, Honesty, and Harmlessness of Large Language Model via Model Merging

Zhejiang University(浙江大学);Ant Group(蚂蚁集团)

Jinluan Yang, Dingnan Jin, Anke Tang, Li Shen, Didi Zhu, Zhengyu Chen, Ziyu Zhao, Daixin Wang, Qing Cui, Zhiqiang Zhang, Jun Zhou, Fei Wu, Kun Kuang

AI总结:

本文提出RESM方法,通过改进的参数融合策略提升大型语言模型在有用性、诚实性和无害性方面的平衡对齐效果。

摘要:

实现大型语言模型(LLMs)在有用性、诚实性和无害性(3H优化)方面的平衡对负责任的AI至关重要。现有方法如数据混合策略面临限制,包括对专家知识的依赖和优化信号的冲突。虽然模型融合通过整合专门模型的参数提供参数层面的冲突解决策略,但其在3H优化中的潜力仍被忽视。本文首次系统比较了模型融合和数据混合方法在构建3H对齐的LLMs中的有效性,揭示了3H维度之间之前被忽视的协作和冲突关系,并讨论了数据混合(数据层面)和模型融合(参数层面)方法在缓解冲突以实现平衡3H优化中的优缺点。特别地,我们提出了一种新的Reweighting Enhanced Task Singular Merging方法(RESM),通过异常值加权和稀疏性感知的排名选择策略来解决3H对齐LLM融合中固有的偏好噪声累积和层稀疏性适应挑战。广泛的评估可以验证RESM与先前的数据混合(2%-5%提升)和模型融合(1%-3%提升)方法在实现平衡LLM对齐方面的有效性和鲁棒性。我们通过https://huggingface.co/Jinluan发布我们的模型以供进一步研究。

英文摘要:

Achieving balanced alignment of large language models (LLMs) in terms of Helpfulness, Honesty, and Harmlessness (3H optimization) constitutes a cornerstone of responsible AI. Existing methods like data mixture strategies face limitations, including heavy reliance on expert knowledge and conflicting optimization signals. While model merging offers parameter-level conflict-resolution strategies through integrating specialized models' parameters, its potential for 3H optimization remains underexplored. This paper systematically compares the effectiveness of model merging and data mixture methods in constructing 3H-aligned LLMs for the first time, revealing previously overlooked collaborative and conflict relationships among the 3H dimensions and discussing the advantages and drawbacks of data mixture (\textit{data-level}) and model merging (\textit{parameter-level}) methods in mitigating the conflict for balanced 3H optimization. Specially, we propose a novel \textbf{R}eweighting \textbf{E}nhanced task \textbf{S}ingular \textbf{M}erging method, \textbf{RESM}, through outlier weighting and sparsity-aware rank selection strategies to address the challenges of preference noise accumulation and layer sparsity adaptation inherent in 3H-aligned LLM merging. Extensive evaluations can verify the effectiveness and robustness of RESM compared to previous data mixture (2\%-5\% gain) and model merging (1\%-3\% gain) methods in achieving balanced LLM alignment. We release our models through \href{https://huggingface.co/Jinluan}{3H\_Merging} for further investigations.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑