HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

Point2Vec for Self-Supervised Representation Learning on Point Clouds

Zeid Karim Abou ; Schult Jonas ; Hermans Alexander ; Leibe Bastian

Point2Vec for Self-Supervised Representation Learning on Point Clouds

Abstract

Recently, the self-supervised learning framework data2vec has shown inspiringperformance for various modalities using a masked student-teacher approach.However, it remains open whether such a framework generalizes to the uniquechallenges of 3D point clouds. To answer this question, we extend data2vec tothe point cloud domain and report encouraging results on several downstreamtasks. In an in-depth analysis, we discover that the leakage of positionalinformation reveals the overall object shape to the student even under heavymasking and thus hampers data2vec to learn strong representations for pointclouds. We address this 3D-specific shortcoming by proposing point2vec, whichunleashes the full potential of data2vec-like pre-training on point clouds. Ourexperiments show that point2vec outperforms other self-supervised methods onshape classification and few-shot learning on ModelNet40 and ScanObjectNN,while achieving competitive results on part segmentation on ShapeNetParts.These results suggest that the learned representations are strong andtransferable, highlighting point2vec as a promising direction forself-supervised learning of point cloud representations.

Code Repositories

kabouzeid/point2vec
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
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

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp