ACM Transactions on Graphics · SIGGRAPH Asia 2026 Journal Track ACM Transactions on Graphics · SIGGRAPH Asia 2026 期刊轨

doi:10.1145/3842525 ↗

Gauge-Equivariant Attention for Rotation-Stable 360° Scene Understanding Gauge-Equivariant Attention for Rotation-Stable 360° Scene Understanding规范等变注意力:实现旋转稳定的 360° 场景理解

Tianjian Zhou1 Yishan Li1 Jie Jiang1,* Yifei Zhang2

1 National University of Defense Technology (NUDT)  ·  2 Northwestern Polytechnical University (NPU) 1 国防科技大学(NUDT)  ·  2 西北工业大学(NPU)

* Corresponding author * 通讯作者

CCF-A ACM TOG SIGGRAPH Asia '26
Teaser plate: rotation-stable spherical segmentation under a 90-degree camera rotation across eight methods, above image-based relighting under camera tilt from 0 to 90 degrees
Teaser总览 Rotation-stable segmentation and relighting. Under a 90° camera rotation, EquiSSL keeps crisp spherical segmentation while spherical and tangent baselines accumulate errors (top); image-based relighting stays within ±0.9 pp across θ = 0°–90° where SphereUFormer drifts by up to +14.6 pp (bottom). 旋转稳定的分割与重光照。相机旋转 90° 时,EquiSSL 仍保持清晰的球面分割,而球面与切平面基线误差显著累积(上);在 θ = 0°–90° 范围内,基于图像的重光照误差始终保持在 ±0.9 pp 以内,而 SphereUFormer 的最大漂移达 +14.6 pp(下)。
0.8%
rotation drop · 90°90° 旋转下性能下降
+2.39
mIoU at 1% labels1% 标签下 mIoU 提升
0
added parameters新增参数
1.5–4.4%
forward latency前向延迟
01 Abstract摘要

Spherical attention should not care where the camera points. 球面注意力本不该在意相机指向何处。

Panoramic 360° 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 90°, and controlled ablations identify gauge dependence in its relative-position bias as a major contributor.

全景 360° 场景理解正日益依赖二十面体球(icosphere)Transformer,但当前最先进的球面模型在相机旋转 90° 时会损失一半以上的分割精度;受控消融实验表明,其相对位置偏置中的规范依赖(gauge dependence)是造成这一现象的主要原因。

We propose gauge-equivariant relative position encoding (GE-RPE): a parameter-free Reynolds average of the bias over a finite cyclic subgroup Cn ⊂ SO(2) of gauge rotations. Plugged into a SphereUFormer backbone the change is invisible at deployment—zero added parameters and only 1.5–4.4% forward latency—and the matched three-seed GE-RPE model records a 1.3% drop (compared to 53% drop in the published checkpoint).

我们提出规范等变相对位置编码(gauge-equivariant relative position encoding,GE-RPE):在规范旋转的有限循环子群 Cn ⊂ SO(2) 上对偏置做无参数的 Reynolds 平均。将其接入 SphereUFormer 骨干后,部署时几乎察觉不到变化——不增加任何参数,前向延迟仅增加 1.5–4.4%——而匹配三随机种子的 GE-RPE 模型仅出现 1.3% 的性能下降(已发表的检查点为 53%)。

Once the gauge defect is removed and a teacher-token permutation πR aligns the SSL views to the rotated student frame, iBOT+MAE pretraining stops being a liability and becomes a clean low-label lever: the full framework EquiSSL (GE-RPE + πR + iBOT+MAE) tightens the drop to 0.8% at 68.30% val mIoU and lifts 1%-label fine-tuning by +2.39 mIoU on the N=373 test split (and by +4.10 on the smaller N=40 val split); the same fix carries over to monocular depth and to zero-shot Structured3D segmentation. The construction is provably Cn-invariant and O(n-2)-close to the continuous SO(2) average, making the resulting model a usable 360° visual-computing primitive across panoramic relighting, immersive video, and cross-dataset transfer.

一旦消除规范缺陷,并由教师 token 置换 πR 将自监督学习的各视图对齐到旋转后的学生坐标系,iBOT+MAE 预训练便不再是负担,而成为提升低标签性能的有效手段:完整框架 EquiSSL(GE-RPE + πR + iBOT+MAE)将性能下降进一步收窄至 0.8%,验证集 mIoU 达 68.30%,并在 N=373 测试划分上使 1% 标签微调提升 +2.39 mIoU(在更小的 N=40 验证划分上提升 +4.10);同样的修正还可迁移到单目深度估计与零样本 Structured3D 分割。该构造可证明具有 Cn 不变性,且与连续 SO(2) 平均相差 O(n-2),使所得模型成为一种可用的 360° 视觉计算基元,可服务于全景重光照、沉浸式视频与跨数据集迁移。

Film影像 15-second paper video. Rotation sensitivity in state-of-the-art spherical transformers — more than 50% mIoU collapse under camera tilt — set against the mathematically guaranteed stability of EquiSSL (0.8% drop). 15 秒论文视频。最先进球面 Transformer 的旋转敏感性——相机倾斜下 mIoU 下降超过 50%——与 EquiSSL 在数学上可保证的稳定性(下降 0.8%)形成对照。
02 The defect缺陷

One scalar carries the whole defect. 缺陷藏在一个标量里。

When projecting a 360° panoramic scene onto a discretized icosphere mesh, each vertex requires a local tangent coordinate system (gauge) to define relative spatial positions. Existing spherical transformers choose local frame orientations arbitrarily. When the camera rotates, the local relative angles shift inconsistently, corrupting the attention weights.

将 360° 全景场景投影到离散的二十面体球网格时,每个顶点都需要一个局部切坐标系(即规范,gauge)来定义相对空间位置。现有球面 Transformer 对局部坐标系的朝向作任意选择;当相机旋转时,各顶点的局部相对角度发生不一致的偏移,注意力权重因此被破坏。

The failure is architectural, not a data or capacity problem: the relative-position bias is read in a per-vertex frame, so the same scene observed under a different global rotation yields a different attention bias.

这一失效是架构层面的,而非数据或容量问题:相对位置偏置在逐顶点的局部坐标系中读取,因此同一场景在不同全局旋转下会得到不同的注意力偏置。

The gauge defect and its fix: camera rotation re-orients per-vertex tangent frames and corrupts the relative-position bias, while Reynolds averaging over a discrete gauge group restores invariance
Fig. 1图 1 Diagnosis and fix of the gauge defect. Rotating the camera re-orients each vertex's local tangent frame, and a relative-position bias read in that frame corrupts attention (−53% mIoU for the SphereUFormer baseline). Reynolds-averaging the bias over a discrete gauge group removes the ambiguity (−0.8% mIoU drop for EquiSSL) with zero added parameters. 规范缺陷的诊断与修正。相机旋转会改变每个顶点局部切坐标系的朝向,而在该坐标系中读取的相对位置偏置会破坏注意力(SphereUFormer 基线 mIoU 下降 53%)。在离散规范群上对偏置做 Reynolds 平均即可消除歧义(EquiSSL mIoU 下降 0.8%),且不增加任何参数。

Key mathematical insight. True spherical equivariance requires attention weights to be invariant under local gauge transformations SO(2). By computing a finite cyclic group Reynolds average (Cn ⊂ SO(2)), GE-RPE eliminates gauge ambiguity without requiring extra model parameters or retraining backbones from scratch.

核心数学洞察。真正的球面等变性要求注意力权重在局部规范变换 SO(2) 下保持不变。GE-RPE 通过在有限循环群上计算 Reynolds 平均(Cn ⊂ SO(2)),在不引入额外模型参数、也无需从头重训练骨干的前提下消除规范歧义。

03 Method方法

Remove it, then exploit it. 先消除,再利用。

EquiSSL approaches the defect from two sides. GE-RPE fixes the architecture: the relative-position bias becomes a Reynolds average over a finite cyclic subgroup of gauge rotations, so it no longer depends on how each vertex's frame was chosen.

EquiSSL 从两个方向处理这一缺陷。GE-RPE 修正架构:把相对位置偏置改为规范旋转有限循环子群上的 Reynolds 平均,使其不再依赖每个顶点坐标系的选取方式。

πR fixes the pretraining: a teacher-token permutation aligns self-supervised views under global rotations, turning iBOT + MAE into a rotation-consistent feature booster. Neither change adds parameters; the deployed backbone keeps its size, and the forward-pass overhead is 1.5–4.4%.

πR 修正预训练:教师 token 置换在全局旋转下对齐自监督视图,使 iBOT + MAE 成为旋转一致的特征增强手段。两项改动都不增加参数;部署时骨干规模不变,前向开销为 1.5–4.4%。

EquiSSL pipeline: a 360-degree panorama projected onto an icosphere processed by SphereUFormer with GE-RPE, rotation-consistent self-supervised pretraining with teacher permutation pi_R, and the attention layer with Reynolds-averaged bias
Fig. 2图 2 The EquiSSL pipeline. Top: a 360° panorama is projected onto a rank-7 icosphere, processed by SphereUFormer with GE-RPE in every attention layer, and decoded into semantic labels. Bottom: rotation-consistent self-supervised pretraining aligns the EMA teacher and the student under a global rotation R via the token permutation πR, with iBOT + MAE losses. Right: the attention bias is the Reynolds average BGE(q,k) = (1/n) Σg B(g·r), which adds no parameters. EquiSSL 流程。上:360° 全景图投影到 7 阶二十面体球,经每个注意力层都含 GE-RPE 的 SphereUFormer 处理,再由分割头解码为语义标签。下:旋转一致的自监督预训练在全局旋转 R 下,通过 token 置换 πR 对齐 EMA 教师网络与学生网络,并以 iBOT + MAE 损失训练。右:注意力偏置为 Reynolds 平均 BGE(q,k) = (1/n) Σg B(g·r),不增加任何参数。

GE-RPE: Reynolds averagingGE-RPE:Reynolds 平均

Averages the relative-position bias across cyclic Cn rotations. Provably Cn-invariant, with O(n-2) convergence to the continuous SO(2) integral.

在循环子群 Cn 的旋转上对相对位置偏置取平均;可证明具有 Cn 不变性,并以 O(n-2) 的收敛速度逼近连续 SO(2) 积分。

Teacher permutation πR教师置换 πR

In multi-crop SSL (iBOT + MAE), rotating the student view used to misalign its tokens with the teacher. The exact icosphere permutation πR removes that mismatch and turns SSL into a rotation-robust feature booster.

在多裁剪自监督学习(iBOT + MAE)中,旋转学生视图曾使其 token 与教师网络错位;二十面体球上的精确置换 πR 消除了这种错配,使自监督学习成为提升旋转鲁棒性的特征增强手段。

04 Evidence证据

Stable at 90°. Efficient at 1%. 90° 处稳定,1% 标签下高效。

Benchmarks follow the paper: the 0° and 90° columns report Stanford2D3D val mIoU, and the 1%-label column reports fine-tuning mIoU on the N=373 test split.

基准结果与论文一致:0° 与 90° 两列为 Stanford2D3D 验证集 mIoU,1% 标签列为 N=373 测试划分上的微调 mIoU。

Benchmark comparison of EquiSSL against spherical baselines under 0° and 90° camera rotation and with 1% of labels EquiSSL 与球面基线在 0° 与 90° 相机旋转以及仅 1% 标签条件下的基准对比
Method方法 Backbone骨干 0° mIoU (%)0° mIoU (%) 90° mIoU (%)90° mIoU (%) Rotation drop (Δ)旋转性能下降(Δ) 1% label mIoU1% 标签 mIoU
SphereUFormer (published)SphereUFormer(已发表) TransformerTransformer 67.50 31.72 -35.78 (-53.0%) 41.20
SphereUFormer (matched 3-seed)SphereUFormer(匹配三随机种子) TransformerTransformer 66.82 65.95 -0.87 (-1.3%) 42.50
EquiSSL (full framework)EquiSSL(完整框架) Transformer + GE-RPETransformer + GE-RPE 68.30 67.75 -0.55 (-0.8%) 44.89 (+2.39)

EquiSSL's 1%-label gain of +2.39 mIoU is measured against the matched three-seed baseline. EquiSSL 在 1% 标签下 +2.39 mIoU 的提升相对于匹配三随机种子的基线测得。

Scroll the table horizontally to read all columns → 左右滑动表格以查看全部列 →

Validation mIoU versus camera-rotation cap from 0 to 90 degrees for EquiSSL and spherical baselines, showing EquiSSL inside the robust band while baselines collapse
Fig. 3图 3 Rotation stability. Validation mIoU under an increasing camera-rotation cap θmax. EquiSSL and its ablations remain inside the <2% robust band from 0° to 90°, while spherical and tangent baselines lose 29–60% mIoU. 旋转稳定性。相机旋转上限 θmax 增大时的验证集 mIoU。EquiSSL 及其消融变体在 0° 至 90° 范围内始终处于 <2% 的稳健区间,而球面与切平面基线损失 29–60% 的 mIoU。
Semantic segmentation and per-pixel error maps for eight methods under rotated views; EquiSSL has the fewest wrong pixels
Fig. 4图 4 Segmentation under rotated views. Semantic predictions (top) and per-pixel error maps (bottom; green correct, red wrong). EquiSSL's error map stays nearly clean where SphereUFormer, standard RPE, Tangent-Img, PanoFormer, HEAL-SWIN and SO3UFormer accumulate region and boundary errors. 旋转视角下的分割结果。上排为语义预测,下排为逐像素误差图(绿色正确,红色错误)。与 SphereUFormer、标准 RPE、Tangent-Img、PanoFormer、HEAL-SWIN 和 SO3UFormer 累积的区域与边界错误相比,EquiSSL 的误差图几乎保持干净。
Image-based relighting comparison under rotations from 0 to 90 degrees, with per-view error deltas growing for SphereUFormer while EquiSSL stays nearly constant
Fig. 5图 5 Image-based relighting under rotation. Per-view error Δ against the ground truth at θ = 0°–90°. SphereUFormer drifts by up to +14.6 pp, while EquiSSL stays within ±0.9 pp. 旋转下的基于图像重光照。θ = 0°–90° 各视角相对于真值的误差 Δ。SphereUFormer 的最大漂移达 +14.6 pp,而 EquiSSL 始终保持在 ±0.9 pp 以内。

Beyond segmentation 分割之外

The same correction carries over to other tasks: zero-shot Structured3D accuracy improves by +3.1%, monocular depth relative RMSE drops by 18.4% under extreme rotation, and on the smaller N=40 val split the 1%-label gain rises to +4.10 mIoU.

同一修正还可迁移到其他任务:零样本 Structured3D 准确率提升 +3.1%,单目深度在极端旋转下的相对 RMSE 下降 18.4%,而在更小的 N=40 验证划分上,1% 标签的增益升至 +4.10 mIoU。

68.30%
val mIoU at 0°0° 验证集 mIoU
+4.10
mIoU at 1% · N=401% 标签 · N=40
+3.1%
zero-shot Structured3D零样本 Structured3D
18.4%
depth RMSE drop深度 RMSE 下降
05 Citation引用

Cite this work. 引用本文。

If you find our work or code useful in your research, please cite: 如果我们的工作或代码对您的研究有所帮助,请引用:

@article{zhou2026gauge,
  title   = {Gauge-Equivariant Attention for Rotation-Stable {360$^\circ$} Scene Understanding},
  author  = {Zhou, Tianjian and Li, Yishan and Jiang, Jie and Zhang, Yifei},
  journal = {ACM Transactions on Graphics},
  volume  = {45},
  number  = {6},
  year    = {2026},
  note    = {Article 242},
  doi     = {10.1145/3842525}
}

ACM Transactions on Graphics, Vol. 45, No. 6, Article 242 · DOI 10.1145/3842525 ACM Transactions on Graphics · 第 45 卷第 6 期 · 文章 242 · DOI 10.1145/3842525

Project项目