UniDrive: A Unified Vision-Language and Grounding Framework for Interpretable Risk Understanding in Autonomous Driving

Xiaowei Gao, Pengxiang Li, Yitai Cheng, Ruihan Xu, James Haworth +2 more
6/23/2026
cs.CVcs.AI

Official code: pixeli99/unidrive-dev (1 star, Apache-2.0, last commit 2026-02-10).

Abstract

Recent multimodal large language models (MLLMs) have shown strong potential for autonomous driving scene understanding, yet existing methods still face a fundamental trade-off between temporal reasoning and spatial precision. Models that rely on single-frame or low-resolution inputs often miss small, distant, or partially occluded hazards, while language-centric driving models frequently provide limited grounded evidence for their explanations. To address this gap, we propose UniDrive, a unified visual-language and grounding framework for interpretable risk understanding in autonomous driving. UniDrive combines a temporal reasoning branch that models scene dynamics from multi-frame visual input with a high-resolution perception branch that preserves fine-grained spatial details from the latest frame. The two branches are integrated through a gated cross-attention fusion module, enabling dynamic context to be aligned with precise spatial evidence. Based on the fused representation, UniDrive jointly generates natural-language risk descriptions and grounded bounding-box outputs for risk objects. Experiments on the DRAMA-Reasoning benchmark show that UniDrive outperforms representative image-based and video-based baselines in both captioning and risk-object grounding. In particular, UniDrive achieves the best overall performance on the validation split and demonstrates clear advantages in small-object localization, zero-shot generalization to NuScenes and BDD100K, and human-rated interpretability and trustworthiness. These results suggest that explicitly combining temporal semantics and high-resolution perception provides a stronger foundation for interpretable and safety-oriented autonomous driving systems. The code is available at https://github.com/pixeli99/unidrive-dev.

View on arXivView PDF

Code Implementations(1)

Apache-2.0
10CSS, HTMLFeb 10, 20268 months agoApache-2.0

Similar Papers

Cite this paper

@article{gao2026unidrive,
  title  = {UniDrive: A Unified Vision-Language and Grounding Framework for Interpretable Risk Understanding in Autonomous Driving},
  author = {Xiaowei Gao and Pengxiang Li and Yitai Cheng and Ruihan Xu and James Haworth and Stephen Law and Yun Ye},
  year   = {2026},
  eprint = {2606.24759},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2606.24759v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2606.24759.svg)](https://codeofpaper.com/paper/2606.24759)

Discussion