Beyond Random Masking: A Dual-Stream Approach for Rotation-Invariant Point Cloud Masked Autoencoders
Not on the map. Why?
Abstract (arXiv)
Existing rotation-invariant point cloud masked autoencoders (MAE) rely on random masking strategies that overlook geometric structure and semantic coherence. Random masking treats patches independently, failing to capture spatial relationships consistent across orientations and overlooking semantic object parts that maintain identity regardless of rotation. We propose a dual-stream masking approach combining 3D Spatial Grid Masking and Progressive Semantic Masking to address these fundamental limitations. Grid masking creates structured patterns through coordinate sorting to capture geometric relationships that persist across different orientations, while semantic masking uses attention-driven clustering to discover semantically meaningful parts and maintain their coherence during masking. These complementary streams are orchestrated via curriculum learning with dynamic weighting, progressing from geometric understanding to semantic discovery. Designed as plug-and-play components, our strategies integrate into existing rotation-invariant frameworks without architectural changes, ensuring broad compatibility across different approaches. Comprehensive experiments on ModelNet40, ScanObjectNN, and OmniObject3D demonstrate consistent improvements across various rotation scenarios, showing substantial performance gains over the baseline rotation-invariant methods.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 25 of its 37 references that are in the map. First posted on arXiv 18 Sep 2025.
Builds on
25 of 37 references.
References · 25
- 2017Attention Is All You Needcited by 60,361
- 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,403
- 2017PointNet++: Deep Hierarchical Feature Learning on Point Sets in a Metric Spacecited by 4,401
- 2015ShapeNet: An Information-Rich 3D Model Repositorycited by 2,897
- 2021BEiT: BERT Pre-Training of Image Transformerscited by 2,375
- 2018Tensor field networks: Rotation- and translation-equivariant neural networks for 3D point cloudscited by 746
- 2019Deep Learning for 3D Point Clouds: A Surveycited by 558
- 2020PointContrast: Unsupervised Pre-training for 3D Point Cloud Understandingcited by 504
- 2022Masked Autoencoders for Point Cloud Self-supervised Learningcited by 459
- 2023OmniObject3D: Large-Vocabulary 3D Object Dataset for Realistic Perception, Reconstruction and Generationcited by 326
- 2022Point-M2AE: Multi-scale Masked Autoencoders for Hierarchical Point Cloud Pre-trainingcited by 264
- 2022CrossPoint: Self-Supervised Cross-Modal Contrastive Learning for 3D Point Cloud Understandingcited by 198
13 more
- 2021Walk in the Cloud: Learning Curves for Point Clouds Shape Analysiscited by 172
- 2022What to Hide from Your Students: Attention-Guided Masked Image Modelingcited by 118
- 2019Rotation Invariant Convolutions for 3D Point Clouds Deep Learningcited by 80
- 2021Exploiting Edge-Oriented Reasoning for 3D Point-based Scene Graph Analysiscited by 49
- 2020Rotation-Invariant Local-to-Global Representation Learning for 3D Point Cloudcited by 38
- 2022RIConv++: Effective Rotation Invariant Convolutions for 3D Point Clouds Deep Learningcited by 29
- 2024Bringing Masked Autoencoders Explicit Contrastive Properties for Point Cloud Self-Supervised Learningcited by 16
- 2024RI-MAE: Rotation-Invariant Masked AutoEncoders for Self-Supervised Point Cloud Representation Learningcited by 15
- 2023PaRot: Patch-Wise Rotation-Invariant Network via Feature Disentanglement and Pose Restorationcited by 7
- 2022Rethinking Rotation Invariance with Point Cloud Registrationcited by 5
- 2024MaskLRF: Self-supervised Pretraining via Masked Autoencoding of Local Reference Frames for Rotation-invariant 3D Point Set Analysiscited by 4
- 2024Point Cloud Understanding via Attention-Driven Contrastive Learningcited by 1
- 2025HFBRI-MAE: Handcrafted Feature Based Rotation-Invariant Masked Autoencoder for 3D Point Cloud Analysiscited by 1
Built on by
Not cited yet.