arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2304.11393cs.CVcs.AI

用于激光雷达语义分割的从3D到鸟瞰视图的知识蒸馏

Knowledge Distillation from 3D to Bird's-Eye-View for LiDAR Semantic Segmentation

Fudan University(复旦大学);Mogo Auto(魔门塔汽车)

Feng Jiang, Heng Gao, Shoumeng Qiu, Haiqiang Zhang, Ru Wan, Jian Pu

AI总结:

提出3D到BEV知识蒸馏框架,通过体素到柱体和标签权重蒸馏,将3D体素模型知识迁移给实时BEV模型,在SemanticKITTI上显著提升分割精度。

摘要:

激光雷达点云分割是自动驾驶场景理解中最基础的任务之一。然而,现有模型难以同时实现高推理速度和高精度。例如,基于体素的方法在精度上表现良好,而基于鸟瞰视图(BEV)的方法能够实现实时推理。为克服这一问题,我们开发了一种有效的从3D到BEV的知识蒸馏方法,将丰富的知识从基于3D体素的模型迁移到基于BEV的模型。我们的框架主要由两个模块组成:体素到柱体蒸馏模块和标签权重蒸馏模块。体素到柱体蒸馏将稀疏3D特征蒸馏到中间层的BEV特征,使基于BEV的模型感知更多结构和几何信息。标签权重蒸馏帮助模型更加关注具有更多高度信息的区域。最后,我们在SemanticKITTI数据集和Paris-Lille-3D上进行了实验。SemanticKITTI上的结果显示测试集提升超过5%,尤其是摩托车和人等类别,提升超过15%。代码可在https://github.com/fengjiang5/Knowledge-Distillation-from-Cylinder3D-to-PolarNet获取。

英文摘要:

LiDAR point cloud segmentation is one of the most fundamental tasks for autonomous driving scene understanding. However, it is difficult for existing models to achieve both high inference speed and accuracy simultaneously. For example, voxel-based methods perform well in accuracy, while Bird's-Eye-View (BEV)-based methods can achieve real-time inference. To overcome this issue, we develop an effective 3D-to-BEV knowledge distillation method that transfers rich knowledge from 3D voxel-based models to BEV-based models. Our framework mainly consists of two modules: the voxel-to-pillar distillation module and the label-weight distillation module. Voxel-to-pillar distillation distills sparse 3D features to BEV features for middle layers to make the BEV-based model aware of more structural and geometric information. Label-weight distillation helps the model pay more attention to regions with more height information. Finally, we conduct experiments on the SemanticKITTI dataset and Paris-Lille-3D. The results on SemanticKITTI show more than 5% improvement on the test set, especially for classes such as motorcycle and person, with more than 15% improvement. The code can be accessed at https://github.com/fengjiang5/Knowledge-Distillation-from-Cylinder3D-to-PolarNet.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑