4 个月前

Point2Vec 用于点云的自监督表示学习

Point2Vec 用于点云的自监督表示学习

摘要

近日,自监督学习框架data2vec在使用掩码学生-教师方法的各种模态上展示了令人鼓舞的性能。然而,该框架是否适用于3D点云的独特挑战仍是一个开放问题。为了解答这一问题,我们将data2vec扩展到点云领域,并在多个下游任务中报告了令人鼓舞的结果。通过深入分析,我们发现位置信息的泄露即使在重度掩码下也能向学生模型揭示整体物体形状,从而阻碍data2vec学习强大的点云表示。为解决这一3D特有的不足,我们提出了point2vec,该方法释放了类似data2vec的预训练在点云上的全部潜力。实验结果表明,point2vec在ModelNet40和ScanObjectNN上的形状分类和少样本学习任务中优于其他自监督方法,并在ShapeNetParts上的部件分割任务中取得了具有竞争力的结果。这些结果表明所学的表示既强大又可迁移,突显了point2vec作为点云表示自监督学习的一个有前景的方向。

代码仓库

kabouzeid/point2vec
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
3d-part-segmentation-on-shapenet-partpoint2vec
Class Average IoU: 84.6
Instance Average IoU: 86.3
3d-point-cloud-classification-on-modelnet40point2vec
Mean Accuracy: 92.0
Overall Accuracy: 94.8
3d-point-cloud-classification-on-scanobjectnnpoint2vec
Mean Accuracy: 86.0
OBJ-BG (OA): 91.2
OBJ-ONLY (OA): 90.4
Overall Accuracy: 87.5
few-shot-3d-point-cloud-classification-on-1point2vec
Overall Accuracy: 97.0
Standard Deviation: 2.8
few-shot-3d-point-cloud-classification-on-2point2vec
Overall Accuracy: 98.7
Standard Deviation: 1.2
few-shot-3d-point-cloud-classification-on-3point2vec
Overall Accuracy: 93.9
Standard Deviation: 4.1
few-shot-3d-point-cloud-classification-on-4point2vec
Overall Accuracy: 95.8
Standard Deviation: 3.1

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
Point2Vec 用于点云的自监督表示学习 | 论文 | HyperAI超神经