Search papers, labs, and topics across Lattice.
This paper introduces OmniVLA-RL, a Vision-Language-Action model that uses a Mix-of-Transformers architecture to improve spatial perception and multimodal fusion. To address instability in reinforcement learning, they propose Flow-GSPO, which combines flow matching as a Stochastic Differential Equation with Group Segmented Policy Optimization. Experiments on LIBERO and LIBERO-Plus show OmniVLA-RL outperforms existing VLA models, demonstrating improved action precision and training robustness.
VLA models can now achieve significantly higher performance and stability in embodied AI tasks by combining a Mixture-of-Transformers architecture with a novel flow-matching-based reinforcement learning algorithm.
Visual-Language-Action (VLA) models represent a paradigm shift in embodied AI, yet existing frameworks often struggle with imprecise spatial perception, suboptimal multimodal fusion, and instability in reinforcement learning. To bridge these gaps, we propose OmniVLA-RL, a novel architecture that leverages a Mix-of-Transformers (MoT) design to synergistically integrate reasoning, spatial, and action experts. Furthermore, we introduce Flow-GSPO, which reformulates flow matching as a Stochastic Differential Equation (SDE) process and integrates it with Group Segmented Policy Optimization (GSPO) to enhance action precision and training robustness. Extensive evaluations on the LIBERO and LIBERO-Plus benchmarks demonstrate that OmniVLA-RL significantly outperforms state-of-the-art methods, effectively overcoming the fundamental limitations of current VLA models.
