跳转至

Lifting Motion to the 3D World via 2D Diffusion

会议: CVPR 2025
arXiv: 2411.18808
代码: 无
领域: 3D视觉/运动估计
关键词: 3D运动估计, 2D扩散模型, 多视角一致性, 全局轨迹, 无3D监督

一句话总结

MVLift提出了一个多阶段框架,仅使用单视角2D姿态序列训练,通过线条件扩散模型→多视角优化→合成数据生成→多视角扩散模型的渐进策略建立多视角一致性,实现无需3D监督的全局3D运动(含关节旋转+根轨迹)估计,在AIST++上根轨迹误差67.6mm超越需要3D监督的WHAM (164.3mm)。

研究背景与动机

领域现状:从2D观测估计3D运动是长期研究挑战。MotionBERT和WHAM等SOTA方法需要在包含3D GT的动捕数据集上训练,泛化能力受限于训练数据覆盖的运动类型。一些方法尝试仅用2D姿态训练(如ElePose、MAS),但ElePose只处理单帧且不稳定,MAS不能预测全局根轨迹。

现有痛点:(1) 依赖3D GT训练的方法无法泛化到分布外运动(如复杂舞蹈、体操、动物运动),因为这些场景难以采集动捕数据;(2) 不依赖3D数据的方法无法预测世界坐标系中的全局根轨迹,只能估计局部姿态;(3) 单视角2D到3D存在严重的深度歧义,直接优化容易产生突变。

核心矛盾:仅有单视角2D数据时,缺乏直接的3D监督或多视角一致性约束来消除深度歧义,而全局3D运动估计(含世界坐标系中的根轨迹)又需要精确的深度和尺度信息。

本文目标 仅使用领域特定的单视角2D姿态序列(如人体/动物/交互),估计包含关节旋转和世界坐标系根轨迹的全局3D运动。

切入角度:作者观察到虽然单视角2D序列提供的3D信息有限,但在大量多样2D运动上训练的扩散模型可以学到关于不同视角下姿态外观的丰富先验。可以利用学到的2D扩散先验+几何约束渐进地建立多视角一致性,从而恢复3D运动。

核心 idea:通过2D运动扩散模型学到的先验结合对极几何约束,渐进地生成多视角一致的2D姿态序列,从而在无3D标注的情况下恢复全局3D运动。

方法详解

整体框架

四阶段渐进框架。Stage 1:训练线条件2D运动扩散模型——输入模拟的极线约束,生成沿极线分布的2D姿态序列,建立基本的成对几何一致性。Stage 2:多视角2D运动优化——联合优化5个未观察视角的2D序列,用SDS保证运动真实性+多视角一致性损失保证几何关系。Stage 3:合成数据生成——从Stage 2的多视角2D序列恢复3D运动,重投影到多个视角得到严格一致的多视角2D数据集。Stage 4:训练多视角2D运动扩散模型——在合成数据上训练,一次前向传播高效生成多视角一致2D序列,用于最终3D运动恢复。

关键设计

  1. 线条件扩散模型 (Line-Conditioned 2D Motion Diffusion):

    • 功能:生成遵循对极线约束的2D姿态序列,建立跨视角的基本几何一致性
    • 核心思路:定义线条件 \(\mathbf{L} \in \mathbb{R}^{T \times J \times 3}\),每个关节的线由系数 \((a_t^j, b_t^j, c_t^j)\) 参数化。训练时通过随机采样虚拟极点模拟极线约束(不需要真实多视角数据),添加Line Matching Loss \(\mathcal{L}_{\text{line}} = \sum_{t,j} |a_t^j \hat{x}_t^j + b_t^j \hat{y}_t^j + c_t^j|\) 确保预测关节落在对应极线上。使用Transformer架构,将线条件拼接到噪声姿态特征一起处理
    • 设计动机:对极几何提供了最基本的跨视角约束,通过在训练时随机模拟极线使模型学会在几何约束下生成合理的2D运动
  2. 多视角2D运动联合优化 (Multi-View Optimization):

    • 功能:从成对一致性提升到全局多视角一致性
    • 核心思路:设置6个摄像机视角(60°间隔环形排布),联合优化5个未观察视角的2D序列。两个优化目标:(1) SDS (Score Distillation Sampling)确保每个视角的2D序列符合学到的线条件扩散先验分布;(2) 多视角一致性损失遍历所有 \(\binom{6}{2}=15\) 个视角对,约束每对视角间的极线距离为零。SDS梯度为 \(\nabla \mathcal{L}_{\text{SDS}} = \mathbb{E}[\omega(n)(\epsilon_\theta - \epsilon)]\)
    • 设计动机:Stage 1的极线仅约束成对一致性,不保证全局一致;通过在所有视角对上施加几何约束+用SDS保持运动真实性的联合优化,显著提升多视角一致性
  3. 合成数据生成与多视角扩散模型:

    • 功能:将优化方法转化为高效的前向生成模型
    • 核心思路:Stage 3从优化得到的多视角2D序列恢复3D关节位置(最小化重投影误差),拟合SMPL参数(用VPoser),重投影到4个90°间隔视角得到严格一致的多视角数据。Stage 4在这些数据上训练多视角扩散模型:输入一个视角的2D序列,同时生成3个其他视角的2D序列。网络在每个Transformer block的自注意力后加跨视角注意力层实现视角间信息交换
    • 设计动机:优化方法慢且不能保证完美一致性,通过"优化→合成数据→训练生成模型"的self-training式管线,将慢速优化蒸馏为高效的前向推理

损失函数 / 训练策略

Stage 1:\(\mathcal{L} = \mathbb{E}[\|\hat{X}_\theta - X_0\|_1] + \mathcal{L}_{\text{line}}\)(去噪重建+线匹配)。Stage 2:SDS损失+多视角一致性损失。Stage 4:多视角扩散模型的去噪重建损失。3D恢复通过最小化重投影误差+VPoser SMPL拟合。

实验关键数据

主实验

AIST++ 数据集(有3D GT):

方法 需要3D监督 \(T_{\text{root}}\) MPJPE↓ PA-MPJPE↓
MotionBERT 101.6 134.0 108.6
WHAM 164.3 104.8 75.1
ElePose N/A 269.4 215.1
MAS N/A 191.1 155.6
SMPLify 77.4 171.6 146.7
MVLift 67.6 110.7 79.2

OMOMO 人物-物体交互数据集:

方法 \(T_{\text{root}}\) MPJPE↓ \(T_{\text{root}}^O\) O-MPJPE↓
SMPLify 97.9 142.0 751.8 106.7
MVLift 54.9 67.0 172.9 76.9

消融实验

配置 \(T_{\text{root}}\) MPJPE↓ PA-MPJPE↓ \(J_{2D}\)
MVLift-Stage 1 73.1 135.2 104.4 31.0
MVLift-Stage 2 65.3 127.4 96.2 19.7
SDS for 3D 72.9 137.3 103.5 25.2
SDS for 3D w/o \(l_{\text{epi}}\) 752.3 230.4 186.2 54.9
MVLift (final) 67.6 110.7 79.2 14.0

关键发现

  • 每个stage都带来显著提升:Stage 1→Stage 2 MPJPE从135.2降到127.4,最终模型降到110.7
  • 不用线条件(无极线约束)的SDS directly在3D上优化完全崩溃(MPJPE 230.4),证明极线约束是成功的关键
  • MVLift在根轨迹精度上(67.6mm)甚至超越了使用3D监督的WHAM(164.3mm)和MotionBERT(101.6mm)
  • 人类感知实验中39%参与者认为MVLift生成的结果优于GT,11%认为无法区分
  • 方法成功泛化到动物姿态(CatPlay)和人物-物体交互(OMOMO),证明领域通用性

亮点与洞察

  • "渐进建立多视角一致性"的四阶段设计堪称精妙——从成对极线约束(Stage 1)→全局多视角优化(Stage 2)→合成严格一致数据(Stage 3)→高效前向生成(Stage 4),每个阶段解决前一阶段的局限,最终将慢速优化蒸馏为快速推理。这种渐进式self-training策略可广泛用于缺乏直接监督信号的任务
  • 仅用2D数据训练却超越3D监督方法——在根轨迹指标上超越WHAM和MotionBERT,说明多视角一致性约束比直接3D回归可能更有效地消除深度歧义
  • 领域通用性强——同一框架适用于人体、动物、人物-物体交互,只需要领域特定的2D姿态序列,不需要任何骨架或模板先验(动物用SMAL/人用SMPL仅在后处理中用)

局限与展望

  • 四阶段训练流程较复杂,Stage 2的优化尤其耗时
  • Stage 2的优化结果不能保证与输入2D序列完全对齐,只保证运动真实性
  • 合成数据中的3D运动质量直接影响Stage 4模型的上限
  • 未处理遮挡情况下的关节可见性问题
  • 相机排列策略(等间隔环形)比较固定,实际场景中的相机分布可能更复杂
  • 可以探索端到端训练替代多阶段流水线,减少误差累积

相关工作与启发

  • vs WHAM: WHAM在AMASS等3D数据集上训练全局轨迹预测模块,对分布外运动泛化差;MVLift无需3D数据在根轨迹上反超2.4倍,但MPJPE略逊(110.7 vs 104.8)
  • vs MAS: MAS用无条件2D运动扩散模型优化3D运动,但不能生成一致的多视角序列且不支持全局轨迹;MVLift通过线条件和多视角一致性约束解决了这两个问题
  • vs ElePose: ElePose用重投影损失+2D姿态先验同时预测3D姿态和相机,但训练不稳定常产生不真实姿态;MVLift通过扩散模型先验更好地约束了生成质量
  • 启发:多视角一致性的渐进建立思路可以迁移到3D物体重建等任务

评分

  • 新颖性: ⭐⭐⭐⭐⭐ 首次实现无3D监督的full global 3D motion估计,多阶段渐进策略设计精巧
  • 实验充分度: ⭐⭐⭐⭐⭐ 5个数据集+3个领域(人/动物/交互)+人类感知实验+详细消融
  • 写作质量: ⭐⭐⭐⭐ 四阶段框架清晰,但细节较多需要仔细阅读
  • 价值: ⭐⭐⭐⭐⭐ 解决了长期的"2D→3D运动lifting without 3D supervision"问题,具有广泛应用前景

相关论文