atlas

Research map / Video / Video segmentation and tracking

STDiff: Spatio-Temporal Diffusion for Continuous Stochastic Video Prediction

AAAI 2024 · Video › Video segmentation and tracking · approach: Stochastic video prediction · 15 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes STDiff, a stochastic video prediction model that uses a neural SDE over motion features (from frame differences via ConvGRU) plus an autoregressive image diffusion model, enabling continuous-time frame prediction with high diversity.

Problem

Stochastic video frame prediction is hard because future outcomes are uncertain; existing models have limited stochastic expressiveness along time and cannot predict at arbitrary (continuous) frame rates with good diversity.

Approach

Decompose video into motion and content; model temporal motion stochasticity with a neural SDE that can be integrated at arbitrary time points, and generate each frame with a conditional image diffusion model conditioned on the predicted motion feature and previous frame, so both spatial and temporal latents are SDE-described.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
use STDiff instead of prior methodsSMMNIST, 5 → 10 · SMMNISTFVD: 23.86 -> 14.93Helped
use STDiff instead of prior methodsBAIR, 2 → 28 · BAIRFVD: 93.6 -> 88.1Helped
use STDiff instead of prior methodsKITTI, 4 → 5 · KITTILPIPS: 260.5 -> 114.6Helped
use STDiff instead of prior methodsCityscapes, 2 → 28 · CityscapesFVD: 141.31 -> 107.31Helped
replace the SDE-based motion predictor with an ODE-based motion predictor (ablation)STDiff · KTH, 10 → 20iLPIPS: 25.08 -> 15.60Hurt
replace the SDE-based motion predictor with an ODE-based motion predictor (ablation)STDiff · KTH, 10 → 20FVD: 89.67 -> 90.68Hurt
replace the SDE-based motion predictor with an ODE-based motion predictor (ablation)STDiff · SMMNIST, 5 → 10iLPIPS: 136.27 -> 91.07Hurt
replace the SDE-based motion predictor with an ODE-based motion predictor (ablation)STDiff · SMMNIST, 5 → 10FVD: 14.93 -> 46.62Hurt
use STDiff instead of NPVP for continuous predictionBAIR, 2 → 28, 2× fps · BAIRFVD: 1159.14 -> 122.85Helped
use STDiff instead of NPVP for continuous predictionKITTI, 4 → 5, 2× fps · KITTILPIPS: 313.94 -> 160.92Helped

Surprising findings

Limitations

Experimental setup

Datasets: KITTI Cityscapes KTH BAIR SMMNIST

Benchmarks: KITTI Cityscapes KTH BAIR SMMNIST

Metrics: FVD SSIM LPIPS iLPIPS

Baselines: SVG-LP Hier-VRNN MCVD NPVP SAVP STMFANet VPTR-NAR FitVid PredNet Voxel Flow SADM DMVFN

Builds on: denoising diffusion probabilistic models neural SDE MCVD SPADE Vid-ODE

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy