arrow
Return

Inter-Modal Masked Autoencoder for Self-Supervised Learning on Point Clouds

delete2024-01-01
delete2
PRE
AI
刘嘉明 cover
刘嘉明 (Jiaming Liu)
Y
Yue Wu *
M
Maoguo Gong
Z
Zhixiao Liu
Q
Qiguang Miao
W
Wenping Ma
DOI:10.1109/TMM.2023.3317998delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Masked autoencoder (MAE) is a recently widely used self-supervised learning method that has achieved great success in NLP and computer vision. However, the potential advantages of masked pre-training for point cloud understanding have not been fully explored. There is preliminary work on MAE-based point clouds using the Transformer architecture to explore low-level geometric representations in 3D space, which is insufficient for fine-grained decoding completion and downstream tasks. Inspired by multimodality, we propose Inter-MAE, a inter-modal MAE method for self-supervised learning on point clouds. Specifically, we first use Point-MAE as a baseline to partition point clouds into random low percentage of visible and high percentage of masked point patches. Then, a standard Transformer-based autoencoder is built by asymmetric design and shifting mask operations, and latent features are learned from the visible point patches aiming to recover the masked point patches. In addition, we generate image features based on ViT after point cloud rendering to form inter-modal contrastive learning with the decoded features of the completed point patches. Extensive experiments show that the proposed Inter-MAE generates pre-trained models that are effective and exhibit superior results in various downstream tasks. For example, an accuracy of 85.4% is achieved on ScanObjectNN and 86.3% on ShapeNetPart, outperforming other state-of-the-art self-supervised learning methods. Notably, our work establishes for the first time the feasibility of applying image modality to masked point clouds.
Keywords:
Point cloud compression
Transformers
Task analysis
Standards
Computer architecture
Decoding
Self-supervised learning
Self-supervision
masked autoencoder
joint multimodality
point cloud understanding

Journal

IEEE Transactions on Multimedia cover
IEEE Transactions on Multimedia
IF:
9.7
Papers:
4.5K
Citations:
2.4W

Organization

H
Harbin Engineering University
Scholars:
1.9W
Papers: 1.3W
Citations: 1.3W
X
Xidian University
Scholars:
2.4W
Papers: 1.9W
Citations: 9.7K