让大模型在推理时同时学会看懂图像,提升多模态任务表现。
Perception-Aware Policy Optimization for Multimodal Reasoning
- 在强化学习中加入隐式感知损失,让模型边推理边学看图。
- 多模态基准上性能提升4.4%至17.5%,视觉依赖任务提升达8.0%-19.1%。
- 无需额外数据或更强教师模型,可无缝接入主流强化学习算法。
基于可验证奖励的强化学习(RLVR)已被证明是赋予大语言模型稳健多步推理能力的有效方法。然而,其设计和优化仍局限于纯文本领域,在多模态推理任务中表现不佳。我们发现当前多模态推理的主要错误来源在于对视觉输入的感知不足。为此,提出PAPO,一种新型策略梯度算法,使模型在推理过程中同时学习感知能力。具体地,引入基于KL散度的隐式感知损失,可无缝集成至GRPO、DAPO等主流RLVR算法中。值得注意的是,PAPO无需额外数据标注、奖励模型或更强教师模型。为增强训练稳定性,引入双熵损失,有效正则化新KL目标而不牺牲性能。尽管结构简单,PAPO在多种多模态基准上实现4.4%–17.5%的整体提升,视觉依赖任务提升达8.0%–19.1%。感知错误减少30.5%,表明模型感知能力显著增强。本工作将感知意识深度融入核心学习目标,为构建视觉具身推理的新强化学习框架奠定基础。代码与数据将公开。项目页:https://mikewangwzhl.github.io/PAPO。
原文摘要 · Abstract (English)
Reinforcement Learning with Verifiable Rewards (RLVR) has proven to be a highly effective strategy for endowing Large Language Models (LLMs) with robust multi-step reasoning abilities. However, its design and optimizations remain tailored to purely textual domains, resulting in suboptimal performance when applied to multimodal reasoning tasks. In particular, we observe that a major source of error in current multimodal reasoning lies in the perception of visual inputs. To address this bottleneck, we propose PAPO, a novel policy gradient algorithm that encourages the model to learn to perceive while learning to reason. Specifically, we introduce the Implicit Perception Loss in the form of a KL divergence term, which can be seamlessly plugged into mainstream RLVR algorithms such as GRPO and DAPO. Notably, PAPO does not rely on additional data curation, reward models, or stronger teacher models. To further enhance the training stability of PAPO, we introduce the Double Entropy Loss, which effectively regularizes the new KL objective without compromising performance. Despite its simplicity, PAPO yields significant overall improvements of 4.4%-17.5% on diverse multimodal benchmarks. The improvements are more pronounced, approaching 8.0%-19.1%, on tasks with high vision dependency. We also observe a substantial reduction of 30.5% in perception errors, indicating improved perceptual capabilities with PAPO. Overall, our work introduces a deeper integration of perception-aware supervision into core learning objectives and lays the groundwork for a new RL framework that encourages visually grounded reasoning. Code and data will be made publicly available for research purposes. Project page: https://mikewangwzhl.github.io/PAPO.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。