MoSE3: Learning World-Space SE(3) at Every Pixel

CV Monocular or Binocular Depth Estimation and Occlusion Handling VSER
稠密三维点跟踪一直是动态场景运动建模的重要范式,但一条点轨迹仅对应每个像素的三维平移曲线(即仅含3个自由度):它只能刻画像素的位置变化,却无法反映其底层部件的旋转运动,也无法揭示哪些像素作为一个刚体共同运动。本文提出MoSE3——首个前馈式模型,能够从单目RGB视频中直接预测稠密的SE(3)运动,为世界坐标系中的每个像素输出完整的六自由度刚体变换。逐像素的SE(3)运动描述提供了对场景运动更丰富的理解:它同时涵盖旋转、平移与像素分组信息。然而,直接预测SE(3)面临两大挑战:其一,旋转位于非欧几里得的弯曲流形上,难以通过常规的欧氏空间回归方法进行建模;其二,SE(3)真值标注本身极难获取。为应对上述挑战,MoSE3通过两个联合学习的中间表征——三维点轨迹与刚性嵌入(rigidity embeddings)——间接预测逐像素SE(3)运动;进而,在每个“软刚性”聚类(soft rigid cluster)内,以可微分方式拟合刚体变换,从而实现端到端的预测与监督训练。为弥补真实数据标注的缺失,我们构建了Art-Kubric——一个大规模合成数据集,其中包含具备丰富物理交互行为的关节式物体的稠密SE(3)真值及刚性标注。实验表明,MoSE3在刚体与关节式两类基准数据集上,于像素级、部件级和物体级SE(3)估计任务中均达到当前最优性能;同时,在三个数据集上的平均三维点跟踪精度亦位居当前最优。尤为值得注意的是,尽管MoSE3仅在合成运动数据上进行训练,其在真实世界视频上仍展现出卓越的泛化能力。
Dense 3D point tracking has been a prominent paradigm for modeling motion in dynamic scenes, but a point track is just a 3-DoF translation curve per pixel: it captures where pixels go, not the rotation of the underlying part, nor which pixels move together as one body. We propose MoSE3, the first feed-forward model that predicts dense SE(3) motion from monocular RGB video, producing full 6-DoF rigid transforms at every pixel in world space. Per-pixel SE(3) motion offers a richer view of how a scene moves: rotation, translation, and grouping all at once. Directly predicting SE(3) is challenging: rotations lie on a curved manifold that is ill-suited to Euclidean regression, and annotations for SE(3) are particularly difficult to acquire. To address these challenges, MoSE3 predicts per-pixel SE(3) through two jointly learned intermediates, 3D point tracks and rigidity embeddings, and recovers SE(3) by differentiably fitting transforms within each soft rigid cluster, enabling end-to-end prediction and supervision. To close the data gap, we introduce Art-Kubric, a large-scale synthetic dataset with dense SE(3) and rigidity labels for articulated objects with rich physical interactions. MoSE3 achieves state-of-the-art SE(3) estimation at pixel, part, and object levels on both rigid and articulated benchmarks, and state-of-the-art average 3D point tracking accuracy across three datasets, while showing strong generalization to real-world videos despite being trained solely on synthetic motion data.
许愿