Point Cloud Self-supervised Learning via 3D to Multi-view Masked Learner

TL;DR

This paper introduces a novel self-supervised learning method for 3D point clouds that leverages multi-view projections and a specialized attention mechanism, eliminating the need for 2D input data and improving performance on multiple tasks.

Contribution

It proposes a 3D to multi-view autoencoder with multi-scale attention and a two-stage self-training strategy, advancing 3D representation learning without relying on 2D modalities.

Findings

Outperforms state-of-the-art methods in 3D classification

Achieves superior results in part segmentation

Improves object detection accuracy

Abstract

Recently, multi-modal masked autoencoders (MAE) has been introduced in 3D self-supervised learning, offering enhanced feature learning by leveraging both 2D and 3D data to capture richer cross-modal representations. However, these approaches have two limitations: (1) they inefficiently require both 2D and 3D modalities as inputs, even though the inherent multi-view properties of 3D point clouds already contain 2D modality. (2) input 2D modality causes the reconstruction learning to unnecessarily rely on visible 2D information, hindering 3D geometric representation learning. To address these challenges, we propose a 3D to Multi-View Learner (Multi-View ML) that only utilizes 3D modalities as inputs and effectively capture rich spatial information in 3D point clouds. Specifically, we first project 3D point clouds to multi-view 2D images at the feature level based on 3D-based pose. Then, we introduce two components: (1) a 3D to multi-view autoencoder that reconstructs point clouds and multi-view images from 3D and projected 2D features; (2) a multi-scale multi-head (MSMH) attention mechanism that facilitates local-global information interactions in each decoder transformer block through attention heads at various scales. Additionally, a novel two-stage self-training strategy is proposed to align 2D and 3D representations. Our method outperforms state-of-the-art counterparts across various downstream tasks, including 3D classification, part segmentation, and object detection.