用扩散模型生成地图并估算不确定性,提升自动驾驶地图构建的可靠性。
MapDiffusion: Generative Diffusion for Vectorized Online HD Map Construction and Uncertainty Estimation in Autonomous Driving
- 基于扩散模型迭代优化初始查询,生成多组可能的地图样本。
- 单样本性能比基线高5%,多样本融合进一步提升ROC曲线表现。
- 不确定性估计能有效识别遮挡区域,适合需要安全决策的自动驾驶系统。
自动驾驶需从传感器数据中理解静态环境。现有方法通常使用学习的鸟瞰图(BEV)编码器融合多源输入,并通过向量解码器从潜在BEV网格生成矢量化地图表示。然而,传统地图构建模型仅输出确定性点估计,无法捕捉真实世界中的不确定性与模糊性(如遮挡、缺失车道线)。本文提出MapDiffusion,一种新型生成式方法,利用扩散范式学习可能地图的完整分布。不同于直接预测单一确定性输出,MapDiffusion在给定BEV潜空间的基础上,迭代优化随机初始化的查询,生成多个合理地图样本。该方法可通过样本聚合提升预测精度,并推导出与场景模糊性直接相关的不确定性估计。在nuScenes数据集上的大量实验表明,MapDiffusion在在线地图构建任务中达到领先性能,单样本表现优于基线5%。进一步验证,多样本聚合可稳定提升ROC曲线上性能,证明分布建模的价值。此外,不确定性估计在遮挡区域显著升高,证实其对传感器输入模糊性的敏感性。通过建模完整地图分布,MapDiffusion增强了在线矢量化高精地图构建的鲁棒性与可靠性,支持自动驾驶车辆在复杂环境中进行不确定性感知决策。
原文摘要 · Abstract (English)
Autonomous driving requires an understanding of the static environment from sensor data. Learned Bird's-Eye View (BEV) encoders are commonly used to fuse multiple inputs, and a vector decoder predicts a vectorized map representation from the latent BEV grid. However, traditional map construction models provide deterministic point estimates, failing to capture uncertainty and the inherent ambiguities of real-world environments, such as occlusions and missing lane markings. We propose MapDiffusion, a novel generative approach that leverages the diffusion paradigm to learn the full distribution of possible vectorized maps. Instead of predicting a single deterministic output from learned queries, MapDiffusion iteratively refines randomly initialized queries, conditioned on a BEV latent grid, to generate multiple plausible map samples. This allows aggregating samples to improve prediction accuracy and deriving uncertainty estimates that directly correlate with scene ambiguity. Extensive experiments on the nuScenes dataset demonstrate that MapDiffusion achieves state-of-the-art performance in online map construction, surpassing the baseline by 5% in single-sample performance. We further show that aggregating multiple samples consistently improves performance along the ROC curve, validating the benefit of distribution modeling. Additionally, our uncertainty estimates are significantly higher in occluded areas, reinforcing their value in identifying regions with ambiguous sensor input. By modeling the full map distribution, MapDiffusion enhances the robustness and reliability of online vectorized HD map construction, enabling uncertainty-aware decision-making for autonomous vehicles in complex environments.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。