Masked Autoencoders for Point Cloud Self-supervised Learning
Yatian Pang Wenxiao Wang Francis E.H. Tay Affiliation: National University of Singapore Affiliation: National University of Singapore Affiliation: ZheJiang University Wei Liu Yonghong Tian Li Yuan Thanks: Corresponding author Affiliation: School of ECE at Peking University, Shenzhen Graduate School Affiliation: Tencent Data Platform Affiliation: School of Computer Science at Peking University、Pengcheng Laboratory E-mail
Abstract
As a promising scheme of self-supervised learning, masked autoencoding has significantly advanced natural language processing and computer vision. Inspired by this, we propose a neat scheme of masked autoencoders for point cloud self-supervised learning, addressing the challenges posed by point cloud’s properties, including leakage of location information and uneven information density. Concretely, we divide the input point cloud into irregular point patches and randomly mask them at a high ratio. Then, a standard Transformer based autoencoder, with an asymmetric design and a shifting mask tokens operation, learns high-level latent features from unmasked point patches, aiming to reconstruct the masked point patches. Extensive experiments show that our approach is efficient during pre-training and generalizes well on various downstream tasks. Specifically, our pre-trained models achieve 85.18% accuracy on ScanObjectNN and 94.04% accuracy on ModelNet40, outperforming all the other self-supervised learning methods. We show with our scheme, a simple architecture entirely based on standard Transformers can surpass dedicated Transformer models from supervised learning. Our approach also
原文 arXiv:2203.06604;中英对照 + 大白话阅读 https://aha.fim.ai/paper/2203.06604v2