arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2502.04411cs.LGcs.AIcs.CL
收藏

Mediator:基于较少参数冲突和不确定性路由的内存高效大语言模型合并

Mediator: Memory-efficient LLM Merging with Less Parameter Conflicts and Uncertainty Based Routing

The Hong Kong University of Science and Technology (Guangzhou)(香港科技大学(广州));The Hong Kong University of Science and Technology(香港科技大学);Platform and Content Group, Tencent(腾讯平台与内容事业群);UCAS(中国科学院大学);Sun Yat-sen University(中山大学)

Kunfeng Lai, Zhenheng Tang, Xinglin Pan, Peijie Dong, Xiang Liu, Haolan Chen, Huacan Wang, Li Shen, Bo Li, Xiaowen Chu

AI总结:

研究针对大语言模型合并中参数冲突致性能下降等问题,提出基于层参数冲突情况分别处理,结合任务算术稀疏性解耦专家,依输入数据任务不确定性选合并专家,实验表明该方法提升性能且降低系统成本。

摘要:

模型合并将在不同任务上微调的大语言模型聚合为一个更强的模型。然而,模型之间的参数冲突会导致平均性能下降。虽然模型路由通过在推理时选择单个模型解决了这个问题,但它带来了过高的存储和计算成本,并且无法利用不同模型的常识。在这项工作中,我们观察到不同层表现出不同程度的参数冲突。基于这一见解,我们对参数冲突最小的层进行平均,并对冲突显著的层使用新颖的任务级专家路由。为了进一步降低存储成本,受任务算术稀疏性的启发,我们将多个微调专家解耦为一个密集专家和几个稀疏专家。考虑到分布外样本,我们根据输入数据的任务不确定性选择并合并合适的专家。我们在不同参数规模的LLaMA和Qwen上进行了广泛的实验,并在实际推理任务上进行评估。结果表明,我们的方法在需要更少系统成本的同时始终实现显著的性能提升。

英文摘要:

Model merging aggregates Large Language Models (LLMs) finetuned on different tasks into a stronger one. However, parameter conflicts between models leads to performance degradation in averaging. While model routing addresses this issue by selecting individual models during inference, it imposes excessive storage and compute costs, and fails to leverage the common knowledge from different models. In this work, we observe that different layers exhibit varying levels of parameter conflicts. Building on this insight, we average layers with minimal parameter conflicts and use a novel task-level expert routing for layers with significant conflicts. To further reduce storage costs, inspired by task arithmetic sparsity, we decouple multiple fine-tuned experts into a dense expert and several sparse experts. Considering the out-of-distribution samples, we select and merge appropriate experts based on the task uncertainty of the input data. We conduct extensive experiments on both LLaMA and Qwen with varying parameter scales, and evaluate on real-world reasoning tasks. Results demonstrate that our method consistently achieves significant performance improvements while requiring less system cost.

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

↑