May, 2023

使用 Minkowski 引擎和知识蒸馏方法的 3D 语义分割更小模型

TL;DR通过应用知识蒸馏技术来减少模型大小,同时保持性能,本文目的是分析和提出不同的损失函数来模拟不同的稀疏卷积神经网络 (Sparse Convolutional NNs) 的表现,特别是在 3D 深度学习的稀疏张量中,我们在标准 ScanNet V2 数据集上进行了实验,并在最新的基于时空卷积模型上,通过四倍模型减少约 2.6%mIoU 差异和 16 倍模型减少约 8%的实验结果。