跳到主要内容
AI News HubLIVE
站内改写2 分钟阅读

AIPO:通过主动交互学习推理

文章摘要

AIPO是一个增强的强化学习框架,通过引入验证、知识和推理三个协作智能体,在训练过程中主动引导策略模型突破推理能力边界,从而提升大语言模型的推理性能。实验表明该方法在多个基准测试中一致优于现有方法。

来源arXiv Computational Linguistics作者: Junnan Liu, Linhao Luo, Thuy-Trang Vu, Gholamreza Haffari
AIPO:通过主动交互学习推理
报告错误

纠错通道尚未开通,可先复制下方文章信息留存。

查看更正说明
直接读正文

近年来,大语言模型(LLM)在推理能力上取得了显著进展,这一进展很大程度上得益于基于可验证奖励的强化学习(RLVR)。然而,现有的强化学习算法存在一个根本性局限:其探索范围受限于策略模型固有的能力边界。虽然已有方法尝试引入外部专家演示来扩展这一边界,但它们通常依赖于完整的轨迹级指导,导致样本效率低下、信息稀疏,并可能将探索限制在静态的指导空间中。

受多智能体系统潜力的启发,研究人员提出了AIPO框架,这是一种增强的强化学习框架,通过在探索过程中进行主动的多智能体交互来提升LLM推理能力。具体而言,AIPO使策略模型在遇到推理瓶颈时能够主动咨询三个功能协作智能体:验证智能体(Verify Agent)、知识智能体(Knowledge Agent)和推理智能体(Reasoning Agent),从而获得细粒度、有针对性的指导,在训练过程中主动扩展能力边界。验证智能体负责检查中间步骤的正确性,知识智能体提供相关事实和背景知识,推理智能体则引导更有效的推理路径。这种协作使得策略模型能够超越自身能力限制,学习更复杂的推理策略。

为了克服从智能体反馈学习时出现的离策略偏差和梯度消失问题,AIPO引入了一个定制的重要性采样系数以及裁剪策略。重要性采样系数调整不同反馈样本的权重,减少偏差;裁剪策略则限制更新幅度,防止梯度消失。训练完成后,策略模型可以独立进行推理,无需依赖协作智能体,因此部署时不会增加额外开销。

在包括AIME、MATH500、GPQA-Diamond和LiveCodeBench在内的多种推理基准上的大量实验表明,AIPO consistently提升了推理性能,在不同策略模型和RLVR算法上具有鲁棒泛化能力,并有效扩展了策略模型的推理能力边界。例如,在AIME(美国数学邀请赛)题目上,AIPO显著提高了准确率;在MATH500上,它超越了现有最佳方法。此外,该方法在代码生成基准LiveCodeBench上也表现优异,显示了其跨领域的泛化能力。这些结果证明,通过主动多智能体交互,可以显著突破LLM的推理能力上限。该方法的代码和数据已公开发布,便于后续研究和应用。

展开要点与分析

文章情报

工程师进阶

要点

  • AIPO框架允许LLM在探索过程中主动咨询三个专业智能体以突破能力边界。
  • 引入重要性采样系数和裁剪策略解决从智能体反馈学习时的偏差和梯度消失问题。
  • 训练后策略模型可独立推理,无需依赖协作智能体。
  • 在AIME、MATH500等基准上显著提升推理性能,且泛化性强。

要点与分析由自动化流程生成,可能有误,请结合原始来源核实。