用图神经网络自动聚类交通场景,提升自动驾驶测试效率。
Exploring Semantic Clustering and Similarity Search for Heterogeneous Traffic Scenario Graph
- 构建时空异构图表示场景,通过自监督学习生成通用嵌入向量。
- 在nuPlan数据集上实现无标签场景聚类,不同场景类型形成独立簇。
- 无需人工标注或关键性筛选,适合大规模自动驾驶仿真测试。
基于场景的测试是自动驾驶车辆全面验证不可或缺的工具。然而,在可扩展、可能无监督的情况下,找到一个管理可控且具有代表性的场景子集极具挑战性。本文提出一种表达性强、灵活的异构时空图模型来表示交通场景,并利用图神经网络设计一种自监督方法,学习统一的场景图嵌入空间,实现聚类与相似性搜索。特别地,采用对比学习结合自举法评估其对场景空间划分的适用性。在nuPlan数据集上的实验表明,该模型能有效捕捉语义信息,尽管无离散类别标签,仍可有意义地分组相关场景,不同场景类型自然形成独立簇。结果证明,可将变长交通场景压缩为单一向量表示,实现邻近检索代表性候选场景。此过程无需人工标注或偏向特定目标(如关键性)。最终,该方法可作为可扩展场景选择的基础,进一步提升自动驾驶仿真测试的效率与鲁棒性。
原文摘要 · Abstract (English)
Scenario-based testing is an indispensable instrument for the comprehensive validation and verification of automated vehicles (AVs). However, finding a manageable and finite, yet representative subset of scenarios in a scalable, possibly unsupervised manner is notoriously challenging. Our work is meant to constitute a cornerstone to facilitate sample-efficient testing, while still capturing the diversity of relevant operational design domains (ODDs) and accounting for the "long tail" phenomenon in particular. To this end, we first propose an expressive and flexible heterogeneous, spatio-temporal graph model for representing traffic scenarios. Leveraging recent advances of graph neural networks (GNNs), we then propose a self-supervised method to learn a universal embedding space for scenario graphs that enables clustering and similarity search. In particular, we implement contrastive learning alongside a bootstrapping-based approach and evaluate their suitability for partitioning the scenario space. Experiments on the nuPlan dataset confirm the model's ability to capture semantics and thus group related scenarios in a meaningful way despite the absence of discrete class labels. Different scenario types materialize as distinct clusters. Our results demonstrate how variable-length traffic scenarios can be condensed into single vector representations that enable nearest-neighbor retrieval of representative candidates for distinct scenario categories. Notably, this is achieved without manual labeling or bias towards an explicit objective such as criticality. Ultimately, our approach can serve as a basis for scalable selection of scenarios to further enhance the efficiency and robustness of testing AVs in simulation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。