Search papers, labs, and topics across Lattice.
The paper introduces DriveFine, a masked diffusion Vision-Language-Action (VLA) model for autonomous driving that aims to address the limitations of both diffusion-based and token-based planners. DriveFine uses a novel plug-and-play block-MoE to inject a refinement expert on top of a generation expert, enabling flexible decoding and self-correction. Experiments on NAVSIM benchmarks demonstrate that DriveFine achieves strong efficacy and robustness compared to existing approaches.
By decoupling generation and refinement experts within a masked diffusion VLA model, DriveFine achieves both flexible decoding and self-correction for autonomous driving.
Vision-Language-Action (VLA) models for autonomous driving increasingly adopt generative planners trained with imitation learning followed by reinforcement learning. Diffusion-based planners suffer from modality alignment difficulties, low training efficiency, and limited generalization. Token-based planners are plagued by cumulative causal errors and irreversible decoding. In summary, the two dominant paradigms exhibit complementary strengths and weaknesses. In this paper, we propose DriveFine, a masked diffusion VLA model that combines flexible decoding with self-correction capabilities. In particular, we design a novel plug-and-play block-MoE, which seamlessly injects a refinement expert on top of the generation expert. By enabling explicit expert selection during inference and gradient blocking during training, the two experts are fully decoupled, preserving the foundational capabilities and generic patterns of the pretrained weights, which highlights the flexibility and extensibility of the block-MoE design. Furthermore, we design a hybrid reinforcement learning strategy that encourages effective exploration of refinement expert while maintaining training stability. Extensive experiments on NAVSIM v1, v2, and Navhard benchmarks demonstrate that DriveFine exhibits strong efficacy and robustness. The code will be released at https://github.com/MSunDYY/DriveFine.
