Search papers, labs, and topics across Lattice.
This paper introduces dVLA-RL, a novel approach that integrates Reinforcement Learning (RL) into Discrete Diffusion Vision-Language-Action (dVLA) models by reformulating the learning objective to focus on the joint probability of sampled generation paths rather than the intractable marginal action probability. By modeling the denoising process as a Markov Decision Process (MDP), the authors create a trajectory-level objective that allows for flexible denoising step scheduling tailored to task complexity. The method achieves an impressive 99.7% success rate on the LIBERO benchmark and a 30.6% performance improvement over supervised fine-tuning on RoboTwin 2.0, demonstrating its efficacy in complex multi-task learning scenarios.
Shifting the focus from marginal probabilities to joint trajectory probabilities, dVLA-RL achieves unprecedented success rates in robotic manipulation tasks.
Vision-Language-Action (VLA) models have established a powerful paradigm for generalist robotic manipulation by grounding control into the semantic reasoning of VLMs. Prevailing architectures typically model actions continuously via diffusion or flow processes, or discretely through either autoregressive generation or parallel decoding. Recently, Discrete Diffusion VLAs (dVLAs) have emerged as a distinct alternative, unifying vision, language, and action into a single discrete token space via masked generative modeling. While combining iterative refinement with unified representations, its training has thus far been restricted to Supervised Fine-Tuning (SFT), leaving the potential of Reinforcement Learning (RL) for further policy refinement largely unexplored. A fundamental challenge in RL for dVLAs is that the marginal probability of the final action generated by dVLAs remains intractable. To solve this problem, we propose \textbf{dVLA-RL}, shifting the learning objective from the marginal action probability to the joint probability of the sampled generation path. Specifically, by modeling the denoising process as a Markov Decision Process (MDP), we mathematically formulate this path probability as a product of step-wise transitions. This trajectory-level objective provides a unified formulation that natively accommodates variable denoising steps. Leveraging this intrinsic fexibility, we introduce a unified step scheduling approach for complex multi-task learning, tailoring denoising steps to specific task complexities to maximize both success rates and computational effciency. Extensive evaluations demonstrate that our approach achieves a success rate of \textbf{99.7\%} on LIBERO. Furthermore, it establishes strong VLA-based results on RoboTwin 2.0 by delivering a \textbf{30.6\%} improvement over the SFT baseline, remaining competitive with strong World-Action Model baselines.
Tsinghua AI