用轨迹级相对优化提升视觉语言动作模型的机器人任务泛化能力
TGRPO :Fine-tuning Vision-Language-Action Model via Trajectory-wise Group Relative Policy Optimization
- 通过大语言模型生成任务分析,自动构建密集奖励函数
- 并行采样多条轨迹进行相对比较,降低强化学习方差
- 无需价值网络,兼顾全局与局部优化信号,适合复杂长程任务
视觉-语言-动作(VLA)模型在大规模预训练和任务微调后,在多种机器人任务中展现出强大的跨场景泛化能力。然而,其训练主要依赖人工收集的成功示范,难以适应分布外(OOD)场景或执行偏差。强化学习虽具闭环优化能力,但面临奖励稀疏、方差高、优化不稳定等问题。为此,本文提出基于轨迹的组相对策略优化(TGRPO),一种面向VLA模型的在线强化学习训练框架。TGRPO利用大语言模型生成的任务分析,自动构造密集奖励函数,提供细粒度反馈以加速收敛并改进信用分配。核心是分组策略:并行采样与归一化多条轨迹,通过相对比较降低方差。结合轨迹级与步骤级优势估计,无需价值网络即可捕捉全局与局部优化信号。在LIBERO基准的四类任务上实验表明,TGRPO平均成功率达80.7%,较监督微调(SFT)高出4.2个百分点,并优于其他代表性强化学习后训练方法。
原文摘要 · Abstract (English)
Visual-Language-Action (VLA) models have demonstrated strong cross-scenario generalization capabilities in various robotic tasks through large-scale pre-training and task-specific fine-tuning. However, their training paradigm mainly relies on manually collected successful demonstrations, making it difficult to adapt to complex environments when encountering out-of-distribution (OOD) scenarios or execution biases. While Reinforcement Learning (RL) provides a closed-loop optimization framework via active trial-and-error mechanism, it suffers from sparse rewards, high variance, and unstable optimization in long-horizon robotic tasks. To address these limitations, we propose Trajectory-based Group Relative Policy Optimization (TGRPO), an online RL-based training framework for VLA models. TGRPO leverages task analysis generated by a large language model to automatically construct dense reward functions, providing fine-grained feedback to accelerate convergence and improve credit assignment. The core of our method is a group-based strategy that samples and normalizes multiple trajectories in parallel, reducing variance through relative comparison. By integrating trajectory-level and step-level advantage estimation, TGRPO captures both global and local optimization signals without relying on a value network. Experiments on four task categories of the LIBERO benchmark demonstrate that TGRPO achieves an average success rate of 80.7\%, which is 4.2\% higher than that of Supervised Fine-Tuning (SFT) and outperforms other representative RL-based post-training methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。