Command Palette
Search for a command to run...
Diverse Human Motion Prediction via Gumbel-Softmax Sampling from an Auxiliary Space
Lingwei Dang Yongwei Nie Chengjiang Long Qing Zhang Guiqing Li

Abstract
Diverse human motion prediction aims at predicting multiple possible future pose sequences from a sequence of observed poses. Previous approaches usually employ deep generative networks to model the conditional distribution of data, and then randomly sample outcomes from the distribution. While different results can be obtained, they are usually the most likely ones which are not diverse enough. Recent work explicitly learns multiple modes of the conditional distribution via a deterministic network, which however can only cover a fixed number of modes within a limited range. In this paper, we propose a novel sampling strategy for sampling very diverse results from an imbalanced multimodal distribution learned by a deep generative model. Our method works by generating an auxiliary space and smartly making randomly sampling from the auxiliary space equivalent to the diverse sampling from the target distribution. We propose a simple yet effective network architecture that implements this novel sampling strategy, which incorporates a Gumbel-Softmax coefficient matrix sampling method and an aggressive diversity promoting hinge loss function. Extensive experiments demonstrate that our method significantly improves both the diversity and accuracy of the samplings compared with previous state-of-the-art sampling approaches. Code and pre-trained models are available at https://github.com/Droliven/diverse_sampling.
Code Repositories
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| human-pose-forecasting-on-human36m | DiverseSampling | ADE: 370 APD: 15310 CMD: 11.692 FDE: 485 FID: 2.083 MMADE: 475 MMFDE: 516 |
| human-pose-forecasting-on-humaneva-i | DHMP | ADE@2000ms: 220 APD@2000ms: 6109 FDE@2000ms: 234 MMADE@2000ms: 342 MMFDE@2000ms: 316 |
Build AI with AI
From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.