OpenShape: Scaling Up 3D Shape Representation Towards Open-World Understanding

Minghua Liu,Ruoxi Shi,Kaiming Kuang,Yinhao Zhu,Xuanlin Li,Shizhong Han,Hong Cai,Fatih Porikli,Hao Su

2023-06-17

Abstract:We introduce OpenShape, a method for learning multi-modal joint representations of text, image, and point clouds. We adopt the commonly used multi-modal contrastive learning framework for representation alignment, but with a specific focus on scaling up 3D representations to enable open-world 3D shape understanding. To achieve this, we scale up training data by ensembling multiple 3D datasets and propose several strategies to automatically filter and enrich noisy text descriptions. We also explore and compare strategies for scaling 3D backbone networks and introduce a novel hard negative mining module for more efficient training. We evaluate OpenShape on zero-shot 3D classification benchmarks and demonstrate its superior capabilities for open-world recognition. Specifically, OpenShape achieves a zero-shot accuracy of 46.8% on the 1,156-category Objaverse-LVIS benchmark, compared to less than 10% for existing methods. OpenShape also achieves an accuracy of 85.3% on ModelNet40, outperforming previous zero-shot baseline methods by 20% and performing on par with some fully-supervised methods. Furthermore, we show that our learned embeddings encode a wide range of visual and semantic concepts (e.g., subcategories, color, shape, style) and facilitate fine-grained text-3D and image-3D interactions. Due to their alignment with CLIP embeddings, our learned shape representations can also be integrated with off-the-shelf CLIP-based models for various applications, such as point cloud captioning and point cloud-conditioned image generation.

Computer Vision and Pattern Recognition

What problem does this paper attempt to address?

The paper aims to address several key issues in 3D shape understanding:

1. **Data Scale Limitation**: Existing 3D training datasets are relatively small, leading to poor generalization of models when encountering unseen 3D shape categories, which limits the deployment of current models in practical applications.

2. **Cross-Modal Representation Alignment**: The method of achieving zero-shot 3D shape classification by projecting 3D point clouds to 2D images and analyzing them through CLIP faces problems of occlusion and information loss. Additionally, it introduces unnecessary delays due to the need for rendering point clouds and multiple CLIP inferences.

3. **Text Quality Improvement**: The 3D shape text descriptions in the main dataset Objaverse used in the paper contain a lot of noise, requiring automatic filtering and enrichment of these texts to improve model performance.

To address these issues, the authors propose the OpenShape method, which combines multiple 3D datasets, enhances the quality of text descriptions, and introduces a hard negative mining strategy to improve contrastive learning. Through these techniques, OpenShape achieves significant performance improvements in multiple benchmarks, particularly excelling in handling long-tail categories. For example, on the Objaverse-LVIS dataset containing 1156 categories, OpenShape achieves a zero-shot classification accuracy of 46.8%, far surpassing existing methods. Additionally, OpenShape demonstrates extensive visual and semantic concept encoding capabilities and can be easily integrated into other CLIP-based models, supporting various cross-modal tasks.