Multi-Angle Point Cloud-VAE: Unsupervised Feature Learning for 3D Point Clouds from Multiple Angles by Joint Self-Reconstruction and Half-to-Half Prediction
About
Unsupervised feature learning for point clouds has been vital for large-scale point cloud understanding. Recent deep learning based methods depend on learning global geometry from self-reconstruction. However, these methods are still suffering from ineffective learning of local geometry, which significantly limits the discriminability of learned features. To resolve this issue, we propose MAP-VAE to enable the learning of global and local geometry by jointly leveraging global and local self-supervision. To enable effective local self-supervision, we introduce multi-angle analysis for point clouds. In a multi-angle scenario, we first split a point cloud into a front half and a back half from each angle, and then, train MAP-VAE to learn to predict a back half sequence from the corresponding front half sequence. MAP-VAE performs this half-to-half prediction using RNN to simultaneously learn each local geometry and the spatial relationship among them. In addition, MAP-VAE also learns global geometry via self-reconstruction, where we employ a variational constraint to facilitate novel shape generation. The outperforming results in four shape analysis tasks show that MAP-VAE can learn more discriminative global or local features than the state-of-the-art methods.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Shape classification | ModelNet40 (test) | -- | 255 | |
| 3D Shape Classification | ModelNet40 (test) | Accuracy90.15 | 227 | |
| Object Classification | ModelNet40 (test) | -- | 180 | |
| 3D Object Part Segmentation | ShapeNet Part (test) | mIoU67.95 | 114 | |
| 3D shape recognition | ModelNet10 (test) | Accuracy94.82 | 64 | |
| Object Classification | ModelNet10 (test) | Accuracy94.8 | 60 | |
| Object Classification | ModelNet40 1.0 (test) | Accuracy90.2 | 19 | |
| Point Cloud Completion | ShapeNet Part Airplane | EMD (per point)0.0323 | 6 | |
| Point Cloud Completion | ShapeNet Part Chair | EMD (per point)0.0557 | 6 |