Model-Predictive Policy Learning with Uncertainty Regularization for Driving in Dense Traffic
Not on the map. Why?
Cited by 22 in End-to-end autonomous driving, but placed by its own abstract, which didn’t fit.
Abstract (arXiv)
Learning a policy using only observational data is challenging because the distribution of states it induces at execution time may differ from the distribution observed during training. We propose to train a policy by unrolling a learned model of the environment dynamics over multiple time steps while explicitly penalizing two costs: the original cost the policy seeks to optimize, and an uncertainty cost which represents its divergence from the states it is trained on. We measure this second cost by using the uncertainty of the dynamics model about its own predictions, using recent ideas from uncertainty estimation for deep networks. We evaluate our approach using a large-scale observational dataset of driving behavior recorded from traffic cameras, and show that we are able to learn effective driving policies from purely observational data, with no environment interaction.
Key facts. Cited by 50 papers in the map (data through 7 Oct 2026). First posted on arXiv 8 Jan 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2020: 2; 2021: 13; 2022: 10; 2023: 12; 2024: 5; 2025: 6; 2026: 2.
50 papers.
Sort:
By area:
- 2023End-to-end Autonomous Driving: Challenges and Frontierscited by 485
- 2022DayDreamer: World Models for Physical Robot Learningcited by 372
- 2023Driving into the Future: Multiview Visual Forecasting and Planning with World Model for Autonomous Drivingcited by 309
- 2021TrafficSim: Learning to Simulate Realistic Multi-Agent Behaviorscited by 192
- 2022Model-Based Imitation Learning for Urban Drivingcited by 149
- 2025End-to-End Driving with Online Trajectory Evaluation via BEV World Modelcited by 145
- 2021Urban Driver: Learning to Drive from Real-world Demonstrations Using Policy Gradientscited by 122
- 2021Reward (Mis)design for Autonomous Drivingcited by 102
- 2022Tutorial on amortized optimizationcited by 83
- 2024World Models for Autonomous Driving: An Initial Surveycited by 79
- 2021SimNet: Learning Reactive Self-driving Simulations from Real-world Observationscited by 77
- 2024DrivingWorld: Constructing World Model for Autonomous Driving via Video GPTcited by 66
38 more
- 2025A Survey of World Models for Autonomous Drivingcited by 60
- 2021A Survey of Deep RL and IL for Autonomous Driving Policy Learningcited by 59
- 2023TrafficBots: Towards World Models for Autonomous Driving Simulation and Motion Predictioncited by 56
- 2021Learning the Predictability of the Futurecited by 48
- 2021MBRL-Lib: A Modular Library for Model-based Reinforcement Learningcited by 41
- 2022Real-time Neural-MPC: Deep Learning Model Predictive Control for Quadrotors and Agile Robotic Platformscited by 36
- 2023MUVO: A Multimodal Generative World Model for Autonomous Driving with Geometric Representationscited by 33
- 2021PredictionNet: Real-Time Joint Probabilistic Traffic Prediction for Planning, Control, and Simulationcited by 29
- 2023Rethinking Closed-loop Training for Autonomous Drivingcited by 29
- 2023MoDem-V2: Visuo-Motor World Models for Real-World Robot Manipulationcited by 27
- 2023Learning Realistic Traffic Agents in Closed-loopcited by 24
- 2021UMBRELLA: Uncertainty-Aware Model-Based Offline Reinforcement Learning Leveraging Planningcited by 23
- 2021Generating and Characterizing Scenarios for Safety Testing of Autonomous Vehiclescited by 21
- 2022Plan Your Target and Learn Your Skills: Transferable State-Only Imitation Learning via Decoupled Policy Optimizationcited by 12
- 2023CCIL: Context-conditioned imitation learning for urban drivingcited by 12
- 2020Triple-GAIL: A Multi-Modal Imitation Learning Framework with Generative Adversarial Netscited by 11
- 2024Learning to Drive via Asymmetric Self-Playcited by 11
- 2022Spatiotemporal Costmap Inference for MPC via Deep Inverse Reinforcement Learningcited by 10
- 2023Explainable Action Prediction through Self-Supervision on Scene Graphscited by 9
- 2021Anti-Concentrated Confidence Bonuses for Scalable Explorationcited by 7
- 2021What classifiers know what they don't?cited by 6
- 2023Learning Interaction-aware Motion Prediction Model for Decision-making in Autonomous Drivingcited by 6
- 2020SAFARI: Safe and Active Robot Imitation Learning with Imaginationcited by 5
- 2026The Surprising Difficulty of Search in Model-Based Reinforcement Learningcited by 5
- 2021Large Scale Autonomous Driving Scenarios Clustering with Self-supervised Feature Extractioncited by 4
- 2022Tailored Uncertainty Estimation for Deep Learning Systemscited by 3
- 2023Learning to Predict Navigational Patterns from Partial Observationscited by 3
- 2023SeRO: Self-Supervised Reinforcement Learning for Recovery from Out-of-Distribution Situationscited by 2
- 2025Dynamical modeling of nonlinear latent factors in multiscale neural activity with real-time inferencecited by 2
- 2022KINet: Unsupervised Forward Models for Robotic Pushing Manipulationcited by 1
- 2022A Transferable Legged Mobile Manipulation Framework Based on Disturbance Predictive Controlcited by 1
- 2025LaMOuR: Leveraging Language Models for Out-of-Distribution Recovery in Reinforcement Learningcited by 1
- 2026Risk-Aware World Model Predictive Control for Generalizable End-to-End Autonomous Drivingcited by 1
- 2022Enhanced Behavioral Cloning with Environmental Losses for Self-Driving Vehiclescited by 0
- 2024MRIC: Model-Based Reinforcement-Imitation Learning with Mixture-of-Codebooks for Autonomous Driving Simulationcited by 0
- 2024SoftCTRL: Soft conservative KL-control of Transformer Reinforcement Learning for Autonomous Drivingcited by 0
- 2025Scalable Offline Metrics for Autonomous Drivingcited by 0
- 2025Planning as Descent: Goal-Conditioned Latent Trajectory Synthesis in Learned Energy Landscapescited by 0
- 2026Risk-Aware World Model Predictive Control for Generalizable End-to-End Autonomous Driving26 Feb 2026
- 2026The Surprising Difficulty of Search in Model-Based Reinforcement Learning29 Jan 2026
- 2025Planning as Descent: Goal-Conditioned Latent Trajectory Synthesis in Learned Energy Landscapes19 Dec 2025
- 2025Dynamical modeling of nonlinear latent factors in multiscale neural activity with real-time inference13 Dec 2025
- 2025Scalable Offline Metrics for Autonomous Driving9 Oct 2025
- 2025End-to-End Driving with Online Trajectory Evaluation via BEV World Model2 Apr 2025
- 2025LaMOuR: Leveraging Language Models for Out-of-Distribution Recovery in Reinforcement Learning21 Mar 2025
- 2025A Survey of World Models for Autonomous Driving20 Jan 2025
- 2024DrivingWorld: Constructing World Model for Autonomous Driving via Video GPT27 Dec 2024
- 2024SoftCTRL: Soft conservative KL-control of Transformer Reinforcement Learning for Autonomous Driving30 Oct 2024
- 2024Learning to Drive via Asymmetric Self-Play26 Sep 2024
- 2024MRIC: Model-Based Reinforcement-Imitation Learning with Mixture-of-Codebooks for Autonomous Driving Simulation29 Apr 2024
38 more
- 2024World Models for Autonomous Driving: An Initial Survey5 Mar 2024
- 2023Driving into the Future: Multiview Visual Forecasting and Planning with World Model for Autonomous Driving29 Nov 2023
- 2023MUVO: A Multimodal Generative World Model for Autonomous Driving with Geometric Representations20 Nov 2023
- 2023SeRO: Self-Supervised Reinforcement Learning for Recovery from Out-of-Distribution Situations7 Nov 2023
- 2023Learning Realistic Traffic Agents in Closed-loop2 Nov 2023
- 2023MoDem-V2: Visuo-Motor World Models for Real-World Robot Manipulation25 Sep 2023
- 2023End-to-end Autonomous Driving: Challenges and Frontiers29 Jun 2023
- 2023Rethinking Closed-loop Training for Autonomous Driving27 Jun 2023
- 2023CCIL: Context-conditioned imitation learning for urban driving4 May 2023
- 2023Learning to Predict Navigational Patterns from Partial Observations26 Apr 2023
- 2023TrafficBots: Towards World Models for Autonomous Driving Simulation and Motion Prediction7 Mar 2023
- 2023Learning Interaction-aware Motion Prediction Model for Decision-making in Autonomous Driving8 Feb 2023
- 2023Explainable Action Prediction through Self-Supervision on Scene Graphs7 Feb 2023
- 2022Model-Based Imitation Learning for Urban Driving14 Oct 2022
- 2022DayDreamer: World Models for Physical Robot Learning28 Jun 2022
- 2022Tailored Uncertainty Estimation for Deep Learning Systems29 Apr 2022
- 2022Real-time Neural-MPC: Deep Learning Model Predictive Control for Quadrotors and Agile Robotic Platforms15 Mar 2022
- 2022Plan Your Target and Learn Your Skills: Transferable State-Only Imitation Learning via Decoupled Policy Optimization4 Mar 2022
- 2022A Transferable Legged Mobile Manipulation Framework Based on Disturbance Predictive Control2 Mar 2022
- 2022KINet: Unsupervised Forward Models for Robotic Pushing Manipulation18 Feb 2022
- 2022Enhanced Behavioral Cloning with Environmental Losses for Self-Driving Vehicles4 Feb 2022
- 2022Tutorial on amortized optimization1 Feb 2022
- 2022Spatiotemporal Costmap Inference for MPC via Deep Inverse Reinforcement Learning17 Jan 2022
- 2021UMBRELLA: Uncertainty-Aware Model-Based Offline Reinforcement Learning Leveraging Planning22 Nov 2021
- 2021Anti-Concentrated Confidence Bonuses for Scalable Exploration21 Oct 2021
- 2021Urban Driver: Learning to Drive from Real-world Demonstrations Using Policy Gradients27 Sep 2021
- 2021PredictionNet: Real-Time Joint Probabilistic Traffic Prediction for Planning, Control, and Simulation23 Sep 2021
- 2021What classifiers know what they don't?13 Jul 2021
- 2021SimNet: Learning Reactive Self-driving Simulations from Real-world Observations26 May 2021
- 2021Reward (Mis)design for Autonomous Driving28 Apr 2021
- 2021MBRL-Lib: A Modular Library for Model-based Reinforcement Learning20 Apr 2021
- 2021Large Scale Autonomous Driving Scenarios Clustering with Self-supervised Feature Extraction30 Mar 2021
- 2021Generating and Characterizing Scenarios for Safety Testing of Autonomous Vehicles12 Mar 2021
- 2021TrafficSim: Learning to Simulate Realistic Multi-Agent Behaviors17 Jan 2021
- 2021A Survey of Deep RL and IL for Autonomous Driving Policy Learning6 Jan 2021
- 2021Learning the Predictability of the Future1 Jan 2021
- 2020SAFARI: Safe and Active Robot Imitation Learning with Imagination18 Nov 2020
- 2020Triple-GAIL: A Multi-Modal Imitation Learning Framework with Generative Adversarial Nets19 May 2020
- 2023End-to-end Autonomous Driving: Challenges and Frontierscited by 485
- 2022Model-Based Imitation Learning for Urban Drivingcited by 149
- 2021Urban Driver: Learning to Drive from Real-world Demonstrations Using Policy Gradientscited by 122
- 2021Reward (Mis)design for Autonomous Drivingcited by 102
- 2021SimNet: Learning Reactive Self-driving Simulations from Real-world Observationscited by 77
- 2021A Survey of Deep RL and IL for Autonomous Driving Policy Learningcited by 59
- 2023Rethinking Closed-loop Training for Autonomous Drivingcited by 29
- 2023CCIL: Context-conditioned imitation learning for urban drivingcited by 12
- 2024Learning to Drive via Asymmetric Self-Playcited by 11
- 2023Learning Interaction-aware Motion Prediction Model for Decision-making in Autonomous Drivingcited by 6
- 2022Enhanced Behavioral Cloning with Environmental Losses for Self-Driving Vehiclescited by 0
- 2024MRIC: Model-Based Reinforcement-Imitation Learning with Mixture-of-Codebooks for Autonomous Driving Simulationcited by 0
12 most cited of 14 All 14 (JSON)
- 2025Scalable Offline Metrics for Autonomous Driving9 Oct 2025
- 2024SoftCTRL: Soft conservative KL-control of Transformer Reinforcement Learning for Autonomous Driving30 Oct 2024
- 2024Learning to Drive via Asymmetric Self-Play26 Sep 2024
- 2024MRIC: Model-Based Reinforcement-Imitation Learning with Mixture-of-Codebooks for Autonomous Driving Simulation29 Apr 2024
- 2023End-to-end Autonomous Driving: Challenges and Frontiers29 Jun 2023
- 2023Rethinking Closed-loop Training for Autonomous Driving27 Jun 2023
- 2023CCIL: Context-conditioned imitation learning for urban driving4 May 2023
- 2023Learning Interaction-aware Motion Prediction Model for Decision-making in Autonomous Driving8 Feb 2023
- 2022Model-Based Imitation Learning for Urban Driving14 Oct 2022
- 2022Enhanced Behavioral Cloning with Environmental Losses for Self-Driving Vehicles4 Feb 2022
- 2021Urban Driver: Learning to Drive from Real-world Demonstrations Using Policy Gradients27 Sep 2021
- 2021SimNet: Learning Reactive Self-driving Simulations from Real-world Observations26 May 2021
12 newest of 14 All 14 (JSON)
- 2022DayDreamer: World Models for Physical Robot Learningcited by 372
- 2023Driving into the Future: Multiview Visual Forecasting and Planning with World Model for Autonomous Drivingcited by 309
- 2025End-to-End Driving with Online Trajectory Evaluation via BEV World Modelcited by 145
- 2024World Models for Autonomous Driving: An Initial Surveycited by 79
- 2024DrivingWorld: Constructing World Model for Autonomous Driving via Video GPTcited by 66
- 2025A Survey of World Models for Autonomous Drivingcited by 60
- 2023TrafficBots: Towards World Models for Autonomous Driving Simulation and Motion Predictioncited by 56
- 2023MUVO: A Multimodal Generative World Model for Autonomous Driving with Geometric Representationscited by 33
- 2023MoDem-V2: Visuo-Motor World Models for Real-World Robot Manipulationcited by 27
- 2026Risk-Aware World Model Predictive Control for Generalizable End-to-End Autonomous Drivingcited by 1
- 2026Risk-Aware World Model Predictive Control for Generalizable End-to-End Autonomous Driving26 Feb 2026
- 2025End-to-End Driving with Online Trajectory Evaluation via BEV World Model2 Apr 2025
- 2025A Survey of World Models for Autonomous Driving20 Jan 2025
- 2024DrivingWorld: Constructing World Model for Autonomous Driving via Video GPT27 Dec 2024
- 2024World Models for Autonomous Driving: An Initial Survey5 Mar 2024
- 2023Driving into the Future: Multiview Visual Forecasting and Planning with World Model for Autonomous Driving29 Nov 2023
- 2023MUVO: A Multimodal Generative World Model for Autonomous Driving with Geometric Representations20 Nov 2023
- 2023MoDem-V2: Visuo-Motor World Models for Real-World Robot Manipulation25 Sep 2023
- 2023TrafficBots: Towards World Models for Autonomous Driving Simulation and Motion Prediction7 Mar 2023
- 2022DayDreamer: World Models for Physical Robot Learning28 Jun 2022
- 2021UMBRELLA: Uncertainty-Aware Model-Based Offline Reinforcement Learning Leveraging Planningcited by 23
- 2022Plan Your Target and Learn Your Skills: Transferable State-Only Imitation Learning via Decoupled Policy Optimizationcited by 12
- 2020Triple-GAIL: A Multi-Modal Imitation Learning Framework with Generative Adversarial Netscited by 11
- 2022Spatiotemporal Costmap Inference for MPC via Deep Inverse Reinforcement Learningcited by 10
- 2020SAFARI: Safe and Active Robot Imitation Learning with Imaginationcited by 5
- 2022Plan Your Target and Learn Your Skills: Transferable State-Only Imitation Learning via Decoupled Policy Optimization4 Mar 2022
- 2022Spatiotemporal Costmap Inference for MPC via Deep Inverse Reinforcement Learning17 Jan 2022
- 2021UMBRELLA: Uncertainty-Aware Model-Based Offline Reinforcement Learning Leveraging Planning22 Nov 2021
- 2020SAFARI: Safe and Active Robot Imitation Learning with Imagination18 Nov 2020
- 2020Triple-GAIL: A Multi-Modal Imitation Learning Framework with Generative Adversarial Nets19 May 2020
- 2021TrafficSim: Learning to Simulate Realistic Multi-Agent Behaviorscited by 192
- 2022Tutorial on amortized optimizationcited by 83
- 2021Learning the Predictability of the Futurecited by 48
- 2021MBRL-Lib: A Modular Library for Model-based Reinforcement Learningcited by 41
- 2022Real-time Neural-MPC: Deep Learning Model Predictive Control for Quadrotors and Agile Robotic Platformscited by 36
- 2021PredictionNet: Real-Time Joint Probabilistic Traffic Prediction for Planning, Control, and Simulationcited by 29
- 2023Learning Realistic Traffic Agents in Closed-loopcited by 24
- 2021Generating and Characterizing Scenarios for Safety Testing of Autonomous Vehiclescited by 21
- 2023Explainable Action Prediction through Self-Supervision on Scene Graphscited by 9
- 2021Anti-Concentrated Confidence Bonuses for Scalable Explorationcited by 7
- 2021What classifiers know what they don't?cited by 6
- 2026The Surprising Difficulty of Search in Model-Based Reinforcement Learningcited by 5
12 most cited of 21 All 21 (JSON)
- 2026The Surprising Difficulty of Search in Model-Based Reinforcement Learning29 Jan 2026
- 2025Planning as Descent: Goal-Conditioned Latent Trajectory Synthesis in Learned Energy Landscapes19 Dec 2025
- 2025Dynamical modeling of nonlinear latent factors in multiscale neural activity with real-time inference13 Dec 2025
- 2025LaMOuR: Leveraging Language Models for Out-of-Distribution Recovery in Reinforcement Learning21 Mar 2025
- 2023SeRO: Self-Supervised Reinforcement Learning for Recovery from Out-of-Distribution Situations7 Nov 2023
- 2023Learning Realistic Traffic Agents in Closed-loop2 Nov 2023
- 2023Learning to Predict Navigational Patterns from Partial Observations26 Apr 2023
- 2023Explainable Action Prediction through Self-Supervision on Scene Graphs7 Feb 2023
- 2022Tailored Uncertainty Estimation for Deep Learning Systems29 Apr 2022
- 2022Real-time Neural-MPC: Deep Learning Model Predictive Control for Quadrotors and Agile Robotic Platforms15 Mar 2022
- 2022A Transferable Legged Mobile Manipulation Framework Based on Disturbance Predictive Control2 Mar 2022
- 2022KINet: Unsupervised Forward Models for Robotic Pushing Manipulation18 Feb 2022
12 newest of 21 All 21 (JSON)