通过知识蒸馏融合多模态与模型快照,低成本提升3D语义分割性能。
Heterogeneous and Adept Snapshot Distillation for 3D Semantic Segmentation

- 用信息筛选策略选关键图像,增强多模态教师模型能力。
- 利用训练中自动生成的模型快照作专家,降低集成成本。
- 在ScanNetV2和S3DIS上达到当前最优,无额外推理开销。
多模态融合与多模型集成在提升3D语义分割性能方面广泛应用,但通常依赖辅助输入或带来高昂计算成本。为在不引入不可接受代价的前提下有效提升分割性能,我们提出通过知识蒸馏将多模态模型(点云与图像)及多个模型专家的知识迁移至基于点云的网络。具体地,提出信息导向的异构蒸馏(IHD),帮助单模态学生模型吸收多模态教师的互补知识;设计信息导向过滤(IOF)策略,从连续图像序列中选取有信息量的图像用于多模态融合,从而提升多模态教师性能,进而促进学生学习。此外,不同于需独立训练每个专家的传统集成方式,提出灵巧快照蒸馏(ASD),将训练过程中生成的免费模型快照作为多个专家,显著降低集成训练成本。对于每个专家教师,仅在其擅长类别上对学生提供监督。最终形成的异构且灵巧快照知识蒸馏(HAS-KD)在ScanNetV2和S3DIS数据集上达到当前最优结果,可无缝集成到现有3D分割算法中,带来显著增益且无额外推理负担。代码将在发表后公开。
原文摘要 · Abstract (English)
Multi-modal fusion and multi-model ensembling are prevalent in enhancing the performance of 3D semantic segmentation. Despite the impressive performance, these methods either rely on auxiliary input signals or suffer from costly computational expense. To efficaciously enhance the segmentation performance without introducing intolerable costs, we propose to transfer the rich knowledge from the multi-modal model (i.e., point clouds and images) and multiple model experts to the point-cloudbased network through knowledge distillation. Specifically, we present Information-oriented Heterogeneous Distillation (IHD) to help the uni-modal model absorb the complementary knowledge from the multi-modal teacher. We design the Information-Oriented Filtering (IOF) strategy to select informative images from the continuous image sequence for multi-modal fusion. This practice can boost the performance of the multi-modal teacher, thus benefiting the learning of the student. Besides, as opposed to vanilla model ensembling that requires the separate training of each expert, we propose Adept Snapshot Distillation (ASD). ASD treats the freely available model snapshots generated during the training phase as multiple experts, which significantly reduces the training cost for model ensembling. For each expert teacher, it only provides supervision to the student in the class where it is adept. The resulting Heterogeneous and Adept Snapshot Knowledge Distillation, dubbed HAS-KD, attains state-of-the-art results on ScanNetV2 and S3DIS datasets. HAS-KD can be seamlessly integrated into contemporary 3D segmentation algorithms and bring considerable gains without introducing extra inference burdens. The code will be made publicly available upon publication.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。