Gauge-Equivariant Attention for Rotation-Stable Scene Understanding
Organizations: National University of Defense Technology (NUDT), Changsha, China · Northwestern Polytechnical University, Xi’an, China
Abstract
Panoramic scene understanding increasingly relies on icosphere transformers, but a state-of-the-art spherical model loses more than half of its segmentation accuracy when the camera rotates by , and controlled ablations identify gauge dependence in its relative-position bias as a major contributor. We propose gauge-equivariant relative position encoding (GE-RPE): a parameter-free Reynolds average of the bias over a finite cyclic subgroup of gauge rotations. Plugged into a SphereUFormer backbone the change is invisible at deployment---zero added parameters and -- forward latency---and the matched three-seed GE-RPE model records a drop; the published SphereUFormer checkpoint records under the same stress protocol but a different training recipe. Once the gauge defect is removed and a teacher-token permutation aligns the SSL views to the rotated student frame, iBOTMAE pretraining stops being a liability and becomes a clean low-label lever: the full framework EquiSSL (GE-RPE iBOTMAE) tightens the drop to at val mIoU and lifts -label fine-tuning by mIoU on the test split (and by on the smaller val split); the same fix carries over to monocular depth and to zero-shot Structured3D segmentation. The construction is provably -invariant and -close to the continuous average, making the resulting model a usable visual-computing primitive across panoramic relighting, immersive video, and cross-dataset transfer. Code is available at https://github.com/Jaywalk18/equissl-release.
Figures & tables
| Method | Venue | Equiv. | Params | Fwd | val mIoU | test | rot. mIoU | Drop ⋄ | Drop std | lab. | Depth |
| (M) | (ms) | (%) | (%) | @ | (%) | val (%) | |||||
| Prior work (single best; same split, val-mIoU context). | |||||||||||
| SFSS ∗ | ICCV’19 | none | |||||||||
| HEAL-SWIN ( Carlsson et al., 2024 ) | CVPR’24 | sampling-only | |||||||||
| SphereUFormer ( Benny et al., 2025 ) | CVPR’25 | none | |||||||||
| SO3UFormer ‡ ( Zhu et al., 2026 ) | 2026 | SO(3) feat. | |||||||||
| RPE | Train aug. | mIoU @ | mIoU @ | Drop |
|---|---|---|---|---|
| Standard | none | |||
| Standard | Haar- (32) | |||
| Radial | none | |||
| GE-RPE | none |
| Seg. val mIoU | test | Low-label val mIoU | Depth | Depth | Rot. | S3D zero-shot | S3D | ||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Pretraining | RPE | @ | @ | mIoU | RMSE | Drop | @ | @ | Drop | ||||
| Random init | No RPE | ||||||||||||
| Random init | Standard | ||||||||||||
| Random init | GE-RPE | ||||||||||||
| Random init | GE-RPE | ||||||||||||
| iBOT MAE | No RPE | ||||||||||||
| val mIoU @ ( ∘ , %) | test | Drop | |||||||
|---|---|---|---|---|---|---|---|---|---|
| Config. | ( ) | ||||||||
| + area | |||||||||
| no-area | |||||||||
| (n.a. a) | pp | ||||||||
| RPE variant | Build | Fwd | Fwd | Peak Mem | Mem |
|---|---|---|---|---|---|
| (s) | (ms) | (%) | (MB) | (%) | |
| No RPE | — | — | |||
| Standard RPE | |||||
| GE-RPE | |||||
| GE-RPE | |||||
| GE-RPE (+area) |
| RPE Variant | S3D val | @ | Drop | @ |
|---|---|---|---|---|
| No RPE | ||||
| Standard RPE | — | |||
| GE-RPE | ||||
| GE-RPE + area | ||||
| GE-RPE no-area |
| RPE Variant | MP3D val | @ | Drop | |
|---|---|---|---|---|
| No RPE | ||||
| Standard RPE | — | |||
| GE-RPE | ||||
| EquiSSL ( no-area) |
| PSNR (dB) | SSIM | LPIPS | ||||
|---|---|---|---|---|---|---|
| EquiSSL | Std | EquiSSL | Std | EquiSSL | Std | |
| Task | Init | S1 | S2 | Loss | Hrs | Seeds | Cap | Src |
|---|---|---|---|---|---|---|---|---|
| (ep) | (ep) | |||||||
| Seg. | random | – | CE+Dice | – | T 1 | |||
| Seg. | iBOT+MAE | CE+Dice | – | T 3 | ||||
| Seg. | random | – | CE+Dice | – | T 21 | |||
| Seg. | iBOT+MAE | CE+Dice | – | T 3 | ||||
| Depth | iBOT+MAE | L1 | – | T 16 |
Appendix figures & tables19 assets
Supplementary material from the paper’s appendix.
Appendix
| 1 | 2 | 3 | 4 | 5 | 6 | ||
|---|---|---|---|---|---|---|---|
| (no avg.) | all | ||||||
| even only | |||||||
| Variant | parity | kills ? | surviving | Build (s) | val @ | val @ | Drop | Drop std | Resid. ⋆ | Bd. ratio | |||
| Identity / negative ablation | |||||||||||||
| Standard RPE | odd | no | |||||||||||
| GE-RPE | even | no | |||||||||||
| Odd- alternatives (measured; no parity preservation) | |||||||||||||
| GE-RPE | odd | yes | |||||||||||
| GE-RPE | odd | yes | |||||||||||
| Variant | Heads | (rel. %) | (rel. %) | Ratio | ||
|---|---|---|---|---|---|---|
| Standard RPE | 1 | 366 | 81.15 | 1041.9 | 0.071 | 1.000 |
| GE-RPE | 2 | 366 | 48.85 | 329.4 | 0.150 | 0.250 |
| GE-RPE | 4 | 366 | 25.44 | 82.3 | 0.309 | 0.062 |
| GE-RPE | 6 | 366 | 14.04 | 36.6 | 0.389 | 0.028 |
| val mIoU | val mIoU | Drop | Drop | |
|---|---|---|---|---|
| Variant | @ | @ | (meas., 3-seed) | ( pred.) |
| no-area (reference) | — | |||
| no-area | ||||
| no-area | ||||
| Haar (FA limit) | — |
| Bias param. | drop | drop | drop |
|---|---|---|---|
| bilinear (canonical) | |||
| MLP |
| Small (14.6M) | Large (35.1M) | |
| Embedding dimension | 32 | 48 |
| Encoder depths | [2, 2, 2, 2] | [2, 2, 6, 2] |
| Encoder heads | [2, 4, 8, 16] | [3, 6, 12, 24] |
| Bottleneck depth | 2 | 2 |
| Window size coef. | 2 | 2 |
| MLP ratio | 4.0 | 4.0 |
| Pretraining | RPE | RMSE | Abs Rel | |
|---|---|---|---|---|
| Random init | No RPE | |||
| Random init | Standard | |||
| Random init | GE-RPE | |||
| Random init | GE-RPE no-area | |||
| iBOT MAE | No RPE | |||
| iBOT MAE | Standard |
| Pretraining | RPE | @ |
|---|---|---|
| Random init | Standard | |
| Random init | GE-RPE no-area | |
| iBOT MAE | Standard | |
| iBOT MAE | EquiSSL |
| Class | No | Std | n.a. | |
|---|---|---|---|---|
| beam | ||||
| board | ||||
| bookcase | ||||
| ceiling | ||||
| chair | ||||
| clutter |
| Seed | (%) | (%) | (paired) |
|---|---|---|---|
| 42 | |||
| 123 | |||
| 456 | |||
| mean | |||
| std |
| SphereUF. | No RPE | Std RPE | +area | (no SSL) | EquiSSL | |||
|---|---|---|---|---|---|---|---|---|
| Label% | None | Std | SSL | SSL | |||
|---|---|---|---|---|---|---|---|
| 10 | |||||||
| 50 | |||||||
| 101 | |||||||
| 1013 |
| Label% | None | Std | SSL | SSL | |||
|---|---|---|---|---|---|---|---|
| 10 | |||||||
| 50 | |||||||
| 101 | |||||||
| 1013 |
| Configuration | Rotation drop |
|---|---|
| Full EquiSSL | |
| SO(3) augmentation | |
| teacher-token permutation | |
| GE-RPE (= Standard RPE iBOT MAE) |
| Pretext objective | Val mIoU @ | Rotation drop |
|---|---|---|
| iBOT MAE (composite, canonical) | ||
| iBOT only ( ) | ||
| MAE only ( ) |
3D Scene Understanding
3D scene understanding builds geometric and semantic representations of environments for spatial reasoning, perception, and interaction.
See all 295 papers