MART-GRPO 用 VLM 截断失败反馈,改进 VLA 强化学习
本条为 AI 依据下方公开信源自动整理生成的摘要,不构成转载,可能存在偏差,请以原文为准。整理者:贝果科技 AI 资讯助手。
全文梳理
AI 摘要依据下方信源原文自动整理,非原文转载问题出在优势分配
论文针对的是视觉-语言-动作(VLA)模型在强化学习中的信用分配问题。在组相对策略优化(GRPO)里,一条 rollout 只得到一个轨迹级优势,这个值被统一贴到该轨迹的每个动作上。作者指出这种做法的副作用:一次失败的 rollout 并不代表每个动作都导致失败,那些试图纠错的恢复性动作会和真正的错误一起被惩罚。这会让模型在失败后放弃本可能有效的补救行为。
用鞅确定截断点
作者用成功概率与策略得分的鞅来刻画失败反馈的理想停止条件。在这些停止点上,被移除的那部分梯度贡献在给定保留信息后的条件期望为零,因此截断不会改变参考 GRPO 组梯度估计的期望,还可能降低其协方差。其中一类停止点出现在条件成功概率为零时,这为选取候选反馈边界提供了依据,也引出用错误与恢复线索来划定区间。
VLM 定位错误区间
具体实现 MART-GRPO 时,作者引入一个视觉语言模型做事件引导的截断。该 VLM 用低秩适配(LoRA)在自建标注训练集上微调,推理时保持冻结,在完整 rollout 视频中定位最后一个「可见错误且其后未见恢复」的位置。失败反馈只保留到选定的错误区间为止,区间之后的部分被移除;成功 rollout 的损失项和原有损失归一化方式都不改动。对于回溯式边界,作者还分析了期望与方差的变化,用来衡量偏离理想停止的程度。
实验与篇幅
实验在 RoboTwin 2.0 上使用 OpenVLA-OFT 进行,结果显示 MART-GRPO 在多个任务上的成功率高于全轨迹 GRPO。论文共 31 页、9 张图,摘要未给出具体任务数量、成功率数值或提升幅度,也未说明训练与推理的算力开销,这些需要查看正文才能确认。
为什么值得看
把「哪些动作该被惩罚」变成可定位的截断问题,给出保持梯度期望的理论条件,对做 VLA 强化学习的人有直接参考价值。
信源1 家
关键事实
相关 · 论文
本页内容由 AI 自动聚合公开信源生成,仅供了解行业动态参考,不构成任何投资或决策建议。如需引用请以原文出处为准。
