PointGPT: Auto-regressively Generative Pre-training from Point Clouds
About
Large language models (LLMs) based on the generative pre-training transformer (GPT) have demonstrated remarkable effectiveness across a diverse range of downstream tasks. Inspired by the advancements of the GPT, we present PointGPT, a novel approach that extends the concept of GPT to point clouds, addressing the challenges associated with disorder properties, low information density, and task gaps. Specifically, a point cloud auto-regressive generation task is proposed to pre-train transformer models. Our method partitions the input point cloud into multiple point patches and arranges them in an ordered sequence based on their spatial proximity. Then, an extractor-generator based transformer decoder, with a dual masking strategy, learns latent representations conditioned on the preceding point patches, aiming to predict the next one in an auto-regressive manner. Our scalable approach allows for learning high-capacity models that generalize well, achieving state-of-the-art performance on various downstream tasks. In particular, our approach achieves classification accuracies of 94.9% on the ModelNet40 dataset and 93.4% on the ScanObjectNN dataset, outperforming all other transformer models. Furthermore, our method also attains new state-of-the-art accuracies on all four few-shot learning benchmarks.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Part Segmentation | ShapeNetPart (test) | mIoU (Inst.)86.6 | 312 | |
| Shape classification | ModelNet40 (test) | OA94.9 | 255 | |
| Object Classification | ScanObjectNN OBJ_BG | Accuracy95.8 | 215 | |
| Part Segmentation | ShapeNetPart | mIoU (Instance)86.6 | 198 | |
| Object Classification | ScanObjectNN PB_T50_RS | Accuracy91.9 | 195 | |
| Object Classification | ScanObjectNN OBJ_ONLY | Overall Accuracy95.2 | 166 | |
| Classification | ModelNet40 (test) | -- | 99 | |
| Few-shot classification | ModelNet40 5-way 20-shot | Accuracy98.6 | 79 | |
| Few-shot classification | ModelNet40 5-way 10-shot | Accuracy96.8 | 79 | |
| Few-shot classification | ModelNet40 10-way 10-shot | Accuracy92.6 | 79 |