用稀疏集合预测代替密集网格,提升自动驾驶3D占位预测效率
OPUS: Occupancy Prediction Using a Sparse Set

- 将占位预测转为可学习查询集的集合预测,避免显式建模空间
- 轻量模型在nuScenes上达近2倍帧率,重模型比前人高6.1点RayIoU
- 适合追求实时性与高精度的自动驾驶场景感知系统
占位预测旨在预测体素化3D环境中的占据状态,在自动驾驶领域迅速兴起。主流方法先将3D环境离散为体素网格,再对密集网格进行分类,但大量体素为空,导致计算资源浪费。为此,本文提出新视角:将占位预测重构为无需显式空间建模或复杂稀疏化处理的集合预测范式。所提框架OPUS采用Transformer编码器-解码器结构,通过可学习查询集同时预测占据位置与类别。首先引入切比雪夫距离损失,使集合到集合的比较可扩展至前所未有的规模,实现端到端训练。随后基于学习到的位置,利用最近邻搜索自适应分配语义类别。此外,OPUS还融合了粗到细学习、一致采样与自适应重加权等策略。相比现有最先进方法,其最轻模型在Occ3D-nuScenes数据集上达到近2倍帧率的优越RayIoU,最重模型更是超越此前最佳结果6.1 RayIoU。
原文摘要 · Abstract (English)
Occupancy prediction, aiming at predicting the occupancy status within voxelized 3D environment, is quickly gaining momentum within the autonomous driving community. Mainstream occupancy prediction works first discretize the 3D environment into voxels, then perform classification on such dense grids. However, inspection on sample data reveals that the vast majority of voxels is unoccupied. Performing classification on these empty voxels demands suboptimal computation resource allocation, and reducing such empty voxels necessitates complex algorithm designs. To this end, we present a novel perspective on the occupancy prediction task: formulating it as a streamlined set prediction paradigm without the need for explicit space modeling or complex sparsification procedures. Our proposed framework, called OPUS, utilizes a transformer encoder-decoder architecture to simultaneously predict occupied locations and classes using a set of learnable queries. Firstly, we employ the Chamfer distance loss to scale the set-to-set comparison problem to unprecedented magnitudes, making training such model end-to-end a reality. Subsequently, semantic classes are adaptively assigned using nearest neighbor search based on the learned locations. In addition, OPUS incorporates a suite of non-trivial strategies to enhance model performance, including coarse-to-fine learning, consistent point sampling, and adaptive re-weighting, etc. Finally, compared with current state-of-the-art methods, our lightest model achieves superior RayIoU on the Occ3D-nuScenes dataset at near 2x FPS, while our heaviest model surpasses previous best results by 6.1 RayIoU.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。