HEAT: Heterogeneous End-to-End Autonomous Driving via Trajectory-Guided World Models

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Cho, Hoonhee, Lee, Giwon, Kang, Jae-Young, Yang, Hyemin, Park, Heejun, Yoon, Kuk-Jin
Formato: Preprint
Publicado: 2026
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866911698308300800
author Cho, Hoonhee
Lee, Giwon
Kang, Jae-Young
Yang, Hyemin
Park, Heejun
Yoon, Kuk-Jin
author_facet Cho, Hoonhee
Lee, Giwon
Kang, Jae-Young
Yang, Hyemin
Park, Heejun
Yoon, Kuk-Jin
contents End-to-end autonomous driving has emerged as a compelling alternative to traditional modular pipelines by directly mapping raw sensor data to driving actions. While recent approaches achieve strong performance on single-domain datasets, their performance degrades significantly when trained jointly across multiple heterogeneous domains. In practice, however, autonomous systems must operate across diverse environments with heterogeneous distributions, including different cities, sensor configurations, and traffic patterns, without domain-specific retraining. This gap highlights a key challenge in multi-domain learning: domain-specific variations across heterogeneous domains introduce conflicting learning signals, driving models toward compromised solutions that are suboptimal across domains. To address this, we propose a trajectory-driven learning paradigm that organizes training around planning trajectories, enabling the model to capture domain-invariant representations of driving intent. Furthermore, we incorporate a world model that predicts future latent features conditioned on ego actions, improving feature consistency and mitigating domain-induced biases. We evaluate our approach on three benchmarks, nuScenes, NAVSIM, and the Waymo end-to-end dataset, and show substantial improvements over existing methods across all domains. Our results demonstrate that a single unified model can be trained on heterogeneous datasets while maintaining strong performance within each domain, highlighting a step toward scalable real-world deployment. We will make our code publicly available.
format Preprint
id arxiv_https___arxiv_org_abs_2605_19631
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle HEAT: Heterogeneous End-to-End Autonomous Driving via Trajectory-Guided World Models
Cho, Hoonhee
Lee, Giwon
Kang, Jae-Young
Yang, Hyemin
Park, Heejun
Yoon, Kuk-Jin
Robotics
Computer Vision and Pattern Recognition
End-to-end autonomous driving has emerged as a compelling alternative to traditional modular pipelines by directly mapping raw sensor data to driving actions. While recent approaches achieve strong performance on single-domain datasets, their performance degrades significantly when trained jointly across multiple heterogeneous domains. In practice, however, autonomous systems must operate across diverse environments with heterogeneous distributions, including different cities, sensor configurations, and traffic patterns, without domain-specific retraining. This gap highlights a key challenge in multi-domain learning: domain-specific variations across heterogeneous domains introduce conflicting learning signals, driving models toward compromised solutions that are suboptimal across domains. To address this, we propose a trajectory-driven learning paradigm that organizes training around planning trajectories, enabling the model to capture domain-invariant representations of driving intent. Furthermore, we incorporate a world model that predicts future latent features conditioned on ego actions, improving feature consistency and mitigating domain-induced biases. We evaluate our approach on three benchmarks, nuScenes, NAVSIM, and the Waymo end-to-end dataset, and show substantial improvements over existing methods across all domains. Our results demonstrate that a single unified model can be trained on heterogeneous datasets while maintaining strong performance within each domain, highlighting a step toward scalable real-world deployment. We will make our code publicly available.
title HEAT: Heterogeneous End-to-End Autonomous Driving via Trajectory-Guided World Models
topic Robotics
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2605.19631