用3D高斯点云环境预探未来轨迹,让自动驾驶模型更聪明地避障。
GSDrive: Reinforcing Driving Policies by Multi-mode Future Trajectory Probing with 3D Gaussian Splatting Environment
- 用3D高斯点云构建可微分仿真环境,预演多种未来路径
- 通过多模态未来探测提升奖励密度,使策略优化更稳定高效
- 适合想改进自动驾驶强化学习训练效率的研究者
端到端自动驾驶直接将感知输入映射为驾驶动作,但其实际部署受限于数据分布变化和持续标注成本。虽然融合模仿学习(IL)与强化学习(RL)是常见策略,但传统RL依赖延迟的事件奖励,仅从碰撞等灾难性结果中学习,导致过早收敛至次优行为。为此,我们提出GSDrive框架,利用可微分的3D高斯点云(3DGS)环境进行未来感知的轨迹探测与奖励塑造。GSDrive首先通过模仿学习(IL)建立多模态轨迹预测器,再使用强化学习(RL)在3DGS环境中评估多个候选未来路径,将模拟回报转化为密集奖励用于策略优化。该方法形成循环式混合IL-RL训练流程,其中IL提供结构化未来先验,而RL提供交互式反馈以迭代优化。在重构的nuScenes数据集上的闭环实验表明,本方法优于其他基于仿真的强化学习方案。代码已公开于https://github.com/ZionGo6/GSDrive。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。