提出旋转鲁棒的全景预测模型,解决相机倾斜时性能崩溃问题。
SO3UFormer: Learning Intrinsic Spherical Features for Rotation-Robust Panoramic Dense Prediction
- 移除绝对纬度编码,用局部切平面角构建位置偏置
- 在随机旋转下保持70.67的mIoU,基准模型降至25.26
- 适合手持或无人机采集的全景场景建模
全景密集预测模型通常依赖重力对齐假设,但实际拍摄中设备抖动或飞行平台姿态变化常导致相机非垂直。此时标准球面Transformer会过拟合全局纬度线索并失效。本文提出SO3UFormer,通过三个几何组件学习与坐标系无关的内在球面特征:(1) 移除绝对纬度编码以打破重力轴依赖;(2) 四阶一致球面注意力修正采样密度不均;(3) 基于局部切平面角的规范感知相对位置偏置。训练阶段引入对数空间的SO(3)一致性正则项,抑制残余离散化效应。为评估鲁棒性,构建了在±35°内随机旋转的Pose35数据集,并进行完整的SO(3)应力测试。基线SphereUFormer在Pose35上从67.53 mIoU降至全旋转测试下的25.26,而SO3UFormer在Pose35上达72.03,全测试下仍保持70.67。在真实数据集Matterport3D上的语义分割与全景深度估计任务中,该模型亦表现出几乎完全的旋转不变性,而重力锚定基线则严重失准。代码与模型已开源。
原文摘要 · Abstract (English)
Panoramic dense-prediction models, spanning semantic segmentation and depth estimation, are typically trained under a strict gravity-aligned assumption. Real-world captures, however, routinely violate it: handheld devices jitter and aerial platforms change attitude, so the camera is rarely upright. Under such 3D reorientation, standard spherical Transformers overfit global latitude cues and collapse. We introduce SO3UFormer, an architecture that learns intrinsic spherical features largely decoupled from the underlying coordinate frame, through three geometric components: (1) removing absolute latitude encoding, which breaks the dependence on the gravity axis; (2) quadrature-consistent spherical attention, which corrects for non-uniform sampling density; and (3) a gauge-aware relative positional bias built from local tangent-plane angles rather than global axes. A logit-space \emph{SO(3)}-consistency regularizer, used only during training, further suppresses residual discretization effects. To benchmark robustness, we introduce Pose35, a variant of Stanford2D3D perturbed by random rotations within $\pm 35^\circ$, and evaluate under a full, arbitrary \emph{SO(3)} stress test. There, the baseline SphereUFormer collapses from 67.53 \emph{mIoU} on Pose35 to 25.26 under the full \emph{SO(3)} test, whereas SO3UFormer reaches 72.03 on Pose35 and retains 70.67 under the same test. Similarly, on a second real-world dataset (Matterport3D) for segmentation and on panoramic depth estimation, SO3UFormer remains essentially rotation-invariant while the gravity-anchored baseline again loses most of its accuracy. Code and models are available at https://github.com/zhuqinfeng1999/SO3UFormer.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。