Towards More Diverse and Challenging Pre-training for Point Cloud Learning: Self-Supervised Cross Reconstruction with Decoupled Views
About
Point cloud learning, especially in a self-supervised way without manual labels, has gained growing attention in both vision and learning communities due to its potential utility in a wide range of applications. Most existing generative approaches for point cloud self-supervised learning focus on recovering masked points from visible ones within a single view. Recognizing that a two-view pre-training paradigm inherently introduces greater diversity and variance, it may thus enable more challenging and informative pre-training. Inspired by this, we explore the potential of two-view learning in this domain. In this paper, we propose Point-PQAE, a cross-reconstruction generative paradigm that first generates two decoupled point clouds/views and then reconstructs one from the other. To achieve this goal, we develop a crop mechanism for point cloud view generation for the first time and further propose a novel positional encoding to represent the 3D relative position between the two decoupled views. The cross-reconstruction significantly increases the difficulty of pre-training compared to self-reconstruction, which enables our method to surpass previous single-modal self-reconstruction methods in 3D self-supervised learning. Specifically, it outperforms the self-reconstruction baseline (Point-MAE) by 6.5%, 7.0%, and 6.7% in three variants of ScanObjectNN with the Mlp-Linear evaluation protocol. The code is available at https://github.com/aHapBean/Point-PQAE.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Semantic segmentation | S3DIS (Area 5) | mIOU61.4 | 799 | |
| Part Segmentation | ShapeNetPart (test) | mIoU (Inst.)86.1 | 312 | |
| Few-shot classification | ModelNet40 10-way 10-shot | Accuracy94.1 | 79 | |
| Few-shot classification | ModelNet40 10-way 20-shot | Accuracy96.3 | 79 | |
| Few-shot classification | ModelNet40 5-way 20-shot | Accuracy98.9 | 79 | |
| Few-shot classification | ModelNet40 5-way 10-shot | Accuracy96.9 | 79 | |
| 3D Object Classification | ModelNet40 1k P | Accuracy94 | 61 | |
| 3D Object Classification | ScanObjectNN PB_T50_RS (FULL Protocol) | Accuracy89.6 | 25 | |
| 3D Object Classification | ScanObjectNN OBJ_ONLY FULL Protocol | Accuracy93.6 | 23 | |
| 3D Object Classification | ScanObjectNN OBJ_BG (FULL Protocol) | Accuracy95 | 23 |