arXiv:2412.03187cs.CL2024-12中稿 · ICLR被引 18

无需对齐词汇表,用奖励优化实现大模型隐式融合

Weighted-Reward Preference Optimization for Implicit Model Fusion

  • 通过偏好优化在源模型与目标模型间传递能力
  • 在AlpacaEval-2上击败GPT-4-Preview,胜率55.9%
  • 适合想低成本融合多个大模型的开发者使用

将不同架构和规模的开源大模型进行融合,可整合各模型优势,但现有方法面临词汇对齐和分布矩阵合并等难题,过程复杂且易引入噪声。本文提出一种隐式融合方法——加权奖励偏好优化(WRPO),通过源模型与目标模型间的偏好优化实现能力迁移,无需词汇对齐和矩阵融合,可高效扩展至多种大模型。为缓解源与目标模型间的分布偏差,WRPO采用渐进适应策略,逐步减少对目标模型偏好样本的依赖,转而增强对源模型的依赖。在MT-Bench、AlpacaEval-2和Arena-Hard等多个基准测试中,WRPO持续优于现有知识融合方法及各类微调基线。以LLaMA3-8B-Instruct为目标模型时,其在AlpacaEval-2上对GPT-4-Preview-1106的长度控制胜率为55.9%,在Arena-Hard上对GPT-4-0314的胜率为46.2%。代码已开源。

原文摘要 · Abstract (English)

While fusing heterogeneous open-source LLMs with varying architectures and sizes can potentially integrate the strengths of different models, existing fusion methods face significant challenges, such as vocabulary alignment and merging distribution matrices. These procedures are not only complex but also prone to introducing noise and errors. In this paper, we propose an implicit fusion method, Weighted-Reward Preference Optimization (WRPO), which leverages preference optimization between the source LLMs and the target LLM to transfer their capabilities effectively. WRPO eliminates the need for vocabulary alignment and matrix fusion and can be efficiently scaled to accommodate various LLMs. To address distributional deviations between the source and target LLMs, WRPO introduces a progressive adaptation strategy that gradually shifts reliance on preferred examples from the target LLM to the source LLMs. Extensive experiments on the MT-Bench, AlpacaEval-2, and Arena-Hard benchmarks demonstrate that WRPO consistently outperforms existing knowledge fusion methods and various fine-tuning baselines. When applied to LLaMA3-8B-Instruct as the target model, WRPO achieves a length-controlled win rate of 55.9% against GPT-4-Preview-1106 on AlpacaEval-2 and a win rate of 46.2% against GPT-4-0314 on Arena-Hard. Our code is available at https://github.com/SLIT-AI/WRPO.

模型融合偏好优化大模型

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