GoIRL: Graph-Oriented Inverse Reinforcement Learning for Multimodal Trajectory Prediction
On the map: Offline RL and imitation learning from data → Imitation and offline RL for driving
Abstract (arXiv)
Trajectory prediction for surrounding agents is a challenging task in autonomous driving due to its inherent uncertainty and underlying multimodality. Unlike prevailing data-driven methods that primarily rely on supervised learning, in this paper, we introduce a novel Graph-oriented Inverse Reinforcement Learning (GoIRL) framework, which is an IRL-based predictor equipped with vectorized context representations. We develop a feature adaptor to effectively aggregate lane-graph features into grid space, enabling seamless integration with the maximum entropy IRL paradigm to infer the reward distribution and obtain the policy that can be sampled to induce multiple plausible plans. Furthermore, conditioned on the sampled plans, we implement a hierarchical parameterized trajectory generator with a refinement module to enhance prediction accuracy and a probability fusion strategy to boost prediction confidence. Extensive experimental results showcase our approach not only achieves state-of-the-art performance on the large-scale Argoverse & nuScenes motion forecasting benchmarks but also exhibits superior generalization abilities compared to existing supervised models.
Key facts. Cited by 7 papers in the map (data through 14 Oct 2026). Placed in Imitation and offline RL for driving, one of 13 directions in Offline RL and imitation learning from data. Builds on 43 of its 55 references that are in the map. First posted on arXiv 26 Jun 2025.
Builds on
43 of 55 references.
Closest ancestors · 2
Same direction, newest first
- 2022Conditional Predictive Behavior Planning with Inverse Reinforcement Learning for Human-like Autonomous Driving17 Dec
- 2018Integrating kinematics and environment context into deep inverse reinforcement learning for predicting off-road vehicle trajectories16 Oct
Other references · 41
- 2017Attention Is All You Needcited by 60,361
- 2016PointNet: Deep Learning on Point Sets for 3D Classification and Segmentationcited by 5,248
- 2016Feature Pyramid Networks for Object Detectioncited by 5,036
- 2014Neural Machine Translation by Jointly Learning to Align and Translatecited by 4,548
- 2019nuScenes: A multimodal dataset for autonomous drivingcited by 4,523
- 2016Generative Adversarial Imitation Learningcited by 1,597
- 2019Argoverse: 3D Tracking and Forecasting with Rich Mapscited by 732
- 2018Social GAN: Socially Acceptable Trajectories with Generative Adversarial Networkscited by 690
- 2020VectorNet: Encoding HD Maps and Agent Dynamics from Vectorized Representationcited by 530
- 2023End-to-end Autonomous Driving: Challenges and Frontierscited by 490
- 2018An Algorithmic Perspective on Imitation Learningcited by 459
- 2016Guided Cost Learning: Deep Inverse Optimal Control via Policy Optimizationcited by 406
29 more
- 2020Learning Lane Graph Representations for Motion Forecastingcited by 382
- 2021DenseTNT: End-to-end Trajectory Prediction from Dense Goal Setscited by 295
- 2018Multimodal Trajectory Predictions for Autonomous Driving using Deep Convolutional Networkscited by 289
- 2022Wayformer: Motion Forecasting via Simple & Efficient Attention Networkscited by 280
- 2022Motion Transformer with Global Intention Localization and Local Movement Refinementcited by 270
- 2021MultiPath++: Efficient Information Fusion and Trajectory Aggregation for Behavior Predictioncited by 246
- 2021End-to-end Interpretable Neural Motion Plannercited by 242
- 2016A Connection between Generative Adversarial Networks, Inverse Reinforcement Learning, and Energy-Based Modelscited by 168
- 2021GOHOME: Graph-Oriented Heatmap Output for future Motion Estimationcited by 146
- 2021Latent Variable Sequential Set Transformers For Joint Multi-Agent Motion Predictioncited by 131
- 2021HOME: Heatmap Output for future Motion Estimationcited by 122
- 2021LaneRCNN: Distributed Representations for Graph-Centric Motion Forecastingcited by 113
- 2021Multimodal Trajectory Prediction Conditioned on Lane-Graph Traversalscited by 113
- 2021THOMAS: Trajectory Heatmap Output with learned Multi-Agent Samplingcited by 101
- 2021TPCN: Temporal Point Cloud Networks for Motion Forecastingcited by 94
- 2020Trajectory Forecasts in Unknown Environments Conditioned on Grid-Based Planscited by 72
- 2021Learning to Predict Vehicle Trajectories with Model-based Planningcited by 63
- 2019Meta-Inverse Reinforcement Learning with Probabilistic Context Variablescited by 50
- 2022GANet: Goal Area Network for Motion Forecastingcited by 49
- 2020Multimodal Deep Generative Models for Trajectory Prediction: A Conditional Variational Autoencoder Approachcited by 47
- 2023MacFormer: Map-Agent Coupled Transformer for Real-time and Robust Trajectory Predictioncited by 32
- 2023Dynamic Scenario Representation Learning for Motion Forecasting with Heterogeneous Graph Convolutional Recurrent Networkscited by 28
- 2023SEPT: Towards Efficient Scene Representation Learning for Motion Predictioncited by 27
- 2024SIMPL: A Simple and Efficient Multi-agent Motion Prediction Baseline for Autonomous Drivingcited by 27
- 2019Scene Compliant Trajectory Forecast with Agent-Centric Spatio-Temporal Gridscited by 24
- 2022End-to-End Trajectory Distribution Prediction Based on Occupancy Grid Mapscited by 22
- 2021Trajectory Prediction with Graph-based Dual-scale Context Fusioncited by 19
- 2024Improving Out-of-Distribution Generalization of Trajectory Prediction for Autonomous Driving via Polynomial Representationscited by 9
- 2023PBP: Path-based Trajectory Prediction for Autonomous Drivingcited by 6
Built on by
Citing papers per year: 2025: 5; 2026: 2.
7 papers.
- 2025Advancing Multi-agent Traffic Simulation via R1-Style Reinforcement Fine-Tuningcited by 14
- 2025ColaVLA: Leveraging Cognitive Latent Reasoning for Hierarchical Parallel Trajectory Planning in Autonomous Drivingcited by 11
- 2025Foresight in Motion: Reinforcing Trajectory Prediction with Reward Heuristicscited by 6
- 2026VG3S: Visual Geometry Grounded Gaussian Splatting for Semantic Occupancy Predictioncited by 5
- 2025ST-GS: Vision-Based 3D Semantic Occupancy Prediction with Spatial-Temporal Gaussian Splattingcited by 4
- 2026Adapter-Augmented Bandits for Online Multi-Constrained Multi-Modal Inference Schedulingcited by 2
- 2025ILNet: Trajectory Prediction with Inverse Learning Attention for Enhancing Intention Capturecited by 0
- 2026Adapter-Augmented Bandits for Online Multi-Constrained Multi-Modal Inference Scheduling6 Mar 2026
- 2026VG3S: Visual Geometry Grounded Gaussian Splatting for Semantic Occupancy Prediction6 Mar 2026
- 2025ColaVLA: Leveraging Cognitive Latent Reasoning for Hierarchical Parallel Trajectory Planning in Autonomous Driving28 Dec 2025
- 2025Advancing Multi-agent Traffic Simulation via R1-Style Reinforcement Fine-Tuning28 Sep 2025
- 2025ST-GS: Vision-Based 3D Semantic Occupancy Prediction with Spatial-Temporal Gaussian Splatting20 Sep 2025
- 2025Foresight in Motion: Reinforcing Trajectory Prediction with Reward Heuristics16 Jul 2025
- 2025ILNet: Trajectory Prediction with Inverse Learning Attention for Enhancing Intention Capture9 Jul 2025
Similar, not cited
Close in wording, no citation either way
- 2022Driving in Real Life with Inverse Reinforcement Learning
- 2025Generalizable Trajectory Prediction via Inverse Reinforcement Learning with Mamba-Graph Architecture
- 2020Efficient Sampling-Based Maximum Entropy Inverse Reinforcement Learning with Application to Autonomous Driving
- 2026Game-Theoretic Inverse Reinforcement Learning for Modeling Competitive Human Driving: A Cut-in Prediction Study
- 2018Probabilistic Prediction of Interactive Driving Behavior via Hierarchical Inverse Reinforcement Learning
- 2025TreeIRL: Safe Urban Driving with Tree Search and Inverse Reinforcement Learning
- 2020Planning on the fast lane: Learning to interact using attention mechanisms in path integral inverse reinforcement learning
- 2023NashFormer: Leveraging Local Nash Equilibria for Semantically Diverse Trajectory Prediction