HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Sonata: Self-Supervised Learning of Reliable Point Representations

Xiaoyang Wu Daniel DeTone Duncan Frost Tianwei Shen Chris Xie Nan Yang Jakob Engel Richard Newcombe Hengshuang Zhao Julian Straub

Sonata: Self-Supervised Learning of Reliable Point Representations

Abstract

In this paper, we question whether we have a reliable self-supervised pointcloud model that can be used for diverse 3D tasks via simple linear probing,even with limited data and minimal computation. We find that existing 3Dself-supervised learning approaches fall short when evaluated on representationquality through linear probing. We hypothesize that this is due to what we termthe "geometric shortcut", which causes representations to collapse to low-levelspatial features. This challenge is unique to 3D and arises from the sparsenature of point cloud data. We address it through two key strategies: obscuringspatial information and enhancing the reliance on input features, ultimatelycomposing a Sonata of 140k point clouds through self-distillation. Sonata issimple and intuitive, yet its learned representations are strong and reliable:zero-shot visualizations demonstrate semantic grouping, alongside strongspatial reasoning through nearest-neighbor relationships. Sonata demonstratesexceptional parameter and data efficiency, tripling linear probing accuracy(from 21.8% to 72.5%) on ScanNet and nearly doubling performance with only 1%of the data compared to previous approaches. Full fine-tuning further advancesSOTA across both 3D indoor and outdoor perception tasks.

Code Repositories

Pointcept/Pointcept
Official
pytorch
Mentioned in GitHub
facebookresearch/sonata
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
3d-semantic-segmentation-on-scannet200Sonata + PTv3
val mIoU: 36.8
semantic-segmentation-on-s3disSonata + PTv3
Mean IoU: 82.3
Number of params: 128M
mAcc: 89.9
oAcc: 93.3
semantic-segmentation-on-s3dis-area5Sonata + PTv3
mAcc: 81.6
mIoU: 76.0
oAcc: 93.0
semantic-segmentation-on-scannetSonata + PTv3
val mIoU: 79.4

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