SIEVE: Overcoming Topological Obstruction in Equivariant Self-Supervised Learning
Organizations: Max Planck Institute of Mathematics Bonn, Germany · SolverX Seoul, Republic of Korea
Abstract
We identify a theoretical incompatibility between 3D point cloud masked autoencoding and SE(3)-equivariant networks. We prove that standard masking triggers a topological collapse, effectively forcing the model to hallucinate orientation without a reference frame. To resolve this, we introduce SIEVE (Scalar Invariant Extraction Via Equivariance), an SE(3)-equivariant self-supervised framework that compresses point cloud geometry into invariant scalar embeddings. At its core is the spherical projection mask, which preserves global pose while obfuscating local semantics, sidestepping the obstruction by construction. Experiments confirm that SIEVE circumvents the predicted collapse where constant and noise masks fail. On downstream point cloud tasks, the resulting embeddings improve over coordinate-based baselines and hand-crafted geometric descriptors, with the largest gains on generalization to unseen shape categories.
Figures & tables
| Model | SIEVE | |||
|---|---|---|---|---|
| PointNet++ | 71.55 | 75.54 | 72.90 | 86.28 |
| DeltaNet | 87.21 | 88.07 | 86.86 | 89.58 |
| DiffusionNet | 81.20 | 85.89 | 85.97 | 89.30 |
| Seen Categories | PCN | PCN + SIEVE | Unseen Categories | PCN | PCN + SIEVE |
|---|---|---|---|---|---|
| Airplane | 1.6508 | 1.8674 | Bus | 4.2137 | 4.2908 |
| Cabinet | 4.8287 | 4.3237 | Bed | 33.1502 | 29.4725 |
| Car | 2.6543 | 2.7663 | Bookshelf | 13.6558 | 12.5375 |
| Chair | 5.7248 | 5.6236 | Bench | 7.4325 | 6.3300 |
| Lamp | 7.4924 | 7.2977 | Guitar | 7.4999 | 5.2607 |
| Sofa | 5.7775 | 5.6728 | Motorbike | 9.4908 | 8.9940 |
| Model | MSE( ) | RMSE( ) | MAE( ) | MSE( ) | RMSE( ) | MAE( ) |
|---|---|---|---|---|---|---|
| Shared Categories | ||||||
| DCP | 1.475760 | 1.214808 | 0.826500 | 0.000004 | 0.002040 | 0.001239 |
| DCP + SIEVE | 0.703465 | 0.838728 | 0.479402 | 0.000002 | 0.001539 | 0.000774 |
| Separate Categories | ||||||
| DCP | 6.948212 | 2.635946 | 1.782947 | 0.000025 | 0.005048 | 0.003669 |
| DCP + SIEVE | 0.874955 | 0.935390 | 0.511852 | 0.000002 | 0.001509 | 0.000813 |
Appendix figures & tables10 assets
Supplementary material from the paper’s appendix.
Appendix
| Module | Hyperparameter | GATr | SE(3)-Transformer |
|---|---|---|---|
| Encoder | Input Channel (s,v) | (1,1) | (1,1) |
| Hidden Channel (s,v) | (32,8) | (32,8) | |
| Output Channel (s,v) | (16,1) | (16,1) | |
| Number of Attention Heads | 4 | 1 | |
| Number of Layers | 6 | 6 | |
| Decoder | Hidden Channel (s,v) | (32,8) | (32,8) |
| seed 42 | seed 67 | seed 721 | |
|---|---|---|---|
| Validation Loss |
| only | radius | No Noise | Simple Noise | Curv and Norm | NN Attention Mask |
|---|---|---|---|---|---|
| 71.55 | 75.54 | 76.08 | 75.94 | 78.05 | 84.99 |
| PointNet++ | DeltaNet | DiffusionNet | |
|---|---|---|---|
| as feature | 71.55 | 87.21 | 81.20 |
| random init SIEVE | 67.47 | 77.59 | 82.52 |
| pre-trained SIEVE | 86.28 | 89.58 | 89.30 |
| injection location | MSE( ) | RMSE( ) | MAE( ) | MSE( ) | RMSE( ) | MAE( ) |
|---|---|---|---|---|---|---|
| Before DGCNN (chosen) | 7.05 | 2.65 | 1.39 | 0.000004 | 0.0020 | 0.00092 |
| DGCNN Intermediate | 7.34 | 2.71 | 1.55 | 0.000006 | 0.0025 | 0.0015 |
| Before Transformer | 12.43 | 3.53 | 2.01 | 0.000029 | 0.0054 | 0.0036 |
| Before SVD | 24.59 | 4.96 | 3.12 | 0.000452 | 0.021 | 0.015 |