HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Diverse Human Motion Prediction via Gumbel-Softmax Sampling from an Auxiliary Space

Lingwei Dang Yongwei Nie Chengjiang Long Qing Zhang Guiqing Li

Diverse Human Motion Prediction via Gumbel-Softmax Sampling from an Auxiliary Space

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

droliven/diverse_sampling
Official
pytorch
Mentioned in GitHub
Droliven/DHMP_jittor
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
human-pose-forecasting-on-human36mDiverseSampling
ADE: 370
APD: 15310
CMD: 11.692
FDE: 485
FID: 2.083
MMADE: 475
MMFDE: 516
human-pose-forecasting-on-humaneva-iDHMP
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.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Diverse Human Motion Prediction via Gumbel-Softmax Sampling from an Auxiliary Space | Papers | HyperAI