Generalizable Trajectory Prediction via Inverse Reinforcement Learning with Mamba-Graph Architecture
Fuente:
arXiv
Saved in:
| Main Authors: | , , , |
|---|---|
| Format: | Preprint |
| Published: |
2025
|
| Subjects: | |
| Online Access: | |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866915774773329920 |
|---|---|
| author | Li, Wenyun Huang, Wenjie Deng, Zejian Sun, Chen |
| author_facet | Li, Wenyun Huang, Wenjie Deng, Zejian Sun, Chen |
| contents | Accurate driving behavior modeling is fundamental to safe and efficient trajectory prediction, yet remains challenging in complex traffic scenarios. This paper presents a novel Inverse Reinforcement Learning (IRL) framework that captures human-like decision-making by inferring diverse reward functions, enabling robust cross-scenario adaptability. The learned reward function is utilized to maximize the likelihood of output by integrating Mamba blocks for efficient long-sequence dependency modeling with graph attention networks to encode spatial interactions among traffic agents. Comprehensive evaluations on urban intersections and roundabouts demonstrate that the proposed method not only outperforms various popular approaches in terms of prediction accuracy but also achieves 2.3 times higher generalization performance to unseen scenarios compared to other baselines, achieving adaptability in Out-of-Distribution settings that is competitive with fine-tuning. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2506_12474 |
| institution | arXiv |
| publishDate | 2025 |
| record_format | arxiv |
| spellingShingle | Generalizable Trajectory Prediction via Inverse Reinforcement Learning with Mamba-Graph Architecture Li, Wenyun Huang, Wenjie Deng, Zejian Sun, Chen Machine Learning Artificial Intelligence Accurate driving behavior modeling is fundamental to safe and efficient trajectory prediction, yet remains challenging in complex traffic scenarios. This paper presents a novel Inverse Reinforcement Learning (IRL) framework that captures human-like decision-making by inferring diverse reward functions, enabling robust cross-scenario adaptability. The learned reward function is utilized to maximize the likelihood of output by integrating Mamba blocks for efficient long-sequence dependency modeling with graph attention networks to encode spatial interactions among traffic agents. Comprehensive evaluations on urban intersections and roundabouts demonstrate that the proposed method not only outperforms various popular approaches in terms of prediction accuracy but also achieves 2.3 times higher generalization performance to unseen scenarios compared to other baselines, achieving adaptability in Out-of-Distribution settings that is competitive with fine-tuning. |
| title | Generalizable Trajectory Prediction via Inverse Reinforcement Learning with Mamba-Graph Architecture |
| topic | Machine Learning Artificial Intelligence |
| url | https://arxiv.org/abs/2506.12474 |