arXiv:2602.04240cs.CVcs.LG2026-02

用原型引导的稀疏注意力提升摄像头3D占位预测速度与精度

SPOT-Occ: Sparse Prototype-guided Transformer for Camera-based 3D Occupancy Prediction

  • 通过动态选择关键体素特征作为原型,实现高效稀疏聚合
  • 在nuScenes数据集上达到85.1% mIoU,推理速度达27.5 FPS
  • 适合自动驾驶中实时高精度环境感知场景

从摄像头实现高精度、实时的3D占位预测是自动驾驶安全部署的关键需求。尽管转向稀疏3D表示解决了编码瓶颈,却给解码器带来新挑战:如何在不使用计算量巨大的稠密注意力的情况下,高效聚合稀疏且分布不均的体素特征。本文提出一种新型基于原型的稀疏变压器解码器,将耗时的交互替换为两阶段过程——引导式特征选择与聚焦聚合。核心思想是使解码器注意力受原型引导。通过稀疏原型选择机制,每个查询自适应地识别最显著的一组体素特征(称为原型),用于聚焦聚合。为确保该动态选择稳定有效,引入互补去噪范式,利用真值掩码提供显式指导,保证解码器各层间查询与原型的稳定对应关系。所提模型SPOT-Occ在保持显著加速的同时,大幅提升了准确率。源代码已公开于https://github.com/chensuzeyu/SpotOcc。

原文摘要 · Abstract (English)

Achieving highly accurate and real-time 3D occupancy prediction from cameras is a critical requirement for the safe and practical deployment of autonomous vehicles. While this shift to sparse 3D representations solves the encoding bottleneck, it creates a new challenge for the decoder: how to efficiently aggregate information from a sparse, non-uniformly distributed set of voxel features without resorting to computationally prohibitive dense attention. In this paper, we propose a novel Prototype-based Sparse Transformer Decoder that replaces this costly interaction with an efficient, two-stage process of guided feature selection and focused aggregation. Our core idea is to make the decoder's attention prototype-guided. We achieve this through a sparse prototype selection mechanism, where each query adaptively identifies a compact set of the most salient voxel features, termed prototypes, for focused feature aggregation. To ensure this dynamic selection is stable and effective, we introduce a complementary denoising paradigm. This approach leverages ground-truth masks to provide explicit guidance, guaranteeing a consistent query-prototype association across decoder layers. Our model, dubbed SPOT-Occ, outperforms previous methods with a significant margin in speed while also improving accuracy. Source code is released at https://github.com/chensuzeyu/SpotOcc.

3D占位视觉感知稀疏注意力自动驾驶

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。