GSDrive: Reinforcing Driving Policies by Multi-mode Future Trajectory Probing with 3D Gaussian Splatting Environment
On the map: End-to-end autonomous driving → Imitation-learning end-to-end planners
Abstract (arXiv)
End-to-end (E2E) autonomous driving aims to directly map sensory observations to driving actions, but its real-world deployment is hindered by evolving data distributions and the high cost of continual annotation. While combining imitation learning (IL) and reinforcement learning (RL) is a common strategy for policy improvement, conventional RL training relies on delayed, event-based rewards, where policies learn only from catastrophic outcomes such as collisions, leading to premature convergence to suboptimal behaviors. To address these limitations, we propose GSDrive, a framework that uses a differentiable 3D Gaussian Splatting (3DGS) environment for future-aware trajectory probing and reward shaping in E2E driving. GSDrive first learns a multi-mode trajectory probe via IL and then uses RL to evaluate multiple candidate futures in the 3DGS environment, converting their simulated returns into dense shaping rewards for policy optimization. This yields a cyclic hybrid IL-RL training loop, where IL supplies structured future priors and RL provides interactive feedback for iterative refinement. Evaluated on the reconstructed nuScenes dataset, our method outperforms other simulation-based RL approaches in closed-loop experiments. Code is available at https://github.com/ZionGo6/GSDrive.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Imitation-learning end-to-end planners, one of 9 directions in End-to-end autonomous driving. Builds on 33 of its 34 references that are in the map. First posted on arXiv 30 Apr 2026.
Builds on
33 of 34 references.
Closest ancestors · 5
Same direction, newest first
- 2025TakeAD: Preference-based Post-optimization for End-to-end Autonomous Driving with Expert Takeover Data19 Dec
- 2025Zero-Human Demonstration End-to-end Autonomous Driving with Trajectory Scorer28 Oct
- 2025DriveDPO: Policy Learning via Safety DPO For End-to-End Autonomous Driving22 Sep
- 2025ReconDreamer-RL: Enhancing Reinforcement Learning via Diffusion-based Scene Reconstruction11 Aug
- 2025RAD: Training an End-to-End Driving Policy via Large-Scale 3DGS-based Reinforcement Learning18 Feb
Other references · 28
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2020Denoising Diffusion Implicit Modelscited by 9,655
- 2022Classifier-Free Diffusion Guidancecited by 5,717
- 2022Flow Matching for Generative Modelingcited by 5,426
- 2025VGGT: Visual Geometry Grounded Transformercited by 1,475
- 2020Lift, Splat, Shoot: Encoding Images From Arbitrary Camera Rigs by Implicitly Unprojecting to 3Dcited by 1,040
- 2023End-to-end Autonomous Driving: Challenges and Frontierscited by 485
- 2024SparseDrive: End-to-End Autonomous Driving via Sparse Scene Representationcited by 239
- 2025ReCogDrive: A Reinforced Cognitive Framework for End-to-End Autonomous Drivingcited by 190
- 2025End-to-End Driving with Online Trajectory Evaluation via BEV World Modelcited by 145
16 more
- 2025Reinforcement Learning with Action Chunkingcited by 97
- 2024Optimal Flow Matching: Learning Straight Trajectories in Just One Stepcited by 74
- 2025ARTEMIS: Autoregressive End-to-End Trajectory Planning with Mixture of Experts for Autonomous Drivingcited by 66
- 2020On Unbalanced Optimal Transport: An Analysis of Sinkhorn Algorithmcited by 61
- 2025Centaur: Robust End-to-End Autonomous Driving with Test-Time Trainingcited by 41
- 2025Data Scaling Laws for End-to-End Autonomous Drivingcited by 14
- 2025UncAD: Towards Safe End-to-end Autonomous Driving via Online Map Uncertaintycited by 14
- 2025DistillDrive: End-to-End Multi-Mode Autonomous Driving Distillation by Isomorphic Hetero-Source Planning Modelcited by 13
- 2025Reinforcement Learning with Inverse Rewards for World Model Post-trainingcited by 12
- 2025GaussianUpdate: Continual 3D Gaussian Splatting Update for Changing Environmentscited by 11
- 2025FutureX: Enhance End-to-End Autonomous Driving via Latent Chain-of-Thought World Modelcited by 7
- 2023Skill-Critic: Refining Learned Skills for Hierarchical Reinforcement Learningcited by 6
- 2025Iterative Label Refinement Matters More than Preference Optimization under Weak Supervisioncited by 5
- 2025Drive&Gen: Co-Evaluating End-to-End Driving and Video Generation Modelscited by 4
- 2025SynAD: Enhancing Real-World End-to-End Autonomous Driving Models through Synthetic Data Integrationcited by 3
- 2026The Era of End-to-End Autonomy: Transitioning from Rule-Based Driving to Large Driving Modelscited by 3
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025DiffusionDriveV2: Reinforcement Learning-Constrained Truncated Diffusion Modeling in End-to-End Autonomous Driving
- 2026Fine-tuning is Not Enough: A Parallel Framework for Collaborative Imitation and Reinforcement Learning in End-to-end Autonomous Driving
- 2025Pseudo-Expert Regularized Offline RL for End-to-End Autonomous Driving in Photorealistic Closed-Loop Environments
- 2025DIVER: Reinforced Diffusion Breaks Imitation Bottlenecks in End-to-End Autonomous Driving
- 2026RoXDrive: Closed-Loop Reinforcement Learning for End-to-End Autonomous Driving via Action-Faithful Rollouts
- 2024From Imitation to Exploration: End-to-end Autonomous Driving based on World Model
- 2026ForeDrive: Foresight-Guided End-to-End Autonomous Driving with a Planning-Relevant Latent World Model
- 2026TerraTransfer: Learning End-to-End Driving Policies Without Expert Demonstrations