# IMR: Iterative Mode-World Weighted Regression for Multi-Agent Trajectory Prediction

> Source: <https://www.machinebrief.com/news/imr-iterative-mode-world-weighted-regression-for-multi-agent-vg4f>
> Published: 2026-07-08 04:00:00+00:00

arXiv:2607.05705v1 Announce Type: cross
Abstract: Multi-agent motion prediction is essential for automated vehicles to understand the intentions of surrounding vehicles. However, previous prediction-based and anchor-based methods have limitations in mode diversity and prediction accuracy, respectively. These limitations may cause inadequate safety assessments and behavioral deviations in automated vehicles. To address this issue, a mode-world weighted regression loss is proposed to bridge the gap between these features. Specifically, this approach mitigates mode collapse while simultaneously improving world ranking and top-1 confidence. Furthermore, the proposed iterative decoder improves prediction accuracy by recurrently and segmentally generating trajectories. Experimental results show the proposed method ranks first in the Argoverse 2 multi-agent motion forecasting benchmark against other methods.
