104

πRLπ_\texttt{RL}: Online RL Fine-tuning for Flow-based Vision-Language-Action Models

Kang Chen
Zhihao Liu
Tonghe Zhang
Zhen Guo
Si Xu
Hao Lin
Hongzhi Zang
Quanlu Zhang
Zhaofei Yu
Guoliang Fan
Tiejun Huang
Yu Wang
Chao Yu
Main:19 Pages
14 Figures
Bibliography:3 Pages
7 Tables
Appendix:2 Pages
Abstract

Vision-Language-Action (VLA) models enable robots to understand and perform complex tasks from multimodal input. Although recent work explores using reinforcement learning (RL) to automate the laborious data collection process in scaling supervised fine-tuning (SFT), applying large-scale RL to flow-based VLAs (e.g., π0\pi_0, π0.5\pi_{0.5}) remains challenging due to intractable action log-likelihoods from iterative denoising.We address this challenge with πRL\pi_{\text{RL}}, an open-source framework for training flow-based VLAs in parallel simulation. πRL\pi_{\text{RL}} implements two RL algorithms: (1) {Flow-Noise} models the denoising process as a discrete-time MDP with a learnable noise network for exact log-likelihood computation. (2) {Flow-SDE} integrates denoising with agent-environment interaction, formulating a two-layer MDP that employs ODE-to-SDE conversion for efficient RL exploration.We evaluate πRL\pi_{\text{RL}} on LIBERO and ManiSkill benchmarks. On LIBERO, πRL\pi_{\text{RL}} boosts few-shot SFT models π0\pi_0 and π0.5\pi_{0.5} from 57.6% to 97.6% and from 77.1% to 98.3%, respectively. In ManiSkill, we train πRL\pi_{\text{RL}} in 320 parallel environments, improving π0\pi_0 from 41.6% to 85.7% and π0.5\pi_{0.5} from 40.0% to 84.8% across 4352 pick-and-place tasks, demonstrating scalable multitask RL under heterogeneous simulation.Overall, πRL\pi_{\text{RL}} achieves significant performance gains and stronger generalization over SFT-models, validating the effectiveness of online RL for flow-based VLAs.

View on arXiv
Comments on this paper