Source-linked AI summary
DiffPose: SpatioTemporal Diffusion Model for Video-Based Human Pose Estimation
Runyang Feng, Yixing Gao, Tze Ho Elden Tse, Xueqing Ma, Hyung Jin Chang
TL;DR
Video pose estimation needs temporal information and representations focused on joint regions, but adapting diffusion models to those needs remains unresolved. DiffPose generates pose heatmaps through conditioned diffusion, and the paper reports state-of-the-art results on three benchmarks. Its sampling supports pose ensembling and adjustable refinement steps without retraining.
Problem
Adapting diffusion models to video pose estimation remains unclear because videos require temporal modeling and accurate localization requires attention to joint regions.
Source: §1 · p. 2Method
DiffPose generates keypoint heatmaps through conditional diffusion, using spatiotemporal features and multi-scale interactions between local joints and global context.
Source: §1 · p. 2Results
DiffPose achieves state-of-the-art results on PoseTrack2017, PoseTrack2018, and PoseTrack21.
Source: §1 · p. 2Takeaways & Limitations
DiffPose can combine predictions from multiple noise samples and adjust refinement steps without retraining, with ensemble benefits noted for challenging joints.
Source: §1 · p. 2, §3.5 · p. 5
Abstract
from arXiv · showhide
Denoising diffusion probabilistic models that were initially proposed for realistic image generation have recently shown success in various perception tasks (e.g., object detection and image segmentation) and are increasingly gaining attention in computer vision. However, extending such models to multi-frame human pose estimation is non-trivial due to the presence of the additional temporal dimension in videos. More importantly, learning representations that focus on keypoint regions is crucial for accurate localization of human joints. Nevertheless, the adaptation of the diffusion-based methods remains unclear on how to achieve such objective. In this paper, we present DiffPose, a novel diffusion architecture that formulates video-based human pose estimation as a conditional heatmap generation problem. First, to better leverage temporal information, we propose SpatioTemporal Representation Learner which aggregates visual evidences across frames and uses the resulting features in each denoising step as a condition. In addition, we present a mechanism called Lookup-based MultiScale Feature Interaction that determines the correlations between local joints and global contexts across multiple scales. This mechanism generates delicate representations that focus on keypoint regions. Altogether, by extending diffusion models, we show two unique characteristics from DiffPose on pose estimation task: (i) the ability to combine multiple sets of pose estimates to improve prediction accuracy, particularly for challenging joints, and (ii) the ability to adjust the number of iterative steps for feature refinement without retraining the model. DiffPose sets new state-of-the-art results on three benchmarks: PoseTrack2017, PoseTrack2018, and PoseTrack21.
1. Introduction
The introduction frames diffusion-based video pose estimation as underdeveloped because existing perception models are static and keypoint-focused representations remain unresolved. DiffPose conditions heatmap denoising on spatiotemporal features and models interactions between joint regions and global context.
- The paper identifies temporal dependencies and attention to body-joint regions as two unresolved challenges for adapting diffusion models to video pose estimation.Source: §1 · p. 2
- STRL extracts spatial information within frames, integrates cross-frame knowledge with cascaded Transformers, and supplies its features as a fixed condition at each denoising step.Source: §1 · p. 2
- LMSFI uses noisy heatmaps to retrieve local joint features and models their multi-scale correlations with global context to focus representations on keypoint regions.The introduction reports that its representations converge consistently around keypoint regions.Source: §1 · p. 2
- Multiple sampled pose estimates can be combined for greater robustness, while the number of refinement steps can change without retraining.The introduction highlights challenging joints such as wrists and ankles for ensemble gains.Source: §1 · p. 2
- DiffPose achieves state-of-the-art results on PoseTrack2017, PoseTrack2018, and PoseTrack21.Source: §1 · p. 2
- DiffPose is presented as the first diffusion model for multi-frame human pose estimation, formulated as conditional generation of keypoint heatmaps.It progressively adds Gaussian noise to ground-truth heatmaps and denoises them with a Pose-Decoder.Source: §1 · p. 2
2. Related Work
Related work describes heatmap prediction as a popular pose-estimation paradigm and reviews video methods that aggregate information across frames. It also situates DiffPose among diffusion-based perception methods while noting the absence of prior successful diffusion adaptations for multi-frame pose estimation.
- Heatmap-based keypoint representations have gained popularity because they are associated with faster optimization convergence.Source: §2 · p. 2
- Earlier video pose methods merge consecutive-frame heatmaps [37] or use deformable convolutions for implicit motion compensation [38].Source: §2 · p. 2
- These video methods typically predict deterministic poses and may lack recalibration, leaving challenging joints vulnerable to localized detection failures.Source: §2 · p. 3
- Diffusion models have been applied to image segmentation [1], object detection [10], and instance segmentation [24], but the paper reports no prior successful adaptation to multi-frame human pose estimation.Source: §2 · p. 3
3. Our Approach
DiffPose casts video-based keyframe pose estimation as conditional heatmap denoising, using spatiotemporal features and a multiscale Pose-Decoder to produce pose estimates. Its diffusion process supports iterative refinement and complementary pose ensembling.
3.1. Preliminaries
- DiffPose trains a Pose-Decoder to recover ground-truth keypoint heatmaps from noisy heatmaps, conditioned on spatiotemporal features extracted from a cropped video sequence.The sequence contains crops of the same person across consecutive frames around a keyframe.Source: §3.1 · p. 3
3.2. Spatiotemporal Representation Learner
- STRL extracts spatial features per frame, embeds and combines frame tokens, then uses cascaded Transformers and an MLP to produce spatiotemporal features.These features provide temporal context to the Pose-Decoder.Source: §3.2 · pp. 3–4
3.3. Pose-Decoder
- LMSFI uses noisy heatmaps to look up local joint features, models their interaction with global context across scales, and produces representations focused on joint regions.The decoder uses Transformer interactions at low resolution and convolutions at higher resolutions; the resulting scale features are integrated to predict a heatmap.Source: §3.3 · pp. 4–5
3.4. Overall Training and Inference Algorithms
- Training corrupts ground-truth heatmaps and supervises denoising with mean-square error; inference progressively refines noisy heatmaps using DDIM updates.At each sampling step, the decoder takes a random noisy heatmap or the preceding step's predicted heatmap as input.Source: §3.4 · p. 5
3.5. Discussion
- DiffPose can average predictions from multiple noisy initializations and adjust inference sampling steps without retraining; additional steps yield more delicate representations that foster accurate pose estimation.The authors report that complementary pose ensembling improves performance, especially for challenging joints.Source: §3.5 · pp. 5–6
4. Experiments
DiffPose is evaluated on three PoseTrack benchmarks, with experiments testing its components, inference choices, and performance in challenging scenes. It achieves the best reported results on all three benchmarks.
4.1. Experimental Settings
- Experiments use PoseTrack2017, PoseTrack2018, and PoseTrack21, evaluate mean average precision, and run inference with 10 noise groups and 4 denoising steps.The implementation uses 256×192 inputs and terminates training within 60 epochs.Source: §4.1 · p. 6
4.2. Comparison with State-of-the-art Approaches
- 86.4 mAP on PoseTrack2017, 83.0 mAP on PoseTrack2018, and 82.9 (↑1.7) mAP over FAMI-Pose on PoseTrack21.On PoseTrack2017, DiffPose gains 1.6 mAP over FAMI-Pose; the PoseTrack2018 results are reported as outperforming all other approaches.Source: §4.2 · pp. 6–7
4.3. Ablation Study
- LMSFI raises Direct-single from 52.5 to 82.4 mAP, and adding spatiotemporal cues improves performance by a further 4.0 mAP.The component ablation attributes the first increase to LMSFI and the further gain to the full DiffPose setting.Source: §4.3 · p. 7, Table 4 · p. 7
- Multi-scale fusion yields 86.4 (↑2.0) mAP, while concatenation and aggregation yields 86.2 mAP.The comparison tests designs within LMSFI; the reported change between the two fusion approaches is marginal.Source: §4.3 · p. 7, Table 5 · p. 7
- Ensembling predictions from 1, 5, and 10 initial noises raises mAP from 80.4 to 84.0 to 86.4, with wrist and ankle gains of 8.9 and 6.1 mAP.The authors attribute the improvement to averaging plausible pose predictions from different noise inputs.Source: §4.3 · pp. 7–8, Table 6 · p. 8
- More denoising iterations improve performance as features are progressively refined toward keypoint regions.The study compares 1, 2, and 4 steps; the supplied passage reports the trend but not the individual scores.Source: §4.3 · p. 8, Table 7 · p. 8
4.4. Qualitative Analyses on DiffPose
- Visualizations contrast DiffPose activations and pose estimates with baselines, and report more robust predictions in challenging scenes.The examples include fast motion and mutual occlusion, with inaccurate detections highlighted in red circles.Source: Figure 3 · p. 7, §4.4 · p. 8, Figure 4 · p. 8
5. Conclusion and Future Works
DiffPose treats multi-frame pose estimation as conditional generation of keypoint heatmaps, using temporal features and keypoint-focused interactions. It achieves state-of-the-art results on three benchmarks, with future work proposed for other pose tasks and faster inference.
Conclusion
- DiffPose combines STRL for temporal cues and LMSFI for keypoint-region attention in conditional heatmap generation.The conclusion describes flexible pose ensembling and iterative refinement as improving performance without retraining.Source: §5 · p. 8
- Evaluations on PoseTrack2017, PoseTrack2018, and PoseTrack21 show state-of-the-art performance.The authors identify 3D human pose estimation, pose tracking, and accelerated inference as future directions.Source: §5 · p. 8
Training and Inference Algorithms
Training corrupts ground-truth heatmaps and trains the Pose-Decoder to reverse that process. At inference, the decoder iteratively refines noisy or previously predicted heatmaps, with DDIM updating the next step.
Training
- Training corrupts ground-truth heatmaps with Gaussian noise and trains the Pose-Decoder to reverse the diffusion process.The training algorithm takes an image sequence and ground-truth heatmap as inputs.Source: Training and Inference Algorithms · p. 13
Inference
- At inference, the Pose-Decoder maps random noisy or previously predicted heatmaps to current-step estimates, and DDIM updates the heatmaps for the next step.The number of iterative denoising steps is controlled by the parameter steps.Source: Training and Inference Algorithms · p. 13
- The inference procedure takes an image sequence, steps, and T, initializes Gaussian noise, and returns a predicted heatmap.The algorithm initializes spatiotemporal features with STRL before sampling.Source: Training and Inference Algorithms · p. 13
Supplementary Experiments
The supplementary experiments examine how the STRL temporal interval affects performance and show qualitative predictions across three PoseTrack benchmarks, including challenging scenes.
Study on temporal interval δ
- 82.4 mAP at δ = 0 rises to 84.5 at δ = 1 and 86.4 at δ = 2; the authors choose δ = 2 to balance performance and computation.The ablation varies the number of supporting frames in STRL; the table caption identifies the comparison as a temporal-interval ablation.Source: Table 8 · p. 13, Supplementary Experiments · p. 13
Visualized results
- Figures 7–9 show DiffPose predictions on PoseTrack2017, PoseTrack2018, and PoseTrack21 in challenging scenes such as fast motion and occlusion.The authors describe the estimates as accurate and robust across these scenes.Source: Supplementary Experiments · p. 13, Figure 7 · p. 14, Figure 8 · p. 15
- Figure 6 presents video comparisons, pose ensembling, and gradual denoising, with inaccurate estimates highlighted in its comparison panel.“DiffPose-M.” denotes multiple pose predictions in pose ensembling.Source: Supplementary Experiments · p. 13, Figure 6 · p. 14
Related papers
This paper cites
- Attention Is All You Need2017
- An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale2020
- U-Net: Convolutional Networks for Biomedical Image Segmentation2015
- Swin Transformer: Hierarchical Vision Transformer using Shifted Windows2021
- Diffusion Models Beat GANs on Image Synthesis2021
- Deep Unsupervised Learning using Nonequilibrium Thermodynamics2015