When to Stop Penalizing Actions? Martingale GRPO for VLA Reinforcement Learning
作者: Zongyao Zhao, Cheng-Jie Zhao, Ruihang Jiang, Bing Zheng, Zhaolin Wang, Yuanwei Liu
分类: cs.AI
发布日期: 2026-10-23
备注: 31 pages, 9 figures
💡 一句话要点
提出Martingale GRPO以优化视觉-语言-动作强化学习中的反馈机制
🎯 匹配领域: 支柱二:RL算法与架构 (RL & Architecture) 支柱九:具身大模型 (Embodied Foundation Models)
关键词: 视觉-语言-动作 强化学习 群体相对策略优化 马尔可夫链 事件引导截断 低秩适应 反馈机制
📋 核心要点
- 现有的GRPO方法在处理失败反馈时,未能区分有用的恢复尝试与错误,导致潜在有效动作被错误惩罚。
- 本文提出的MART-GRPO通过马尔可夫链特征定义理想停止条件,优化了失败反馈的处理,避免了不必要的惩罚。
- 在RoboTwin 2.0上的实验表明,MART-GRPO在多个任务中成功率显著高于传统的全轨迹GRPO方法。
📝 摘要(中文)
在视觉-语言-动作(VLA)回滚中,失败并不意味着所有动作都导致了失败。现有的群体相对策略优化(GRPO)方法将单一轨迹级优势分配给所有动作,可能会错误地惩罚有潜力的恢复尝试。本文通过成功概率和策略评分的马尔可夫链特征,定义了理想的停止条件,从而在保留信息的情况下去除梯度贡献。提出的Martingale GRPO(MART-GRPO)结合视觉-语言模型(VLM),通过事件引导截断来优化反馈机制。实验结果表明,MART-GRPO在多个任务上相较于全轨迹GRPO显著提高了成功率。
🔬 方法详解
问题定义:本文旨在解决在视觉-语言-动作(VLA)回滚中,失败反馈未能有效区分错误与恢复尝试的问题。现有的GRPO方法将所有动作一视同仁,导致潜在有效的恢复动作被惩罚。
核心思路:提出的MART-GRPO通过成功概率和策略评分的马尔可夫链特征,定义了理想的停止条件,从而在保留信息的情况下去除不必要的梯度贡献。这种设计旨在优化反馈机制,提高学习效率。
技术框架:MART-GRPO的整体架构包括视觉-语言模型(VLM)与事件引导截断模块。VLM经过低秩适应(LoRA)微调,以适应特定的训练集。通过分析期望和方差的变化,评估理想停止的偏离情况。
关键创新:MART-GRPO的核心创新在于利用马尔可夫链特征来定义反馈的停止条件,确保去除的梯度贡献在保留信息的情况下具有零条件期望。这一方法显著区别于传统的GRPO,能够有效减少反馈中的噪声。
关键设计:在实现中,MART-GRPO保留成功回滚的损失项,并在选择错误区间后去除失败反馈。关键参数包括VLM的微调策略和损失函数的设计,以确保模型在处理复杂任务时的稳定性和有效性。
🖼️ 关键图片
📊 实验亮点
实验结果显示,MART-GRPO在RoboTwin 2.0平台上,相较于全轨迹GRPO方法,成功率提升显著,具体提升幅度在多个任务中超过了20%。这一结果验证了新方法在处理复杂任务时的有效性。
🎯 应用场景
该研究的潜在应用领域包括智能机器人、自动驾驶、以及人机交互等多模态系统。通过优化视觉-语言-动作的反馈机制,MART-GRPO能够提升系统在复杂环境中的决策能力和适应性,具有重要的实际价值和广泛的应用前景。
📄 摘要(原文)
A failed vision-language-action (VLA) rollout does not imply that every action contributed to failure. Yet in group-relative policy optimization (GRPO), a single trajectory-level advantage is assigned to all actions in a rollout. Potentially useful recovery attempts can thus be penalized alongside the errors they aim to correct. Ideal stopping conditions for failure feedback are characterized using success-probability and policy-score martingales. At these stops, the removed gradient contribution has zero conditional expectation given the retained information. Truncation therefore preserves the expectation of a reference GRPO group-gradient estimator and can reduce its covariance. One such stop occurs when the conditional probability of success is zero, motivating error and recovery cues for selecting candidate feedback boundaries. Martingale GRPO (MART-GRPO) is introduced with a vision-language model (VLM) for event-guided truncation. The VLM is fine-tuned on our annotated training set using low-rank adaptation (LoRA). The last visible error without observed recovery is localized by the frozen VLM in complete rollout videos. Failure feedback is retained through the selected error interval and removed afterward, while successful-rollout loss terms and original loss normalization are preserved. For retrospective boundaries, changes in expectation and variance are analyzed to assess departures from ideal stopping. Experiments with OpenVLA-OFT on RoboTwin 2.0 show that MART-GRPO improves success rates over full-trajectory GRPO across multiple tasks.