arXiv:2410.15287cs.CL2024-10被引 16

用多智能体反馈训练模型,让大模型更会自我批评。

Training Language Models to Critique With Multi-agent Feedback

  • 用多个模型协同生成高质量批评意见,避免单个模型的偏误。
  • 在两个基准测试中,70亿参数模型表现接近700亿参数模型。
  • 适合想提升模型反思能力的研究者与开发者参考。

批评能力是人类元认知的重要体现,但对大语言模型而言仍具挑战。现有方法主要依赖单一模型(如GPT-4)生成的批评进行监督微调(SFT),但此类批评常因复杂性而存在缺陷,导致微调后模型性能受限且缺陷被继承。为此,本文提出新型数据生成流程MultiCritique,通过在SFT和强化学习(RL)阶段引入多智能体反馈,提升模型批评能力。该流程从多个智能体聚合高质量批评,并以关键信息为输入简化内容;同时通过多智能体反馈提升批评质量偏好准确率,增强RL效果。基于此,我们构建了用于SFT和RL微调的MultiCritiqueDataset。在两个基准测试上的实验表明:1)所构建的SFT数据集质量优于现有批评数据集;2)强化学习阶段进一步提升了模型的批评能力。值得注意的是,我们微调的70亿参数模型显著超越其他70亿至130亿参数开源模型,逼近先进700亿参数模型及GPT-4水平。代码、数据集与模型权重将公开可用。

原文摘要 · Abstract (English)

Critique ability, a meta-cognitive capability of humans, presents significant challenges for LLMs to improve. Recent works primarily rely on supervised fine-tuning (SFT) using critiques generated by a single LLM like GPT-4. However, these model-generated critiques often exhibit flaws due to the inherent complexity of the critique. Consequently, fine-tuning LLMs on such flawed critiques typically limits the model's performance and propagates these flaws into the learned model. To overcome these challenges, this paper proposes a novel data generation pipeline, named MultiCritique, that improves the critique ability of LLMs by utilizing multi-agent feedback in both the SFT and reinforcement learning (RL) stages. First, our data generation pipeline aggregates high-quality critiques from multiple agents instead of a single model, with crucial information as input for simplifying the critique. Furthermore, our pipeline improves the preference accuracy of critique quality through multi-agent feedback, facilitating the effectiveness of RL in improving the critique ability of LLMs. Based on our proposed MultiCritique data generation pipeline, we construct the MultiCritiqueDataset for the SFT and RL fine-tuning stages. Extensive experimental results on two benchmarks demonstrate: 1) the superior quality of our constructed SFT dataset compared to existing critique datasets; 2) additional improvements to the critique ability of LLMs brought by the RL stage. Notably, our fine-tuned 7B model significantly surpasses other advanced 7B-13B open-source models, approaching the performance of advanced 70B LLMs and GPT-4. Codes, datasets and model weights will be publicly available.

大模型批评能力多智能体强化学习

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