Video Prediction at Multiple Scales with Hierarchical Recurrent Networks

17 Mar 2022  ·  Ani Karapetyan, Angel Villar-Corrales, Andreas Boltres, Sven Behnke ·

Autonomous systems not only need to understand their current environment, but should also be able to predict future actions conditioned on past states, for instance based on captured camera frames. For certain tasks, detailed predictions such as future video frames are required in the near future, whereas for others it is beneficial to also predict more abstract representations for longer time horizons. However, existing video prediction models mainly focus on forecasting detailed possible outcomes for short time-horizons, hence being of limited use for robot perception and spatial reasoning. We propose Multi-Scale Hierarchical Prediction (MSPred), a novel video prediction model able to forecast future possible outcomes of different levels of granularity at different time-scales simultaneously. By combining spatial and temporal downsampling, MSPred is able to efficiently predict abstract representations such as human poses or object locations over long time horizons, while still maintaining a competitive performance for video frame prediction. In our experiments, we demonstrate that our proposed model accurately predicts future video frames as well as other representations (e.g. keypoints or positions) on various scenarios, including bin-picking scenes or action recognition datasets, consistently outperforming popular approaches for video frame prediction. Furthermore, we conduct an ablation study to investigate the importance of the different modules and design choices in MSPred. In the spirit of reproducible research, we open-source VP-Suite, a general framework for deep-learning-based video prediction, as well as pretrained models to reproduce our results.

PDF Abstract

Results from the Paper

 Ranked #1 on Video Prediction on KTH (LPIPS metric)

     Get a GitHub badge
Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Video Prediction KTH MSPred LPIPS 0.029 # 1
PSNR 27.81 # 10
SSIM 0.951 # 1
MSE 23.18 # 1
Video Prediction Moving MNIST MSPred MSE 34.44 # 5
SSIM 0.975 # 1
LPIPS 0.024 # 1
PSNR 26.82 # 1
Video Prediction SynpickVP MSPred MSE 53.09 # 3
PSNR 27.89 # 1
SSIM 0.881 # 3
LPIPS 0.033 # 1