Revisiting Triplet Contrastive Learning for 3D Shape Understanding
Organizations: Xiamen University · Shanghai AI Lab
Abstract
In multi-modal 3D pre-training, aligning 3D shapes, their 2D image counterparts, and language descriptions has made tremendous progress. However, we find that supervision from rendered images is unreliable, the semantics encoded in synthetic views are distorted due to domain gap, self-occlusion, and rendering artifacts. Instead, we learn transferable 3D representations from language only. We observe that, in real world, 3D shapes are organized in a tree-like structure, ranging from general concepts to fine-grained instances with specific attributes, while language typically denotes generic concepts and subsumes these various 3D instantiations. Hence, we propose to embed 3D representations in hyperbolic space, whose geometry is well-suited for modeling tree-like data without distortion. To capture the asymmetric hierarchy between language and 3D shapes, we further impose a partial-order relation that language entails 3D shapes. Our method, Hyperbolic Contrastive Learning (HyperCL), jointly optimizes a hyperbolic geodesic-based contrastive objective and an entailment loss to learn discriminative cross-modal representations while explicitly preserving this hierarchy. We also introduce a rigorously verified benchmark for comprehensively evaluating cross-modal 3D alignment. Across two widely used benchmarks and ours, we improve the performance up to 4.6% absolute percentage on 3D shape classification and 5.5% on cross-modal text-to-3D retrieval under the same parameter budget. The code and evaluation benchmark will be made publicly available.
Figures & tables
| Method | Data | OLVIS | c-OLVIS | MN40 | ||||||
| Acc@1 | Acc@3 | Acc@5 | Acc@1 | Acc@3 | Acc@5 | Acc@1 | Acc@3 | Acc@5 | ||
| ReCon Qi et al. (2023) | ShapeNet | 1.1 | 2.7 | 3.7 | - | - | - | 61.2 | 73.9 | 78.1 |
| CLIP2Point Huang et al. (2023) | 2.7 | 5.8 | 7.9 | - | - | - | 49.5 | 71.3 | 81.2 | |
| ULIP Xue et al. (2023) | 6.2 | 13.6 | 17.9 | 9.9 | 19.1 | 24.1 | 60.4 | 79.0 | 84.4 | |
| OpenShape Liu et al. (2023) | 10.8 | 20.2 | 25.0 | 16.5 | 27.3 | 32.1 | 70.3 | 86.9 | 91.3 | |
| TAMM Zhang et al. (2024) | 13.7 | 24.2 | 29.2 | 23.9 | 35.5 | 41.2 | 73.1 | 88.5 | 91.9 | |
| Method | Data | P T | T P | ||||
| R@1 | R@3 | R@5 | R@1 | R@3 | R@5 | ||
| ULIP2 Xue et al. (2024) | ShapeNet | 1.0 | 2.4 | 3.6 | 1.0 | 2.7 | 3.9 |
| TAMM Zhang et al. (2024) | 3.9 | 9.4 | 12.8 | 4.9 | 10.5 | 14.4 | |
| HyperCL (Ours) | 6.0 | 12.1 | 17.1 | 7.0 | 14.7 | 19.2 | |
| OpenShape Liu et al. (2023) | Ensembled (w/o LVIS) | 19.6 | 35.2 | 45.5 | 20.4 | 37.1 | 46.0 |
| Uni3D-B Zhou et al. (2024) | 22.0 | 39.3 | 48.8 | 18.5 | 34.1 | 42.7 | |
Appendix figures & tables5 assets
Supplementary material from the paper’s appendix.
Appendix
| Hyperparameters | Value | Descriptions |
| Normalization | ✓ | normalize xyz coordinates of point cloud into [-1, 1] |
| Point dropout | 0.875 | the maximum ratio of dropped points is 0.875 |
| Random scale | [0.8, 1.25] | uniformly scale points |
| Random shift | [-0.1, 0.1] | shift x, y, z coordinates independently |
| Random rotation | ✓ | small rotation jittering around x, y, z axes |
| Random rotation around gravity-axis | ✓ | rotate around gravity-axis for learning rotation invariance |
| Speed (it/s) | Peak Memory (GB) | GFLOPs | |
| 3.30 | 46.65 | 72.05 | |
| 3.43 | 46.65 | 72.05 | |
| 3.36 | 46.65 | 72.05 |
3D Representation Learning
3D representation learning builds geometric features for recognizing, reconstructing, generating, and reasoning about objects and scenes.
See all 187 papers