arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.02007cs.CV

PMT-MAE:面向高效点云分类的带蒸馏双分支自监督学习

PMT-MAE: Dual-Branch Self-Supervised Learning with Distillation for Efficient Point Cloud Classification

Qiang Zheng, Chao Zhang, Jian Sun

AI总结:

本文提出PMT-MAE双分支自监督点云分类框架,融合Transformer与MLP分支特征,结合Point-M2AE指导的蒸馏策略,在ModelNet40上达93.6%准确率且训练效率高,适配资源有限场景。

摘要:

自监督学习的进展对提升点云处理中的特征提取与理解能力至关重要。本文提出了PMT-MAE(Point MLP-Transformer Masked Autoencoder,点云MLP-Transformer掩码自编码器),这是一种用于点云分类的新型自监督学习框架。PMT-MAE采用双分支架构,融合了Transformer与MLP组件以捕获丰富特征。Transformer分支利用全局自注意力实现复杂的特征交互,而并行的MLP分支通过共享全连接层处理token,提供互补的特征变换路径。随后的融合机制将这些特征结合,增强了模型学习全面3D表征的能力。在成熟的教师模型Point-M2AE的指导下,PMT-MAE采用了包含预训练阶段特征蒸馏和微调阶段logit蒸馏的蒸馏策略,确保有效的知识迁移。在ModelNet40分类任务上,PMT-MAE未使用投票策略便达到了93.6%的准确率,超过了基线模型Point-MAE(93.2%)和教师模型Point-M2AE(93.4%),凸显了其学习具有判别性的3D点云表征的能力。此外,该框架展现出高效率,预训练和微调均仅需40个epoch。PMT-MAE的有效性与高效性使其非常适合计算资源有限的场景,是实用点云分析领域极具潜力的解决方案。

英文摘要:

Advances in self-supervised learning are essential for enhancing feature extraction and understanding in point cloud processing. This paper introduces PMT-MAE (Point MLP-Transformer Masked Autoencoder), a novel self-supervised learning framework for point cloud classification. PMT-MAE features a dual-branch architecture that integrates Transformer and MLP components to capture rich features. The Transformer branch leverages global self-attention for intricate feature interactions, while the parallel MLP branch processes tokens through shared fully connected layers, offering a complementary feature transformation pathway. A fusion mechanism then combines these features, enhancing the model's capacity to learn comprehensive 3D representations. Guided by the sophisticated teacher model Point-M2AE, PMT-MAE employs a distillation strategy that includes feature distillation during pre-training and logit distillation during fine-tuning, ensuring effective knowledge transfer. On the ModelNet40 classification task, achieving an accuracy of 93.6\% without employing voting strategy, PMT-MAE surpasses the baseline Point-MAE (93.2\%) and the teacher Point-M2AE (93.4\%), underscoring its ability to learn discriminative 3D point cloud representations. Additionally, this framework demonstrates high efficiency, requiring only 40 epochs for both pre-training and fine-tuning. PMT-MAE's effectiveness and efficiency render it well-suited for scenarios with limited computational resources, positioning it as a promising solution for practical point cloud analysis.

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

↑