Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

DOS: Distilling Observable Softmaps of Zipfian Prototypes for Self-Supervised Point Representation

About

Recent advances in self-supervised learning (SSL) have shown tremendous potential for learning 3D point cloud representations without human annotations. However, SSL for 3D point clouds still faces critical challenges due to irregular geometry, shortcut-prone reconstruction, and unbalanced semantics distribution. In this work, we propose DOS (Distilling Observable Softmaps), a novel SSL framework that self-distills semantic relevance softmaps only at observable (unmasked) points. This strategy prevents information leakage from masked regions and provides richer supervision than discrete token-to-prototype assignments. To address the challenge of unbalanced semantics in an unsupervised setting, we introduce Zipfian prototypes and incorporate them using a modified Sinkhorn-Knopp algorithm, Zipf-Sinkhorn, which enforces a power-law prior over prototype usage and modulates the sharpness of the target softmap during training. DOS outperforms current state-of-the-art methods on semantic segmentation and 3D object detection across multiple benchmarks, including nuScenes, Waymo, SemanticKITTI, ScanNet, and ScanNet200, without relying on extra data or annotations. Our results demonstrate that observable-point softmaps distillation offers a scalable and effective paradigm for learning robust 3D representations.

Mohamed Abdelsamad, Michael Ulrich, Bin Yang, Miao Zhang, Yakov Miron, Abhinav Valada• 2025

Related benchmarks

TaskDatasetResultRank
Semantic segmentationS3DIS (Area 5)
mIOU75.1
799
Semantic segmentationScanNet (val)
mIoU79.7
231
Semantic segmentationnuScenes (val)
mIoU (Segmentation)0.818
212
Semantic segmentationSemanticKITTI (val)
mIoU73.5
117
Semantic segmentationScanNet200 (val)
mIoU37.1
74
Semantic segmentationWaymo Open Dataset (val)
mIoU73.9
63
Object DetectionnuScenes (val)
mAP65.5
41
3D Semantic SegmentationParisLuco (test)
nuScenes Score51.5
3
Showing 8 of 8 rows

Other info

Follow for update