Source-linked AI summary

MotionMixer: MLP-based 3D Human Body Pose Forecasting

Arij Bouazizi, Adrian Holzbock, Ulrich Kressel, Klaus Dietmayer, Vasileios Belagiannis

arXiv:2207.00499v1cs.CVcs.LG

TL;DR

Forecasting future 3D human poses from past poses is challenging because prior-based methods struggle with complex body dynamics. MotionMixer uses spatial and temporal MLPs with squeeze-and-excitation blocks, and reports improved short- and long-term forecasting on three benchmark datasets with fewer parameters than current best-performing approaches.

  • Problem

    Prior-based approaches struggle to forecast reliable 3D poses amid complex biomechanics and may not generalize to new environments and subjects.

    Source: §1 · p. 1
  • Method

    MotionMixer mixes spatial and temporal pose features with MLPs, then decodes future motion, using squeeze-and-excitation blocks to calibrate time-step influence.

    Source: §1 · p. 2, §3.1 · p. 3, Spatial-Temporal Mixing · p. 3
  • Results

    MotionMixer improved short- and long-term forecasting on three large-scale benchmark datasets and has fewer parameters than current best-performing approaches.

    Source: §5 · p. 7
  • Takeaways & Limitations

    The reported results support MLP-based spatial-temporal mixing as an effective approach to 3D human pose forecasting across the evaluated benchmarks.

    Source: §5 · p. 7

    Human motion is highly uncertain, and long-term inter-joint and inter-frame dependencies are complex; the WalkDog example also shows a mismatch in the last three predicted frames.

    Source: §4.6 · p. 7

Abstract

from arXiv · show

In this work, we present MotionMixer, an efficient 3D human body pose forecasting model based solely on multi-layer perceptrons (MLPs). MotionMixer learns the spatial-temporal 3D body pose dependencies by sequentially mixing both modalities. Given a stacked sequence of 3D body poses, a spatial-MLP extracts fine grained spatial dependencies of the body joints. The interaction of the body joints over time is then modelled by a temporal MLP. The spatial-temporal mixed features are finally aggregated and decoded to obtain the future motion. To calibrate the influence of each time step in the pose sequence, we make use of squeeze-and-excitation (SE) blocks. We evaluate our approach on Human3.6M, AMASS, and 3DPW datasets using the standard evaluation protocols. For all evaluations, we demonstrate state-of-the-art performance, while having a model with a smaller number of parameters. Our code is available at: https://github.com/MotionMLP/MotionMixer

1 Introduction

3D human motion forecasting predicts future body poses from past poses, but prior-based methods struggle with complex biomechanics and generalization. MotionMixer uses spatial and temporal MLPs to model body-pose dependencies, with an efficient architecture evaluated on three large-scale datasets.

1 Introduction

  • Prior-based methods struggle to forecast reliable 3D poses because of complex biomechanics, and expert knowledge does not ensure generalization to new environments and subjects.Forecasting motion also supports applications such as virtual reality, autonomous driving, and robotics.Source: §1 · p. 1
  • RNN-based approaches model temporal correlations between joints, but the passage notes challenges including gradient problems, first-frame discontinuity, and stationary predictions at shorter forecast horizons.The passage also notes difficulty processing longer input sequences.Source: §1 · pp. 1–2
  • MotionMixer alternates MLP operations across body poses and time steps to model spatial-temporal dependencies and capture structural and temporal relationships explicitly.The operations allow direct access to current and past information.Source: §1 · p. 2
  • The paper proposes a spatial-temporal MLP for jointly modeling body-joint locations and temporal dependencies, and reports state-of-the-art short- and long-term forecasting on three large-scale datasets.The authors also identify reduced computational cost as a contribution.Source: §1 · p. 2

2 Related Work

Prior human-motion forecasting work uses recurrent, convolutional, graph, and attention architectures to capture temporal or spatial-temporal dependencies. MotionMixer instead explores an MLP-only architecture for human dynamics.

Recurrent-based Motion Prediction

  • RNN methods use encoder-decoder structures, residual connections, or motion-context modeling, while recurrent-free backbones for human motion prediction are described as rarely studied.The cited approaches include [Fragkiadaki et al., 2015], [Martinez et al., 2017], and.Source: §2 · p. 2

Convolutional-based Motion Prediction

  • CNN-based motion-prediction approaches have drawn interest for their hierarchical structure and effectiveness at capturing spatial and temporal correlations, but convolution-based approaches remain bottlenecked by quite high computational cost.[Mao et al., 2019] converted the joint trajectory to the frequency domain to handle temporal information.Source: §2 · p. 2

Attention- and MLP-based Architectures

  • Attention methods model relationships among poses but incur increasing memory overhead with longer histories; MLP-Mixer replaces self-attention with MLPs, while MotionMixer targets human spatial-temporal dynamics.The passage describes attention approaches in [Mao et al., 2020] and [Aksan et al., 2020], and MLP-Mixer in [Tolstikhin et al., 2021].Source: §2 · p. 2

3 Method

MotionMixer maps a history of 3D body poses to future poses using a pure MLP-based network. It embeds each pose, mixes spatial and temporal features, then decodes the mixed features into predictions.

3.1 MotionMixer

  • MotionMixer couples pose embedding and spatial-temporal mixing in repeated STMixer blocks, then uses a two-layer fully connected decoder to forecast future motion.Each block contains spatial and temporal MLPs with skip connections.Source: §3.1 · p. 3

Pose Prediction

  • The decoder aggregates the mixed features into a global vector and applies an MLP to forecast future motion.The decoder projects each mixed feature to a vector of length Tf using a two-layer nonlinear feedforward network.Source: §3.1 · p. 3, Pose Prediction · p. 3

Pose Embedding

  • Each observed skeleton is flattened to length K = 3×J and linearly projected to hidden dimension C to form the learnable pose embedding.The projection is a single fully connected layer.Source: Pose Embedding · p. 3

Spatial-Temporal Mixing

  • Spatial mixing models dependencies among body-joint embeddings, while temporal mixing models long-range interactions across frames.The temporal operation is shared across rows representing individual joints; SE blocks reweight the influence of each time step after each mixing operation.Source: Spatial-Temporal Mixing · p. 3

3.2 Training

  • Training predicts future joint displacements and adds them to the most recent pose to produce the full-body pose sequence.The stated losses include MPJPE and an angle-based loss comparing predicted joint angles with ground truth in the exponential-map representation.Source: §3.2 · p. 4

4 Experimental Evaluation

MotionMixer is evaluated on Human3.6M, AMASS, and 3DPW using standard pose-error measures and comparisons with recurrent, convolutional, graph-convolutional, and attention-based methods. It reports strong benchmark results, ablations of its architecture, and a limitation involving uncertain long-term motion.

4.1 Datasets and Metrics

  • Human3.6M uses 22 joints for 3D poses, AMASS uses 18, and the official 3DPW test set measures generalization from AMASS training.Evaluation reports joint-angle distance and 3D MPJPE, including particular-frame and average-over-frames measures.Source: §4.1 · pp. 4–5

4.4 Results

  • MotionMixer leads average-frame Human3.6M results, performs best short-term on AMASS, and outperforms prior approaches for long-term AMASS prediction and untuned 3DPW transfer.On Human3.6M, its gain over the second-best approach ranges from 13% at 400ms to 5% at 1000ms; AMASS gains range from 36% at 80ms to 4% at 1000ms.Source: Table 1 · p. 5, §4.4 · pp. 5–6, Table 2 · p. 6

4.2 Implementation Details

  • The model uses three STMixer blocks with 60 channels, dropout 0.1, and Adam training for 50 epochs.The learning rate starts at 10^-2 and decays by a factor of 0.1 every 10 epochs; batch sizes are 50 for Human3.6M and 256 for AMASS.Source: §4.2 · p. 5

4.3 Baselines

  • Comparisons cover nine recurrent, convolutional, graph-convolutional, and attention-based methods at 400ms and 1000ms prediction horizons.These correspond to 10-frame short-term and 25-frame long-term forecasts.Source: §4.3 · p. 5

4.5 Ablation Studies

  • Predicting pose displacement improves performance by 5mm over direct 3D-joint prediction, while removing temporal or spatial mixing raises 1000ms error by 4% or 6%, respectively.The best layer count is N = 3; stacking more layers did not empirically improve performance.Source: §4.5 · p. 6, Figure 3 · p. 6, Table 4 · p. 7
  • SE blocks yield a 2mm gain across horizons, and MotionMixer reaches nearly the performance of Mao et al. with 1.4% of its parameters.It also outperforms Sofianos et al. and Dang et al. by 4% and 3%, respectively, using 40% and 0.5% of their parameters.Source: §4.5 · p. 7, Table 4 · p. 7, Table 5 · p. 7

4.6 Limitations

  • Long-term motion remains uncertain: the final three predicted WalkDog frames mismatch ground truth, a failure also seen in previous methods.The paper notes that past poses can imply multiple possible futures, making long-term inter-joint and inter-frame dependencies more complex.Source: §4.6 · p. 7

5 Conclusion

MotionMixer forecasts 3D body poses by mixing spatial and temporal features with an MLP-based approach. It improves short- and long-term forecasting results across three benchmark datasets while using fewer parameters than current best-performing approaches.

  • MotionMixer mixes features across spatial and temporal domains to exploit 3D human body pose dependencies.Source: §5 · p. 7
  • The method improves state-of-the-art short- and long-term forecasting on three large-scale benchmark datasets.Source: §5 · p. 7
  • Squeeze-and-excitation blocks calibrate the influence of each time step, while the model uses fewer parameters than current best-performing approaches.Source: §5 · p. 7
Loading 2207.00499v1…