SGDrive: Scene-to-Goal Hierarchical World Cognition for Autonomous Driving

Jingyu Li, Junjie Wu, Dongnan Hu, Xiangkai Huang, Bin Sun +4 more
1/9/2026
cs.CV

No official code found. Best implementation: LogosRoboticsGroup/SGDrive (71 stars, MIT, last commit 2026-04-15).

Abstract

Recent end-to-end autonomous driving approaches have leveraged Vision-Language Models (VLMs) to enhance planning capabilities in complex driving scenarios. However, VLMs are inherently trained as generalist models, lacking specialized understanding of driving-specific reasoning in 3D space and time. When applied to autonomous driving, these models struggle to establish structured spatial-temporal representations that capture geometric relationships, scene context, and motion patterns critical for safe trajectory planning. To address these limitations, we propose SGDrive, a novel framework that explicitly structures the VLM's representation learning around driving-specific knowledge hierarchies. Built upon a pre-trained VLM backbone, SGDrive decomposes driving understanding into a scene-agent-goal hierarchy that mirrors human driving cognition: drivers first perceive the overall environment (scene context), then attend to safety-critical agents and their behaviors, and finally formulate short-term goals before executing actions. This hierarchical decomposition provides the structured spatial-temporal representation that generalist VLMs lack, integrating multi-level information into a compact yet comprehensive format for trajectory planning. Extensive experiments on the NAVSIM benchmark demonstrate that SGDrive achieves state-of-the-art performance among camera-only methods on both PDMS and EPDMS, validating the effectiveness of hierarchical knowledge structuring for adapting generalist VLMs to autonomous driving.

View on arXivView PDF

Code Implementations(1)

MIT

[CVPR2026] SGDrive: Scene-to-Goal Hierarchical World Cognition for Autonomous Driving

715Shell, PythonJan 9, 20266 months agoMIT

Similar Papers

Cite this paper

@article{li2026sgdrive,
  title  = {SGDrive: Scene-to-Goal Hierarchical World Cognition for Autonomous Driving},
  author = {Jingyu Li and Junjie Wu and Dongnan Hu and Xiangkai Huang and Bin Sun and Zhihui Hao and Xianpeng Lang and Xiatian Zhu and Li Zhang},
  year   = {2026},
  eprint = {2601.05640},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2601.05640v2}
}

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/2601.05640.svg)](https://codeofpaper.com/paper/2601.05640)

Discussion