arXiv Machine Learning

Uncertainty-Aware Optimization for Physics-Aware Highway Trajectory Prediction

The paper introduces uncertainty-aware extensions of the X-TRACK physics‑aware trajectory prediction framework, namely X-TRACK-DE and X-TRACK-MCD. These extensions predict future vehicle motion variables while explicitly modeling and propagating both aleatoric and epistemic uncertainties through vehicle dynamics, rather than only in trajectory space. Conformal prediction is then applied to the trajectory‑space covariance to create uncertainty regions that achieve a desired marginal coverage level, and experiments on the highD dataset demonstrate improved prediction accuracy and calibrated uncertainty estimates.

arXiv AI
Aug 20

DA-WAM: Decision-Aligned Future Latents for Driving World Models

DA‑WAM is a framework that integrates predictive representation learning, action‑conditioned future modeling, and trajectory scoring into a single decision‑making objective for autonomous driving. It uses an online encoder with a stable momentum target to keep future representations aligned with the driving task, generating a distinct future latent for each trajectory candidate. A future‑latent‑conditioned scorer evaluates these latents, with expert‑matched trajectories supervised by observed futures and safety‑critical hard negatives providing additional guidance, achieving state‑of‑the‑art results on NAVSIM‑v1 and NAVSIM‑v2.

By Ruiguo Zhong, Benshan Ma, Xiaolong Chen, Lang Zhang, Mingyue Feng, Yaonong Wang, Pei Liu, Jun Ma
arXiv Machine Learning
Aug 11

Particle-Based Conformal Prediction for Contact-Aware Uncertainty Calibration in Stratified Configuration Spaces

arXiv:2608. 09166v1 Announce Type: cross Abstract: Reliable uncertainty representation is essential for deploying autonomous systems that interact with their environment, as robots must reason about how uncertainty arising from both stochasticity and model mismatch is impacted by contacts with obstacles (e.

By Lu\'is Marques, Kristian Popov, Dmitry Berenson
arXiv Machine Learning
Sep 25

Active Client Selection in Federated Trajectory Prediction with Uncertainty-Awareness and Heterogeneous Complexity

The paper introduces active client selection strategies for federated learning in autonomous vehicle trajectory prediction, addressing challenges of high scene uncertainty and heterogeneous complexity across different driving environments. It proposes uncertainty-aware selectors that use per-client negative log-likelihood and aleatoric uncertainty, as well as a joint selector that balances scene complexity and uncertainty to prioritize informative clients. Experiments on the Argoverse dataset show that federated models outperform local training, with uncertainty-aware selection speeding convergence and improving key metrics, while the joint selector yields the best generalization under strong heterogeneity.

By Yiming Xie, Muzi Peng, Fei Miao, Ningfang Mi, Lili Su