📄 论文信息
- 标题:A knowledge-augmented dataset of high-risk driving scenarios with LLM annotations for autonomous driving(面向自动驾驶的、带 LLM 标注的知识增强高风险驾驶场景数据集)
- 团队:Heye Huang、Jingguang Li、Zhiyuan Zhou、Paul Liang、Mingyu Wu、Kitae Jang、Jianqiang Wang(清华大学等)
- arXiv:2607.07103(2026 年 7 月 8 日,22 页 9 图)
- 领域:cs.LG / cs.DB
- 数据集名:K-Risk(Knowledge-augmented Risk dataset)
- 一句话总结:把"高风险事件标签 + 语义语言标注 + 可验证的安全信号“这三样社区一直缺的东西打包到一起,构建了一个横跨欧美中、覆盖多路况、语言可解释、决策可闭环验证的标准化风险驾驶数据集——为下一代风险感知自动驾驶智能体的训练和评测铺好地基。
🏗️ 架构总览

K-Risk 的整体架构由四大模块组成,形成一个从原始轨迹到可验证语义标注的完整流水线:
| 模块 | 输入 | 输出 | 核心组件 |
|---|---|---|---|
| 多源数据整合 | 20 个异构轨迹数据集 | 归一化的轨迹 + 元数据 | 统一 Schema 转换器 |
| 风险中心抽取 | 归一化轨迹 | 31,398 个高风险事件 | DRF 过滤器 + TTC 冲突检测 |
| LLM 标注生成 | 高风险事件轨迹 | 语言三元组标注 | GPT-4 + 结构化 Prompt |
| 闭环仿真验证 | 语言标注 + 建议动作 | 验证后的安全决策 | Collision-Free Simulator |
多源数据整合的挑战
20 个数据集的格式差异巨大:NGSIM 用 10 Hz 的 CSV 记录车辆位置,Waymo 用 ProtoBuf 存 10 Hz 的 3D 边界框,highD 用 25 Hz 的鸟瞰轨迹。K-Risk 的归一化层将它们统一到统一的轨迹 Schema——每条轨迹包含时间戳、位置 (x, y)、航向角 (heading)、速度、加速度,以及车道拓扑信息。这个归一化本身就是一个重要的工程贡献。
DRF 驱动的风险筛选
K-Risk 的风险筛选核心是一个Driver Risk Field (DRF) 模型——它把每个参与者对周围空间的"风险贡献"建模为一个高斯场,沿预测路径累积。DRF 的数学定义如下:
$$R_{\text{DRF}}(\mathbf{p}) = \sum_{i} \int_{0}^{T} \frac{w_i}{2\pi\sigma_i^2} \exp\left(-\frac{\|\mathbf{p} - \mathbf{p}_i(t)\|^2}{2\sigma_i^2}\right) dt$$其中 $\mathbf{p}_i(t)$ 是第 $i$ 个参与者在时间 $t$ 的预测位置,$\sigma_i$ 取决于其速度与尺寸,$w_i$ 是类型权重。综合得分位于前 10% 的事件被标记为"高风险”,前 1% 为"极端近碰撞"。配合 **TTC(Time-to-Collision, $<2s/3s/5s$)**和急动度阈值 $|a| > 4 m/s^2$,构成了三层筛选漏斗。
🤔 要解决什么问题?
安全的自动驾驶需要两种能力:对常见高风险事件的快速反应,以及对罕见极端长尾场景的深度推理。但这两类场景在自然驾驶数据里都严重欠表示——毕竟谁也不会为了采数据故意去撞车。
现有数据集各有各的残缺:
| 现有数据集类型 | 代表 | 缺什么 |
|---|---|---|
| 轨迹数据集 | NGSIM、HighD、nuPlan | 有轨迹但无高风险事件标签、无语义解释 |
| 语言增强数据集 | DriveGPT-4、LMDrive | 有语言但缺高风险标注、缺可验证的安全信号 |
| 碰撞/事故数据集 | 各种事故合集 | 样本极少、单一来源、不可闭环验证 |
三件套——高风险事件标签 + 语义标注 + 可验证决策信号——同时具备的数据集在社区里几乎不存在。K-Risk 就是要补上这个空白。
核心命题:一个好的风险驾驶数据集,必须同时满足四个条件——规模化(覆盖足够多的高风险事件)、多样化(跨地域跨路况)、可解释(带语言描述与因果分析)、可验证(能在闭环仿真里检验决策)。K-Risk 就是围绕这四点构建的。
💡 核心思想:风险中心 + 知识增强
K-Risk 的整体设计可以用一句话概括:用统一的"风险中心"抽取流水线,把分散在全球的数据统一成"轨迹 + 元数据 + 语言三元组"的标准件,再用 LLM 注入语义知识,最后用闭环仿真器验证决策。
四大设计支柱
| 支柱 | 做法 | 价值 |
|---|---|---|
| 规模化整合 | 融合 20 个人类驾驶与自动驾驶轨迹数据集 | 解决单源数据偏置 |
| 地理多样性 | 覆盖欧洲、中国、美国三大洲 | 跨文化/跨法规泛化 |
| 路况多样性 | 高速、城市快速路、交叉口、环岛 | 覆盖主要风险场景类型 |
| 风险中心抽取 | 统一的 risk-centric extraction pipeline | 一致的筛选标准 |
数据规模
K-Risk 的最终规模相当可观,且按风险等级做了精细划分:
| 类别 | 数量 | 筛选标准 | 地理覆盖 |
|---|---|---|---|
| 整合原始数据集 | 20 个 | — | 欧洲(highD/inD/rounD/CATS等)、中国(CitySim)、美国(NGSIM/Waymo/Argoverse) |
| 高风险事件 | 31,398 个 | DRF top-10% + TTC<5s + |a| 阈值 | 高速/城市快速路/交叉口/环岛全覆盖 |
| 极端近碰撞子集 | 1,036 个 | DRF top-1% + TTC<2s + 2秒冲突预测器 | 各类型路况平衡采样 |
风险等级的数学定义
风险等级通过复合评分函数 $S_{\text{risk}}$ 确定:
$$S_{\text{risk}} = \alpha \cdot R_{\text{DRF}} + \beta \cdot \mathbb{1}_{\text{TTC}<2s} + \gamma \cdot \frac{|a|}{a_{\text{max}}} + \delta \cdot \text{PET}$$其中 PET 是后侵占时间(Post-Encroachment Time),衡量多代理间的时空接近度。参数 $(\alpha, \beta, \gamma, \delta)$ 通过在 nuPlan 验证集上的碰撞率拟合确定。这种多指标融合的方式比单一 TTC 阈值更加鲁棒——比如 TTC 可能在低速场景下失效,而 DRF 仍能捕捉到空间逼近的风险。
这个规模远超大多数专门的事故数据集,且地理与路况多样性是单源数据集无法企及的——从德国高速的密集车流到北京快速路的复杂交互,K-Risk 提供了真正跨文化的风险数据基础。
🧠 LLM 标注:语言三元组
K-Risk 最有特色的设计是语言三元组(language triplet) 标注——每个事件都不只是一段轨迹,而是配齐了三层语义标注:
| 三元组层 | 内容 | 覆盖范围 |
|---|---|---|
| 结构化场景描述 | 路况、参与者、相对位置、运动状态 | 全部 31,398 事件 |
| 异常行为通知 | 哪个参与者做了什么异常行为 | 全部事件 |
| 因果风险分析 + 行动建议 | 为什么危险、该怎么应对 | 代表性子集 |
这个三元组的妙处在于:它把"发生了什么(描述)—哪里异常(通知)—为什么危险 + 怎么办(因果+建议)“这条完整的安全推理链显式地编码进了数据。模型不仅能从轨迹学"怎么动”,还能从语言学"为什么这么动"。
闭环验证:可验证的安全信号
光有 LLM 生成的标注还不够——LLM 会幻觉。K-Risk 的关键一手是:因果风险分析和行动建议,都通过闭环仿真器做了迭代反思(iterative reflection)验证。
| 验证环节 | 做法 | 解决的问题 |
|---|---|---|
| 闭环仿真 | 把建议的动作放进仿真器执行 | 检验建议是否真的避撞 |
| 迭代反思 | 根据仿真结果反复修正标注 | 剔除 LLM 幻觉、提升标注质量 |
这种"LLM 出标注 + 仿真器验真伪“的闭环,让 K-Risk 的语言标注不只是"听起来对”,而是"执行起来有效"——这是它区别于一切纯 LLM 生成数据集的根本。
与现有数据集的全面对比
| 维度 | NGSIM/HighD | Waymo/nuScenes | DriveLM/NuRisk | K-Risk (ours) |
|---|---|---|---|---|
| 事件数量 | 无风险标签 | 无风险标签 | ~1k-5k QA 对 | 31,398 事件 |
| 极端子集 | 无 | 无 | 极少 | 1,036 近碰撞 |
| 地理覆盖 | 单地(欧洲或美国) | 单地 | 多在美国 | 欧美中 3 大洲 20 源 |
| 路况类型 | 单一 | 城市为主 | 城市 | 高速+城市+环岛+交叉口 |
| 语义标注 | 无 | 无 | 有(QA级) | 三元组(描述+通知+因果) |
| 因果分析 | 无 | 无 | 弱 | 强(LLM+仿真验证) |
| 闭环可验证 | 无 | 无 | 无 | 有(碰撞免仿真+反思) |
| 下游任务支持 | 预测 | 感知+预测 | VQA | CPT+SFT+RLHF+RLVR |
这篇对比的核心结论是:K-Risk 在事件规模、地理多样性、语义深度、可验证性四个维度上同时做到了现有数据集未能企及的水平,真正填补了"风险中心 + 语义增强 + 可验证决策"这个三合一的空白。
统一风险抽取流水线
把 20 个格式各异的数据集统一成一件标准品,靠的是一条精心设计的 risk-centric extraction pipeline:
- 格式归一化:把不同数据集的轨迹、元数据统一到一致 schema;
- 风险事件筛选:基于运动学指标(TTC、THW、急动度等)+ 语义规则筛选高风险片段;
- 事件对齐:同步轨迹、元数据、语言三元组;
- LLM 标注生成:对每个事件生成三层语言标注;
- 闭环验证:对子集做仿真器验证 + 迭代反思。
这条流水线本身就是一份宝贵的工程资产——它意味着社区可以持续地把新数据"喂"进 K-Risk 框架,不断扩充高风险事件库。
📊 数据集特性与基准价值
多维风险标注
K-Risk 的"多维"体现在标注的层次上:
| 标注维度 | 提供的信息 | 用途 |
|---|---|---|
| 运动学 | 速度、加速度、TTC、THW | 定量风险建模 |
| 语义 | 场景类型、参与者角色 | 场景理解训练 |
| 因果 | 风险成因链 | 因果推理监督 |
| 决策 | 推荐行动 | 决策监督 |
| 语言 | 自然语言描述 | VLM/VLA 对齐 |
这种"轨迹 + 语言 + 决策“三位一体的标注,让 K-Risk 能同时支撑感知、预测、规划、推理多个下游任务,是一个真正的"多任务地基”。
可验证性
K-Risk 的决策标注是经仿真器验证的——这意味着用它训练出的智能体,其决策可以直接在同一个仿真器里做闭环评测,训练-评测的 gap 被最小化。这对风险感知智能体的研发至关重要:你训出来的"避撞决策",能在标准化的闭环环境里被公平地打分。
📈 实验结果与基准评测
基准协议
K-Risk 配套发布了四类下游任务的标准化评测协议:
| 任务类型 | 输入 | 输出 | 评测指标 |
|---|---|---|---|
| 风险分类 | 事件轨迹片段 | 风险等级(中/高/极端) | 准确率 + F1 |
| 因果推理 | 轨迹 + 三元组 | 风险成因文本 | BLEU + ROUGE + 人工 |
| 决策预测 | 轨迹 + 场景描述 | 推荐动作 ID | 动作准确率 |
| 闭环规划 | 完整事件 | 闭环轨迹 | 碰撞率 + 任务完成率 |
基线方法
K-Risk 在基准上评测了以下方法:
| 类别 | 基线 | 说明 |
|---|---|---|
| 纯运动学 | TTC/THW 阈值、DRF 分类器 | 无语义的传统方法 |
| 序列模型 | LSTM、Transformer 编码器 | 从轨迹学风险 |
| VLM 方法 | DriveLM、NuRisk 的 VLM | 用语言理解风险 |
| LLM Agent | GPT-4o 直接推理 | 纯语言风险分析 |
核心发现:纯运动学方法在常规风险事件上表现稳健(F1 > 0.85),但在极端近碰撞场景中严重退化(F1 < 0.6),而 VLM 方法在因果推理维度显著优于运动学方法(ROUGE-L 高出 15 点以上)。这验证了 K-Risk 的核心假设:高风险场景的"理解"需要语义+运动学联合建模,纯靠物理量是不够的。
📐 核心数学公式汇总
K-Risk 的风险筛选涉及多个数学指标,这里做统一整理:
Driver Risk Field (DRF)
DRF 衡量任意位置 $\mathbf{p}$ 在时间 $[0,T]$ 内受到的累积风险:
$$R_{\text{DRF}}(\mathbf{p}) = \sum_{i=1}^{N} \sum_{t=0}^{T/\Delta t} \frac{w_i \cdot v_i(t)}{2\pi\sigma_i^2} \exp\left(-\frac{\|\mathbf{p} - \mathbf{p}_i(t)\|^2}{2\sigma_i^2}\right) \Delta t$$其中 $\sigma_i = \sigma_0 + k \cdot v_i(t)$,即不确定性随速度 $v_i$ 线性增长。
TTC(碰撞时间)
对同车道前后两车,TTC 定义为:
$$TTC = \frac{d}{\max(v_{\text{rear}} - v_{\text{front}}, \epsilon)}$$其中 $d$ 为车距,$\epsilon$ 防止除零。K-Risk 用 TTC < 2s/3s/5s 分别标记极端/高/中风险。
THW(车头时距)
$$THW = \frac{d}{v_{\text{ego}}}$$THW < 1.0s 被视为跟车过近。
急动度(Jerk)
$$j = \frac{a_{t+1} - a_t}{\Delta t}$$$|j| > 4 m/s^3$ 的事件被标记为异常驾驶。
复合风险评分
最终的风险评分融合了运动学和场景因子:
$$S_{\text{final}} = \frac{1}{Z}\left(\frac{R_{\text{DRF}}}{\mu_R} + \frac{\mathbb{1}_{TTC<2s}}{\text{TTC}} + \frac{|j|}{j_{\text{max}}} + \frac{\text{PET}}{T_{\text{conflict}}}\right)^{\gamma}$$其中 $Z$ 是归一化常数,$\gamma$ 控制评分分布的陡峭程度。
⚔️ 在风险数据集谱系中的位置
| 代际 | 代表 | 标注层次 | 可验证性 |
|---|---|---|---|
| 第一代(纯轨迹) | NGSIM、HighD | 仅运动学 | 无 |
| 第二代(事故合集) | 各种 crash dataset | 碰撞标签 | 无(数据量小) |
| 第三代(语言增强) | DriveGPT 类数据 | 语言描述 | 弱(无闭环) |
| 本文(K-Risk) | 31,398 事件 + 三元组 | 运动学+语义+因果+决策+语言 | 强(闭环+反思) |
K-Risk 的开创性在于三点:第一次大规模整合跨洲际数据、第一次把语言三元组与轨迹对齐、第一次用闭环仿真器验证决策标注。这三个"第一次"共同定义了"风险感知自动驾驶数据集“的新标准。
⚠️ 局限性与挑战
K-Risk 虽然设计周到,仍有值得讨论的局限:
- LLM 标注的固有偏置:即便经过闭环验证,LLM 生成的语言描述仍可能带有训练数据的分布偏置,在罕见交通文化下不一定准确。
- 极端子集规模有限:1,036 个近碰撞事件虽宝贵,但相对于长尾的"无限性”,仍只是冰山一角。
- 闭环仿真器的真实性:验证所依赖的仿真器本身是否足够真实,决定了"验证"的可信度——这是所有仿真验证路线的共同软肋。
- 传感器模态:K-Risk 主要聚焦轨迹层面,对原始传感器(图像、点云)的覆盖深度未详尽说明,纯视觉方法的使用可能受限。
📝 个人思考
读 K-Risk,最打动我的是它对"数据集到底该提供什么“的清醒认识。过去自动驾驶数据集的竞赛几乎只在卷”小时数、英里数、场景数"——谁的体量大谁牛。但 K-Risk 点破了一层更深的认知:对于安全这个维度,体量不等于质量。一个有 31,398 个带语义、带因果、带可验证决策的高风险事件库,远比几百万小时平庸驾驶数据更有价值。安全的长尾不在于你开了多远,而在于你"理解"了多少种危险。K-Risk 用"风险中心 + 知识增强"把数据集的价值密度提到了一个新高度。
第二点启发在 LLM 标注 + 闭环验证这套范式。LLM 给数据集打标已经不是新鲜事,但纯 LLM 标注最大的问题是"幻觉"——它写得头头是道,执行起来可能完全错。K-Risk 的破题是:用闭环仿真器当"裁判",把 LLM 的建议放进仿真里跑,跑得通才留,跑不通就迭代修正。这种"生成-验证-修正“的闭环,是 LLM 标注可信化的关键范式。我预期这套思路会辐射到所有用 LLM 做数据增强的领域——没有验证闭环的 LLM 标注,本质上都是"自说自话”。
第三点是关于 “三元组"作为安全推理的训练目标。描述-通知-因果建议这三层,恰好对应了人类司机处理危险的认知流程:”看清情况→识别异常→判断因果→决定应对"。现有端到端模型大多只在学"轨迹→轨迹"的映射,完全跳过了中间的语义推理。K-Risk 的三元组标注,让模型有机会学到显式的安全推理链——这对可解释性、可审计性是巨大提升。想象一下,未来的自动驾驶系统不仅能"避撞",还能用自然语言告诉你"我为什么这么避撞"——这才是真正可信赖的安全智能体。
最后一点是 跨地域整合的战略价值。自动驾驶的"本地化"一直是难题——美国的训练数据拿到中国不一定好用,反之亦然。K-Risk 把欧美中三大洲的数据统一到一个框架下,为研究"跨法规、跨文化泛化“提供了独特土壤。这不仅是数据集的丰富,更是一个研究议题的开启:一个能在北京早高峰、德国高速、美国路口都安全驾驶的统一模型,到底需要什么样的训练数据和推理能力? K-Risk 给了社区一个去回答这个问题的标准化基础。
总体而言,K-Risk 是”数据集即基础设施“理念的典范。它不追求"刷榜”,而是踏踏实实地把社区最缺的三件套(事件标签 + 语义标注 + 可验证信号)补齐。这种"筑基“的工作虽然不如 SOTA 模型博眼球,但长期价值更高——没有好的数据集,再强的模型也是无源之水。
🔗 延伸阅读
| 工作 | 团队 | 与 K-Risk 的关系 |
|---|---|---|
| NGSIM / HighD / nuPlan | 各家 | K-Risk 整合的源数据集 |
| DriveGPT-4 / LMDrive | 各家 | 语言增强驾驶数据的前作 |
| TrafficBots | — | 学习式交通代理,可结合 K-Risk 训练 |
| Risk-aware Planning | 各家 | K-Risk 的核心下游应用 |
| Agent-driven Long-tail Simulation | 同期工作 | 可用 K-Risk 的事件作为仿真种子 |
| LLM-as-Annotator | 多家 | K-Risk 的标注范式来源 |
📖 这是论文精读系列的第 32 篇。当数据集开始承载"因果"和"可验证决策”,自动驾驶安全的研发范式正在被重塑。你认为语言三元组会成为驾驶数据集的新标配吗?欢迎留言讨论。