用双分支编码+原型查询,3D占位预测又快又准
ProtoOcc: Accurate, Efficient 3D Occupancy Prediction Using Dual Branch Encoder-Prototype Query Decoder

- 双分支编码融合体素与鸟瞰图特征,提升感知范围与效率
- 单步解码实现45.02% mIoU,单帧推理达12.83 FPS
- 原型噪声学习增强鲁棒性,适合实时3D场景理解任务
本文提出ProtoOcc,一种用于3D占位预测的新模型,通过深度语义理解场景,预测3D体素的占用状态与语义类别。模型由双分支编码器(DBE)和原型查询解码器(PQD)组成。DBE通过双分支结构在多尺度上融合3D体素与鸟瞰图(BEV)表示,提升BEV感受野的同时保持体素小感受野,兼顾性能与效率。PQD引入原型查询加速解码:场景自适应原型从输入样本的3D体素特征中提取,场景无关原型通过训练阶段对场景自适应原型进行指数移动平均得到。利用这些原型查询可一步直接预测3D占位,无需迭代式Transformer解码。此外,提出鲁棒原型学习,在原型生成中注入噪声并训练模型去噪。ProtoOcc在Occ3D-nuScenes基准上达到45.02% mIoU;作为单帧方法,推理速度达12.83 FPS(NVIDIA RTX 3090)。代码已开源。
原文摘要 · Abstract (English)
In this paper, we introduce ProtoOcc, a novel 3D occupancy prediction model designed to predict the occupancy states and semantic classes of 3D voxels through a deep semantic understanding of scenes. ProtoOcc consists of two main components: the Dual Branch Encoder (DBE) and the Prototype Query Decoder (PQD). The DBE produces a new 3D voxel representation by combining 3D voxel and BEV representations across multiple scales through a dual branch structure. This design enhances both performance and computational efficiency by providing a large receptive field for the BEV representation while maintaining a smaller receptive field for the voxel representation. The PQD introduces Prototype Queries to accelerate the decoding process. Scene-Adaptive Prototypes are derived from the 3D voxel features of input sample, while Scene-Agnostic Prototypes are computed by applying Scene-Adaptive Prototypes to an Exponential Moving Average during the training phase. By using these prototype-based queries for decoding, we can directly predict 3D occupancy in a single step, eliminating the need for iterative Transformer decoding. Additionally, we propose the Robust Prototype Learning, which injects noise into prototype generation process and trains the model to denoise during the training phase. ProtoOcc achieves state-of-the-art performance with 45.02% mIoU on the Occ3D-nuScenes benchmark. For single-frame method, it reaches 39.56% mIoU with an inference speed of 12.83 FPS on an NVIDIA RTX 3090. Our code can be found at https://github.com/SPA-junghokim/ProtoOcc.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。