用视觉模型反馈指导离线强化学习,提升策略鲁棒性。
Offline RLAIF: Piloting VLM Feedback for RL via SFO
- 基于子轨迹的过滤优化,利用视觉模型判断成功画面
- 通过回溯过滤剔除失败前动作,显著减少学习干扰
- 简单行为克隆方法优于复杂强化学习调优,适合实际部署
尽管互联网规模的图文数据推动了视觉语言模型(VLM)的强大泛化能力,但缺乏互联网规模的控制数据阻碍了标准强化学习(RL) agent 的类似泛化。尽管 VLM 因缺少动作条件训练数据而无法直接解决控制任务,但其图像理解能力可用来识别成功结果并提供反馈。在从 AI 反馈中进行强化学习(RLAIF)时,如何有效整合 VLM 信号是关键挑战。本文聚焦离线强化学习,提出一类名为子轨迹过滤优化(SFO)的方法。我们发现三个关键洞察:第一,轨迹长度在离线 RL 中至关重要,全轨迹偏好学习会加剧拼接问题,需采用子轨迹;第二,即使在马尔可夫环境中,仍需依赖图像序列的非马尔可夫奖励信号来评估轨迹改进,因 VLM 无法理解控制动作,只能依赖视觉线索;第三,一种简单有效的策略——加权过滤行为克隆,持续优于更复杂的 RLHF 方法。我们提出子轨迹过滤行为克隆(SFBC),利用 VLM 对子轨迹的反馈,并引入回溯过滤机制,剔除失败前的子轨迹以增强鲁棒性,避免学习过程震荡。
原文摘要 · Abstract (English)
While internet-scale image and textual data have enabled strong generalization in Vision-Language Models (VLMs), the absence of internet-scale control data has impeded the development of similar generalization in standard reinforcement learning (RL) agents. Although VLMs are fundamentally limited in their ability to solve control tasks due to their lack of action-conditioned training data, their capacity for image understanding allows them to provide valuable feedback in RL tasks by recognizing successful outcomes. A key challenge in Reinforcement Learning from AI Feedback (RLAIF) is determining how best to integrate VLM-derived signals into the learning process. We explore this question in the context of offline RL and introduce a class of methods called Sub-Trajectory Filtered Optimization (SFO). We identify three key insights. First, trajectory length plays a crucial role in offline RL, as full-trajectory preference learning exacerbates the stitching problem, necessitating the use of sub-trajectories. Second, even in Markovian environments, a non-Markovian reward signal from a sequence of images is required to assess trajectory improvement, as VLMs do not interpret control actions and must rely on visual cues over time. Third, a simple yet effective approach--filtered and weighted behavior cloning--consistently outperforms more complex RLHF-based methods. We propose Sub-Trajectory Filtered Behavior Cloning (SFBC), a method that leverages VLM feedback on sub-trajectories while incorporating a retrospective filtering mechanism that removes sub-trajectories preceding failures to improve robustness and prevent turbulence. Please enjoy our airport puns.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。