STDiff: Spatio-temporal Diffusion for Continuous Stochastic Video Prediction

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Ye, Xi, Bilodeau, Guillaume-Alexandre
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911778688991232
author Ye, Xi
Bilodeau, Guillaume-Alexandre
author_facet Ye, Xi
Bilodeau, Guillaume-Alexandre
contents Predicting future frames of a video is challenging because it is difficult to learn the uncertainty of the underlying factors influencing their contents. In this paper, we propose a novel video prediction model, which has infinite-dimensional latent variables over the spatio-temporal domain. Specifically, we first decompose the video motion and content information, then take a neural stochastic differential equation to predict the temporal motion information, and finally, an image diffusion model autoregressively generates the video frame by conditioning on the predicted motion feature and the previous frame. The better expressiveness and stronger stochasticity learning capability of our model lead to state-of-the-art video prediction performances. As well, our model is able to achieve temporal continuous prediction, i.e., predicting in an unsupervised way the future video frames with an arbitrarily high frame rate. Our code is available at \url{https://github.com/XiYe20/STDiffProject}.
format Preprint
id arxiv_https___arxiv_org_abs_2312_06486
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle STDiff: Spatio-temporal Diffusion for Continuous Stochastic Video Prediction
Ye, Xi
Bilodeau, Guillaume-Alexandre
Computer Vision and Pattern Recognition
Predicting future frames of a video is challenging because it is difficult to learn the uncertainty of the underlying factors influencing their contents. In this paper, we propose a novel video prediction model, which has infinite-dimensional latent variables over the spatio-temporal domain. Specifically, we first decompose the video motion and content information, then take a neural stochastic differential equation to predict the temporal motion information, and finally, an image diffusion model autoregressively generates the video frame by conditioning on the predicted motion feature and the previous frame. The better expressiveness and stronger stochasticity learning capability of our model lead to state-of-the-art video prediction performances. As well, our model is able to achieve temporal continuous prediction, i.e., predicting in an unsupervised way the future video frames with an arbitrarily high frame rate. Our code is available at \url{https://github.com/XiYe20/STDiffProject}.
title STDiff: Spatio-temporal Diffusion for Continuous Stochastic Video Prediction
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2312.06486