arXiv · 2312.06486
STDiff: Spatio-temporal Diffusion for Continuous Stochastic Video Prediction
Abstract
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}.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Xi Ye, Guillaume-Alexandre Bilodeau. 2023-12-11. STDiff: Spatio-temporal Diffusion for Continuous Stochastic Video Prediction. https://arxiv.org/abs/2312.06486
Cite the original work for its findings. Save a collection to share your selection of sources.