R-Pred: Two-Stage Motion Prediction Via Tube-Query Attention-Based Trajectory Refinement
Not on the map. Why?
Abstract (arXiv)
Predicting the future motion of dynamic agents is of paramount importance to ensuring safety and assessing risks in motion planning for autonomous robots. In this study, we propose a two-stage motion prediction method, called R-Pred, designed to effectively utilize both scene and interaction context using a cascade of the initial trajectory proposal and trajectory refinement networks. The initial trajectory proposal network produces M trajectory proposals corresponding to the M modes of the future trajectory distribution. The trajectory refinement network enhances each of the M proposals using 1) tube-query scene attention (TQSA) and 2) proposal-level interaction attention (PIA) mechanisms. TQSA uses tube-queries to aggregate local scene context features pooled from proximity around trajectory proposals of interest. PIA further enhances the trajectory proposals by modeling inter-agent interactions using a group of trajectory proposals selected by their distances from neighboring agents. Our experiments conducted on Argoverse and nuScenes datasets demonstrate that the proposed refinement network provides significant performance improvements compared to the single-stage baseline and that R-Pred achieves state-of-the-art performance in some categories of the benchmarks.
Key facts. Cited by 23 papers in the map (data through 7 Oct 2026). Builds on 45 of its 56 references that are in the map. First posted on arXiv 16 Nov 2022.
Builds on
45 of 56 references.
References · 45
- 2017Attention Is All You Needcited by 59,932
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2015Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networkscited by 9,617
- 2017Mask R-CNNcited by 7,259
- 2016Layer Normalizationcited by 5,579
- 2019nuScenes: A multimodal dataset for autonomous drivingcited by 4,485
- 2015Fast R-CNNcited by 2,930
- 2014Empirical Evaluation of Gated Recurrent Neural Networks on Sequence Modelingcited by 2,341
- 2019Argoverse: 3D Tracking and Forecasting with Rich Mapscited by 728
- 2018Social GAN: Socially Acceptable Trajectories with Generative Adversarial Networkscited by 687
- 2020VectorNet: Encoding HD Maps and Agent Dynamics from Vectorized Representationcited by 528
- 2019MultiPath: Multiple Probabilistic Anchor Trajectory Hypotheses for Behavior Predictioncited by 417
33 more
- 2020Learning Lane Graph Representations for Motion Forecastingcited by 380
- 2020TNT: Target-driveN Trajectory Predictioncited by 345
- 2021AgentFormer: Agent-Aware Transformers for Socio-Temporal Multi-Agent Forecastingcited by 327
- 2017DESIRE: Distant Future Prediction in Dynamic Scenes with Interacting Agentscited by 310
- 2018SoPhie: An Attentive GAN for Predicting Paths Compliant to Social and Physical Constraintscited by 296
- 2021DenseTNT: End-to-end Trajectory Prediction from Dense Goal Setscited by 292
- 2018Multimodal Trajectory Predictions for Autonomous Driving using Deep Convolutional Networkscited by 289
- 2022Wayformer: Motion Forecasting via Simple & Efficient Attention Networkscited by 279
- 2019CoverNet: Multimodal Behavior Prediction using Trajectory Setscited by 259
- 2021MultiPath++: Efficient Information Fusion and Trajectory Aggregation for Behavior Predictioncited by 246
- 2018Convolutional Social Pooling for Vehicle Trajectory Predictioncited by 221
- 2021Multimodal Motion Prediction with Stacked Transformerscited by 207
- 2021GOHOME: Graph-Oriented Heatmap Output for future Motion Estimationcited by 146
- 2019Multi-Agent Tensor Fusion for Contextual Trajectory Predictioncited by 131
- 2021Latent Variable Sequential Set Transformers For Joint Multi-Agent Motion Predictioncited by 131
- 2022GroupNet: Multiscale Hypergraph Neural Networks for Trajectory Prediction with Relational Reasoningcited by 128
- 2021HOME: Heatmap Output for future Motion Estimationcited by 122
- 2021LaneRCNN: Distributed Representations for Graph-Centric Motion Forecastingcited by 112
- 2021Multimodal Trajectory Prediction Conditioned on Lane-Graph Traversalscited by 112
- 2021THOMAS: Trajectory Heatmap Output with learned Multi-Agent Samplingcited by 101
- 2021TPCN: Temporal Point Cloud Networks for Motion Forecastingcited by 94
- 2018Uncertainty-aware Short-term Motion Prediction of Traffic Actors for Autonomous Drivingcited by 90
- 2020Trajectory Forecasts in Unknown Environments Conditioned on Grid-Based Planscited by 72
- 2020Diverse and Admissible Trajectory Forecasting through Multimodal Context Understandingcited by 68
- 2021LaPred: Lane-Aware Prediction of Multi-Modal Future Trajectories of Dynamic Agentscited by 64
- 2021Learning to Predict Vehicle Trajectories with Model-based Planningcited by 63
- 2017Probabilistic Vehicle Trajectory Prediction over Occupancy Grid Map via Recurrent Neural Networkcited by 56
- 2022Adaptive Trajectory Prediction via Transferable GNNcited by 51
- 2017Soft + Hardwired Attention: An LSTM Framework for Human Trajectory Prediction and Abnormal Event Detectioncited by 42
- 2020Map-Adaptive Goal-Based Trajectory Predictioncited by 29
- 2020Probabilistic Multi-modal Trajectory Prediction with Lane Attention for Autonomous Vehiclescited by 28
- 2020Interaction-Based Trajectory Prediction Over a Hybrid Traffic Graphcited by 17
- 2021AIR2 for Interaction Predictioncited by 9
Built on by
Citing papers per year: 2024: 9; 2025: 8; 2026: 6.
23 papers.
Sort:
By area:
- 2024HPNet: Dynamic Trajectory Forecasting with Historical Prediction Attentioncited by 37
- 2024SmartRefine: A Scenario-Adaptive Refinement Framework for Efficient Motion Predictioncited by 33
- 2024Can Language Beat Numerical Regression? Language-Based Multimodal Trajectory Predictioncited by 31
- 2024Multi-agent Long-term 3D Human Pose Forecasting via Interaction-aware Trajectory Conditioningcited by 29
- 2024SIMPL: A Simple and Efficient Multi-agent Motion Prediction Baseline for Autonomous Drivingcited by 27
- 2024Mask2Map: Vectorized HD Map Construction Using Bird's Eye View Segmentation Maskscited by 26
- 2024T4P: Test-Time Training of Trajectory Prediction via Masked Autoencoder and Actor-specific Token Memorycited by 20
- 2025Challenger: Affordable Adversarial Driving Video Generationcited by 16
- 2024Motion Forecasting in Continuous Drivingcited by 11
- 2024SmartPretrain: Model-Agnostic and Dataset-Agnostic Representation Learning for Motion Predictioncited by 7
- 2025Continuous Locomotive Crowd Behavior Generationcited by 7
- 2025DONUT: A Decoder-Only Model for Trajectory Predictioncited by 6
11 more
- 2025Future-Aware Interaction Network For Motion Forecastingcited by 4
- 2026Recover to Predict: Progressive Retrospective Learning for Variable-Length Trajectory Predictioncited by 2
- 2026Streaming Real-Time Trajectory Prediction Using Endpoint-Aware Modelingcited by 1
- 2026Hyperspectral Image Classification via Efficient Global Spectral Supertoken Clusteringcited by 1
- 2025LTMSformer: A Local Trend-Aware Attention and Motion State Encoding Transformer for Multi-Agent Trajectory Predictioncited by 0
- 2025ILNet: Trajectory Prediction with Inverse Learning Attention for Enhancing Intention Capturecited by 0
- 2025BEVTraj: Map-Free End-to-End Trajectory Prediction in Bird's-Eye View with Deformable Attention and Sparse Goal Proposalscited by 0
- 2025Ensemble of Pre-Trained Models for Long-Tailed Trajectory Predictioncited by 0
- 2026TaPD: Temporal-adaptive Progressive Distillation for Observation-Adaptive Trajectory Forecasting in Autonomous Drivingcited by 0
- 2026SHARP: Short-Window Streaming for Accurate and Robust Prediction in Motion Forecastingcited by 0
- 2026Ego-Human Motion Prediction with 3D-Aware LLMcited by 0
- 2026Ego-Human Motion Prediction with 3D-Aware LLM8 Jul 2026
- 2026Hyperspectral Image Classification via Efficient Global Spectral Supertoken Clustering30 Apr 2026
- 2026SHARP: Short-Window Streaming for Accurate and Robust Prediction in Motion Forecasting30 Mar 2026
- 2026Recover to Predict: Progressive Retrospective Learning for Variable-Length Trajectory Prediction11 Mar 2026
- 2026TaPD: Temporal-adaptive Progressive Distillation for Observation-Adaptive Trajectory Forecasting in Autonomous Driving6 Mar 2026
- 2026Streaming Real-Time Trajectory Prediction Using Endpoint-Aware Modeling2 Mar 2026
- 2025Ensemble of Pre-Trained Models for Long-Tailed Trajectory Prediction17 Sep 2025
- 2025BEVTraj: Map-Free End-to-End Trajectory Prediction in Bird's-Eye View with Deformable Attention and Sparse Goal Proposals12 Sep 2025
- 2025ILNet: Trajectory Prediction with Inverse Learning Attention for Enhancing Intention Capture9 Jul 2025
- 2025LTMSformer: A Local Trend-Aware Attention and Motion State Encoding Transformer for Multi-Agent Trajectory Prediction7 Jul 2025
- 2025DONUT: A Decoder-Only Model for Trajectory Prediction7 Jun 2025
- 2025Challenger: Affordable Adversarial Driving Video Generation21 May 2025
11 more
- 2025Continuous Locomotive Crowd Behavior Generation7 Apr 2025
- 2025Future-Aware Interaction Network For Motion Forecasting9 Mar 2025
- 2024SmartPretrain: Model-Agnostic and Dataset-Agnostic Representation Learning for Motion Prediction11 Oct 2024
- 2024Motion Forecasting in Continuous Driving8 Oct 2024
- 2024Mask2Map: Vectorized HD Map Construction Using Bird's Eye View Segmentation Masks18 Jul 2024
- 2024HPNet: Dynamic Trajectory Forecasting with Historical Prediction Attention9 Apr 2024
- 2024Multi-agent Long-term 3D Human Pose Forecasting via Interaction-aware Trajectory Conditioning8 Apr 2024
- 2024Can Language Beat Numerical Regression? Language-Based Multimodal Trajectory Prediction27 Mar 2024
- 2024SmartRefine: A Scenario-Adaptive Refinement Framework for Efficient Motion Prediction18 Mar 2024
- 2024T4P: Test-Time Training of Trajectory Prediction via Masked Autoencoder and Actor-specific Token Memory15 Mar 2024
- 2024SIMPL: A Simple and Efficient Multi-agent Motion Prediction Baseline for Autonomous Driving4 Feb 2024
- 2025Challenger: Affordable Adversarial Driving Video Generationcited by 16
- 2025Challenger: Affordable Adversarial Driving Video Generation21 May 2025
- 2025Future-Aware Interaction Network For Motion Forecastingcited by 4
- 2025Future-Aware Interaction Network For Motion Forecasting9 Mar 2025
- 2024HPNet: Dynamic Trajectory Forecasting with Historical Prediction Attentioncited by 37
- 2024SmartRefine: A Scenario-Adaptive Refinement Framework for Efficient Motion Predictioncited by 33
- 2024Can Language Beat Numerical Regression? Language-Based Multimodal Trajectory Predictioncited by 31
- 2024Multi-agent Long-term 3D Human Pose Forecasting via Interaction-aware Trajectory Conditioningcited by 29
- 2024SIMPL: A Simple and Efficient Multi-agent Motion Prediction Baseline for Autonomous Drivingcited by 27
- 2024Mask2Map: Vectorized HD Map Construction Using Bird's Eye View Segmentation Maskscited by 26
- 2024T4P: Test-Time Training of Trajectory Prediction via Masked Autoencoder and Actor-specific Token Memorycited by 20
- 2024Motion Forecasting in Continuous Drivingcited by 11
- 2024SmartPretrain: Model-Agnostic and Dataset-Agnostic Representation Learning for Motion Predictioncited by 7
- 2025Continuous Locomotive Crowd Behavior Generationcited by 7
- 2025DONUT: A Decoder-Only Model for Trajectory Predictioncited by 6
- 2026Recover to Predict: Progressive Retrospective Learning for Variable-Length Trajectory Predictioncited by 2
12 most cited of 21 All 21 (JSON)
- 2026Ego-Human Motion Prediction with 3D-Aware LLM8 Jul 2026
- 2026Hyperspectral Image Classification via Efficient Global Spectral Supertoken Clustering30 Apr 2026
- 2026SHARP: Short-Window Streaming for Accurate and Robust Prediction in Motion Forecasting30 Mar 2026
- 2026Recover to Predict: Progressive Retrospective Learning for Variable-Length Trajectory Prediction11 Mar 2026
- 2026TaPD: Temporal-adaptive Progressive Distillation for Observation-Adaptive Trajectory Forecasting in Autonomous Driving6 Mar 2026
- 2026Streaming Real-Time Trajectory Prediction Using Endpoint-Aware Modeling2 Mar 2026
- 2025Ensemble of Pre-Trained Models for Long-Tailed Trajectory Prediction17 Sep 2025
- 2025BEVTraj: Map-Free End-to-End Trajectory Prediction in Bird's-Eye View with Deformable Attention and Sparse Goal Proposals12 Sep 2025
- 2025ILNet: Trajectory Prediction with Inverse Learning Attention for Enhancing Intention Capture9 Jul 2025
- 2025LTMSformer: A Local Trend-Aware Attention and Motion State Encoding Transformer for Multi-Agent Trajectory Prediction7 Jul 2025
- 2025DONUT: A Decoder-Only Model for Trajectory Prediction7 Jun 2025
- 2025Continuous Locomotive Crowd Behavior Generation7 Apr 2025
12 newest of 21 All 21 (JSON)