A knowledge-augmented dataset of high-risk driving scenarios with LLM annotations for autonomous driving

Heye Huang, Jingguang Li, Zhiyuan Zhou, Paul Liang, Mingyu Wu +2 more
7/8/2026
cs.LGcs.DB

No official code found. Best implementation: benmagnifico/K-Risk (3 stars, MIT, last commit 2026-07-04).

Abstract

Safe autonomous driving requires both rapid responses to common high-risk events and deeper reasoning over rare, extreme long-tail scenarios in traffic safety. These scenarios are severely under-represented in naturalistic driving data, and existing trajectory and language-augmented datasets seldom provide high-risk event labels, semantic annotations, and verifiable safety signals. Here we present K-Risk, a knowledge-augmented dataset that combines structured driving trajectories with large language model generated semantic annotations for safety-critical driving scenarios. K-Risk integrates 20 human-driven and autonomous-vehicle trajectory datasets from Europe, China, and the United States, covering highways, urban freeways, intersections, and roundabouts. Using a unified risk-centric extraction pipeline, K-Risk curates 31,398 high-risk events, together with a 1,036-event extreme subset of near-collision cases. Each event is released as a synchronized trajectory, metadata, and language triplet containing structured scenario descriptions, abnormal-behavior notifications, and, for a representative subset, causal risk analyses and action recommendations validated through a closed-loop simulator with iterative reflection. By combining multi-dimensional risk annotations, interpretable language supervision, and verifiable decisions, K-Risk bridges structured traffic trajectories, semantic reasoning, and decision supervision, providing a standardized foundation for developing and evaluating next-generation risk-aware autonomous driving agents.

View on arXivView PDF

Code Implementations(1)

MIT

A knowledge-augmented dataset of high-risk driving scenarios with LLM annotations for autonomous driving.

31Jun 25, 20263 months agoMIT

Similar Papers

Cite this paper

@article{huang2026knowledgeaugmented,
  title  = {A knowledge-augmented dataset of high-risk driving scenarios with LLM annotations for autonomous driving},
  author = {Heye Huang and Jingguang Li and Zhiyuan Zhou and Paul Liang and Mingyu Wu and Kitae Jang and Jianqiang Wang},
  year   = {2026},
  eprint = {2607.07103},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2607.07103v1}
}

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

Discussion