FastMap加速高精地图重建,用单层双阶段结构提升效率。
FastMap: Fast Queries Initialization Based Vectorized HD Map Reconstruction Framework
- 采用单层双阶段Transformer减少解码器冗余
- 查询生成引入热力图引导,提速3.2倍且精度领先
- 适合需要高效高精地图的自动驾驶系统
高精地图重建是感知自动驾驶环境的关键任务,其精度直接影响下游预测与规划的可靠性。现有基于DETR框架的矢量地图重建方法因解码器结构冗余,需堆叠六层解码器以维持性能,严重制约计算效率。为此,我们提出FastMap,一种创新框架,通过单层双阶段Transformer优化解码器结构,实现多层次表征能力。该框架摒弃随机初始化查询,改用解码阶段的热力图引导查询生成模块,利用可学习位置编码将图像特征映射为结构化查询向量。此外,提出几何约束点到线损失机制,有效解决传统点对点损失在高度同质特征下的区分难题。大量实验表明,FastMap在nuScenes和Argoverse2数据集上均达到最先进性能,解码器速度比基线快3.2倍。代码与演示详见https://github.com/hht1996ok/FastMap。
原文摘要 · Abstract (English)
Reconstruction of high-definition maps is a crucial task in perceiving the autonomous driving environment, as its accuracy directly impacts the reliability of prediction and planning capabilities in downstream modules. Current vectorized map reconstruction methods based on the DETR framework encounter limitations due to the redundancy in the decoder structure, necessitating the stacking of six decoder layers to maintain performance, which significantly hampers computational efficiency. To tackle this issue, we introduce FastMap, an innovative framework designed to reduce decoder redundancy in existing approaches. FastMap optimizes the decoder architecture by employing a single-layer, two-stage transformer that achieves multilevel representation capabilities. Our framework eliminates the conventional practice of randomly initializing queries and instead incorporates a heatmap-guided query generation module during the decoding phase, which effectively maps image features into structured query vectors using learnable positional encoding. Additionally, we propose a geometry-constrained point-to-line loss mechanism for FastMap, which adeptly addresses the challenge of distinguishing highly homogeneous features that often arise in traditional point-to-point loss computations. Extensive experiments demonstrate that FastMap achieves state-of-the-art performance in both nuScenes and Argoverse2 datasets, with its decoder operating 3.2 faster than the baseline. Code and more demos are available at https://github.com/hht1996ok/FastMap.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。