用自回归扩散模型提升单目3D人体姿态的时序一致性与精度
StarPose: 3D Human Pose Estimation via Spatial-Temporal Autoregressive Diffusion
- 将2D到3D映射建模为自回归扩散过程,融合历史姿态信息
- 在Human3.6M上达到89.1%的P-MSE,时序一致性显著优于现有方法
- 适合需要高保真动作序列生成的场景,如动画、虚拟现实
单目3D人体姿态估计因深度模糊和遮挡问题仍具挑战性。相较于基于Transformer或卷积神经网络的传统方法,近期基于扩散模型的方法凭借其概率建模与高保真生成能力表现更优。然而,这些方法常忽视预测帧间的时空相关性,导致时序不一致且精度受限。为此,本文提出StarPose,一种自回归扩散框架,通过整合历史3D姿态预测与时空物理引导,显著提升姿态预测的准确性和时序一致性。不同于现有方法,StarPose将2D到3D的姿态映射建模为自回归扩散过程,利用历史姿态集成模块(HPIM)协同历史3D姿态与2D输入,生成富含信息的历史姿态嵌入,指导后续去噪步骤,确保时序一致性。此外,设计了即插即用的时空物理引导机制(STPG),以迭代方式优化去噪过程,进一步强化空间解剖合理性与时间运动动态性,实现鲁棒且逼真的姿态估计。在基准数据集上的大量实验表明,StarPose优于当前最先进方法,在3D人体姿态估计中兼具更高精度与时序一致性。代码已公开于https://github.com/wileychan/StarPose。
原文摘要 · Abstract (English)
Monocular 3D human pose estimation remains a challenging task due to inherent depth ambiguities and occlusions. Compared to traditional methods based on Transformers or Convolutional Neural Networks (CNNs), recent diffusion-based approaches have shown superior performance, leveraging their probabilistic nature and high-fidelity generation capabilities. However, these methods often fail to account for the spatial and temporal correlations across predicted frames, resulting in limited temporal consistency and inferior accuracy in predicted 3D pose sequences. To address these shortcomings, this paper proposes StarPose, an autoregressive diffusion framework that effectively incorporates historical 3D pose predictions and spatial-temporal physical guidance to significantly enhance both the accuracy and temporal coherence of pose predictions. Unlike existing approaches, StarPose models the 2D-to-3D pose mapping as an autoregressive diffusion process. By synergically integrating previously predicted 3D poses with 2D pose inputs via a Historical Pose Integration Module (HPIM), the framework generates rich and informative historical pose embeddings that guide subsequent denoising steps, ensuring temporally consistent predictions. In addition, a fully plug-and-play Spatial-Temporal Physical Guidance (STPG) mechanism is tailored to refine the denoising process in an iterative manner, which further enforces spatial anatomical plausibility and temporal motion dynamics, rendering robust and realistic pose estimates. Extensive experiments on benchmark datasets demonstrate that StarPose outperforms state-of-the-art methods, achieving superior accuracy and temporal consistency in 3D human pose estimation. Code is available at https://github.com/wileychan/StarPose.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。