FALO让激光雷达3D目标检测在低功耗设备上又快又准。
FALO: Fast and Accurate LiDAR 3D Object Detection on Resource-Constrained Devices
- 将点云转为一维序列,用大卷积核和哈达玛积混合特征
- 在nuScenes和Waymo上精度接近顶尖,移动端推理快1.6~9.8倍
- 适合部署在手机GPU或嵌入式NPU等资源受限设备
现有激光雷达3D目标检测方法主要依赖稀疏卷积和/或变换器,因不规则内存访问和高计算开销,在资源受限的边缘设备上难以运行。本文提出FALO,一种面向硬件友好的激光雷达3D检测方法,兼具顶尖检测精度与快速推理速度。给定3D点云并体素化后,FALO首先根据坐标和邻近性将稀疏体素排列成一维序列,再通过提出的ConvDotMix模块处理,该模块包含大卷积核、哈达玛积和线性层。ConvDotMix在空间和嵌入维度均具备充分的特征混合能力,并引入高阶非线性交互。同时,通过隐式分组平衡张量维度,兼顾不断增长的感受野,提升推理效率。所有操作均适配资源受限平台,可直接部署于小型嵌入式设备。在nuScenes和Waymo等激光雷达3D检测基准上的大量评估表明,FALO性能具有竞争力,且在移动GPU和移动NPU上比最新SOTA快1.6~9.8倍。
原文摘要 · Abstract (English)
Existing LiDAR 3D object detection methods predominantely rely on sparse convolutions and/or transformers, which can be challenging to run on resource-constrained edge devices, due to irregular memory access patterns and high computational costs. In this paper, we propose FALO, a hardware-friendly approach to LiDAR 3D detection, which offers both state-of-the-art (SOTA) detection accuracy and fast inference speed. More specifically, given the 3D point cloud and after voxelization, FALO first arranges sparse 3D voxels into a 1D sequence based on their coordinates and proximity. The sequence is then processed by our proposed ConvDotMix blocks, consisting of large-kernel convolutions, Hadamard products, and linear layers. ConvDotMix provides sufficient mixing capability in both spatial and embedding dimensions, and introduces higher-order nonlinear interaction among spatial features. Furthermore, when going through the ConvDotMix layers, we introduce implicit grouping, which balances the tensor dimensions for more efficient inference and takes into account the growing receptive field. All these operations are friendly to run on resource-constrained platforms and proposed FALO can readily deploy on compact, embedded devices. Our extensive evaluation on LiDAR 3D detection benchmarks such as nuScenes and Waymo shows that FALO achieves competitive performance. Meanwhile, FALO is 1.6~9.8x faster than the latest SOTA on mobile Graphics Processing Unit (GPU) and mobile Neural Processing Unit (NPU).
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。