arXiv:2409.00633cs.CV2024-09ECCV被引 10

通过压缩视觉令牌提升ViT多视角3D检测速度,兼顾精度与实时性。

Make Your ViT-based Multi-view 3D Detectors Faster via Token Compression

论文配图:Make Your ViT-based Multi-view 3D Detectors Faster via Token Compression
图 1 · 摘自论文原文
  • 利用历史查询作为前景先验,动态评估图像令牌重要性。
  • 引入动态路由机制,对关键前景令牌分配更多计算资源。
  • 在nuScenes上实现30%提速,且模型越大效果越明显。

多视角3D检测器的推理速度慢是自动驾驶等高实时性任务部署的主要瓶颈。尽管已有稀疏查询方法提升效率,但大多忽视了骨干网络,尤其在采用性能更优的视觉变换器(ViT)时。为此,本文提出一种简单有效的令牌压缩方法TokenCompression3D(ToC3D),通过将历史目标查询作为高质量前景先验,建模其3D运动信息,并与图像令牌通过注意力机制交互,从而有效判断图像令牌的信息密度并分割出显著前景令牌。结合动态路由设计,ToC3D可将更多计算资源集中于重要前景令牌,同时最小化信息损失,显著提升基于ViT的多视角3D检测效率。在大规模nuScenes数据集上的大量实验表明,该方法可在近乎保持最新最优性能的前提下实现最高30%的推理加速,且在扩大ViT规模和输入分辨率时仍保持一致增益。代码将公开于https://github.com/DYZhang09/ToC3D。

原文摘要 · Abstract (English)

Slow inference speed is one of the most crucial concerns for deploying multi-view 3D detectors to tasks with high real-time requirements like autonomous driving. Although many sparse query-based methods have already attempted to improve the efficiency of 3D detectors, they neglect to consider the backbone, especially when using Vision Transformers (ViT) for better performance. To tackle this problem, we explore the efficient ViT backbones for multi-view 3D detection via token compression and propose a simple yet effective method called TokenCompression3D (ToC3D). By leveraging history object queries as foreground priors of high quality, modeling 3D motion information in them, and interacting them with image tokens through the attention mechanism, ToC3D can effectively determine the magnitude of information densities of image tokens and segment the salient foreground tokens. With the introduced dynamic router design, ToC3D can weigh more computing resources to important foreground tokens while compressing the information loss, leading to a more efficient ViT-based multi-view 3D detector. Extensive results on the large-scale nuScenes dataset show that our method can nearly maintain the performance of recent SOTA with up to 30% inference speedup, and the improvements are consistent after scaling up the ViT and input resolution. The code will be made at https://github.com/DYZhang09/ToC3D.

3D检测ViT令牌压缩加速

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