Greedy Hierarchical Variational Autoencoders for Large-Scale Video Prediction
Not on the map. Why?
Abstract (arXiv)
A video prediction model that generalizes to diverse scenes would enable intelligent agents such as robots to perform a variety of tasks via planning with the model. However, while existing video prediction models have produced promising results on small datasets, they suffer from severe underfitting when trained on large and diverse datasets. To address this underfitting challenge, we first observe that the ability to train larger video prediction models is often bottlenecked by the memory constraints of GPUs or TPUs. In parallel, deep hierarchical latent variable models can produce higher quality predictions by capturing the multi-level stochasticity of future observations, but end-to-end optimization of such models is notably difficult. Our key insight is that greedy and modular optimization of hierarchical autoencoders can simultaneously address both the memory constraints and the optimization challenges of large-scale video prediction. We introduce Greedy Hierarchical Variational Autoencoders (GHVAEs), a method that learns high-fidelity video predictions by greedily training each level of a hierarchical autoencoder. In comparison to state-of-the-art models, GHVAEs provide 17-55% gains in prediction performance on four video datasets, a 35-40% higher success rate on real robot tasks, and can improve performance monotonically by simply adding more modules.
Key facts. Cited by 75 papers in the map (data through 7 Oct 2026). Builds on 60 of its 87 references that are in the map. First posted on arXiv 6 Mar 2021.
Builds on
60 of 87 references.
References · 60
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2018The Unreasonable Effectiveness of Deep Features as a Perceptual Metriccited by 11,029
- 2020Denoising Diffusion Implicit Modelscited by 9,655
- 2016The Cityscapes Dataset for Semantic Urban Scene Understandingcited by 4,427
- 2017Progressive Growing of GANs for Improved Quality, Stability, and Variationcited by 2,967
- 2018Towards Accurate Generative Models of Video: A New Metric & Challengescited by 1,204
- 2016Conditional Image Generation with PixelCNN Decoderscited by 859
- 2020NVAE: A Deep Hierarchical Variational Autoencodercited by 635
- 2019Multi-Object Representation Learning with Iterative Variational Inferencecited by 359
- 2016Dynamic Filter Networkscited by 318
- 2016Ladder Variational Autoencoderscited by 315
38 more
- 2015Deep multi-scale video prediction beyond mean square errorcited by 297
- 2016Deep Visual Foresight for Planning Robot Motioncited by 276
- 2019RoboNet: Large-Scale Multi-Robot Learningcited by 268
- 2016Unsupervised Learning for Physical Interaction through Video Predictioncited by 256
- 2018Stochastic Video Generation with a Learned Priorcited by 244
- 2017Stochastic Variational Video Predictioncited by 234
- 2019GENESIS: Generative Scene Inference and Sampling with Object-Centric Latent Representationscited by 227
- 2018Visual Foresight: Model-Based Deep Reinforcement Learning for Vision-Based Robotic Controlcited by 220
- 2017Video Frame Synthesis using Deep Voxel Flowcited by 209
- 2019Contrastive Learning of Structured World Modelscited by 206
- 2016Deep Predictive Coding Networks for Video Prediction and Unsupervised Learningcited by 203
- 2018RoboTurk: A Crowdsourcing Platform for Robotic Skill Learning through Imitationcited by 202
- 2018Stochastic Adversarial Video Predictioncited by 173
- 2015Action-Conditional Video Prediction using Deep Networks in Atari Gamescited by 171
- 2017Self-Supervised Visual Planning with Temporal Skip Connectionscited by 137
- 2018Learning to Decompose and Disentangle Representations for Video Predictioncited by 120
- 2018Greedy Layerwise Learning Can Scale to ImageNetcited by 113
- 2019Entity Abstraction in Visual Model-Based Reinforcement Learningcited by 110
- 2019BIVA: A Very Deep Hierarchy of Latent Variables for Generative Modelingcited by 109
- 2016Video Pixel Networkscited by 108
- 2020Stochastic Latent Residual Video Predictioncited by 106
- 2019Improved Conditional VRNNs for Video Predictioncited by 104
- 2019High Fidelity Video Prediction with Large Stochastic Recurrent Neural Networkscited by 88
- 2016An Uncertain Future: Forecasting from Static Images using Variational Autoencoderscited by 86
- 2019Putting An End to End-to-End: Gradient-Isolated Learning of Representationscited by 85
- 2019Hierarchical Foresight: Self-Supervised Learning of Long-Horizon Tasks via Visual Subgoal Generationcited by 83
- 2017Dual Motion GAN for Future-Flow Embedded Video Predictioncited by 69
- 2018Latent Space Policies for Hierarchical Reinforcement Learningcited by 66
- 2017Towards Deeper Understanding of Variational Autoencoding Modelscited by 56
- 2019COBRA: Data-Efficient Model-Based RL through Unsupervised Object Discovery and Curiosity-Driven Explorationcited by 55
- 2019Deep Visual MPC-Policy Learning for Navigationcited by 54
- 2018Reasoning About Physical Interactions with Object-Oriented Prediction and Planningcited by 51
- 2018Hierarchical Long-term Video Prediction without Supervisioncited by 50
- 2018SDCNet: Video Prediction Using Spatially-Displaced Convolutioncited by 50
- 2016SE3-Nets: Learning Rigid Body Motion using Deep Neural Networkscited by 48
- 2016Visual Dynamics: Probabilistic Future Frame Synthesis via Cross Convolutional Networkscited by 46
- 2020Goal-Aware Prediction: Learning to Model What Matterscited by 45
- 2019Compositional Video Predictioncited by 43
50 most cited of 60 All 60 (JSON)
Built on by
Citing papers per year: 2021: 9; 2022: 27; 2023: 20; 2024: 8; 2025: 8; 2026: 3.
75 papers.
Sort:
By area:
- 2023Driving into the Future: Multiview Visual Forecasting and Planning with World Model for Autonomous Drivingcited by 309
- 2022MCVD: Masked Conditional Video Diffusion for Prediction, Generation, and Interpolationcited by 307
- 2022Diffusion Probabilistic Modeling for Video Generationcited by 237
- 2023A Comprehensive Survey of AI-Generated Content (AIGC): A History of Generative AI from GAN to ChatGPTcited by 221
- 2021PredRNN: A Recurrent Neural Network for Spatiotemporal Predictive Learningcited by 181
- 2023Open-World Object Manipulation using Pre-trained Vision-Language Modelscited by 155
- 2021Physion: Evaluating Physical Prediction from Vision in Humans and Machinescited by 145
- 2024iVideoGPT: Interactive VideoGPTs are Scalable World Modelscited by 136
- 2022Diffusion Models for Video Prediction and Infillingcited by 134
- 2021Learning Language-Conditioned Robot Behavior from Offline Data and Crowd-Sourced Annotationcited by 117
- 2022MaskViT: Masked Visual Pre-Training for Video Predictioncited by 112
- 2021GENESIS-V2: Inferring Unordered Object Representations without Iterative Refinementcited by 109
38 more
- 2023VDT: General-purpose Video Diffusion Transformers via Mask Modelingcited by 102
- 2021FitVid: Overfitting in Pixel-Level Video Predictioncited by 86
- 2021CCVS: Context-aware Controllable Video Synthesiscited by 68
- 2022Iso-Dream: Isolating and Leveraging Noncontrollable Visual Dynamics in World Modelscited by 53
- 2022Transframer: Arbitrary Frame Prediction with Generative Modelscited by 37
- 2022Do better ImageNet classifiers assess perceptual similarity better?cited by 34
- 2022Rethinking Learning Approaches for Long-Term Action Anticipationcited by 29
- 2022The Free Energy Principle for Perception and Action: A Deep Learning Perspectivecited by 24
- 2021Sawtooth Factorial Topic Embeddings Guided Gamma Belief Networkcited by 23
- 2022HARP: Autoregressive Latent Video Prediction with High-Fidelity Image Generatorcited by 22
- 2023Blockwise Self-Supervised Learning at Scalecited by 22
- 2023PointCMP: Contrastive Mask Prediction for Self-supervised Learning on Point Cloud Videoscited by 22
- 2023Slot-VAE: Object-Centric Scene Generation with Slot Attentioncited by 20
- 2022Unsupervised Image Representation Learning with Deep Latent Particlescited by 19
- 2023DDLP: Unsupervised Object-Centric Video Prediction with Deep Dynamic Latent Particlescited by 18
- 2022STAP: Sequencing Task-Agnostic Policiescited by 17
- 2025World Models That Know When They Don't Know - Controllable Video Generation with Calibrated Uncertaintycited by 16
- 2023Video Diffusion Models with Local-Global Context Guidancecited by 15
- 2023Neural Foundations of Mental Simulation: Future Prediction of Latent Representations on Dynamic Scenescited by 13
- 2021Example-Driven Model-Based Reinforcement Learning for Solving Long-Horizon Visuomotor Taskscited by 12
- 2022MSPred: Video Prediction at Multiple Spatio-Temporal Scales with Hierarchical Recurrent Networkscited by 12
- 2023MOSO: Decomposing MOtion, Scene and Object for Video Predictioncited by 12
- 20233D-IntPhys: Towards More Generalized 3D-grounded Visual Intuitive Physics under Challenging Scenescited by 12
- 2023STDiff: Spatio-temporal Diffusion for Continuous Stochastic Video Predictioncited by 12
- 2026Latent Particle World Models: Self-supervised Object-centric Stochastic Dynamics Modelingcited by 11
- 2023Recent Advances of Deep Robotic Affordance Learning: A Reinforcement Learning Perspectivecited by 10
- 2021Variational Predictive Routing with Nested Subjective Timescalescited by 9
- 2024Momentum Auxiliary Network for Supervised Local Learningcited by 9
- 2025STARFlow-V: End-to-End Video Generative Modeling with Normalizing Flowscited by 9
- 2022Continual Predictive Learning from Videoscited by 8
- 2024Scaling Supervised Local Learning with Augmented Auxiliary Networkscited by 8
- 2025Dynamical Diffusion: Learning Temporal Dynamics with Diffusion Modelscited by 8
- 2022A unified model for continuous conditional video predictioncited by 7
- 2023Towards End-to-End Generative Modeling of Long Videos with Memory-Efficient Bidirectional Transformerscited by 7
- 2024A Survey on Future Frame Synthesis: Bridging Deterministic and Generative Approachescited by 7
- 2023Block-local learning with probabilistic latent representationscited by 6
- 2024PredBench: Benchmarking Spatio-Temporal Prediction across Diverse Disciplinescited by 6
- 2022MS-RNN: A Flexible Multi-Scale Framework for Spatiotemporal Predictive Learningcited by 5
50 most cited of 75 All 75 (JSON)
- 2026HMPDM: A Diffusion Model for Driving Video Prediction with Historical Motion Priors28 Mar 2026
- 2026GMT: Goal-Conditioned Multimodal Transformer for 6-DOF Object Trajectory Synthesis in 3D Scenes18 Mar 2026
- 2026Latent Particle World Models: Self-supervised Object-centric Stochastic Dynamics Modeling4 Mar 2026
- 2025Autoregressive Flow Matching for Motion Prediction27 Dec 2025
- 2025World Models That Know When They Don't Know - Controllable Video Generation with Calibrated Uncertainty5 Dec 2025
- 2025STARFlow-V: End-to-End Video Generative Modeling with Normalizing Flows25 Nov 2025
- 2025SAMPO:Scale-wise Autoregression with Motion PrOmpt for generative world models19 Sep 2025
- 2025MAN++: Scaling Momentum Auxiliary Network for Supervised Local Learning in Vision Tasks22 Jul 2025
- 2025One Step Closer: Creating the Future to Boost Monocular Semantic Scene Completion18 Jul 2025
- 2025Foresight Diffusion: Improving Sampling Consistency in Predictive Diffusion Models22 May 2025
- 2025Dynamical Diffusion: Learning Temporal Dynamics with Diffusion Models2 Mar 2025
- 2024Advances in Transformers for Robotic Applications: A Review13 Dec 2024
38 more
- 2024PredBench: Benchmarking Spatio-Temporal Prediction across Diverse Disciplines11 Jul 2024
- 2024Momentum Auxiliary Network for Supervised Local Learning8 Jul 2024
- 2024MLAAN: Scaling Supervised Local Learning with Multilaminar Leap Augmented Auxiliary Network24 Jun 2024
- 2024CAVACHON: a hierarchical variational autoencoder to integrate multi-modal single-cell data28 May 2024
- 2024iVideoGPT: Interactive VideoGPTs are Scalable World Models24 May 2024
- 2024Scaling Supervised Local Learning with Augmented Auxiliary Networks27 Feb 2024
- 2024A Survey on Future Frame Synthesis: Bridging Deterministic and Generative Approaches26 Jan 2024
- 2023STDiff: Spatio-temporal Diffusion for Continuous Stochastic Video Prediction11 Dec 2023
- 2023Driving into the Future: Multiview Visual Forecasting and Planning with World Model for Autonomous Driving29 Nov 2023
- 2023Slot-VAE: Object-Centric Scene Generation with Slot Attention12 Jun 2023
- 2023DDLP: Unsupervised Object-Centric Video Prediction with Deep Dynamic Latent Particles9 Jun 2023
- 2023Video Diffusion Models with Local-Global Context Guidance5 Jun 2023
- 2023Block-local learning with probabilistic latent representations24 May 2023
- 2023VDT: General-purpose Video Diffusion Transformers via Mask Modeling22 May 2023
- 2023Neural Foundations of Mental Simulation: Future Prediction of Latent Representations on Dynamic Scenes19 May 2023
- 2023PointCMP: Contrastive Mask Prediction for Self-supervised Learning on Point Cloud Videos6 May 2023
- 20233D-IntPhys: Towards More Generalized 3D-grounded Visual Intuitive Physics under Challenging Scenes22 Apr 2023
- 2023MS-LSTM: Exploring Spatiotemporal Multiscale Representations in Video Prediction Domain16 Apr 2023
- 2023Model-Based Reinforcement Learning with Isolated Imaginations27 Mar 2023
- 2023Towards End-to-End Generative Modeling of Long Videos with Memory-Efficient Bidirectional Transformers20 Mar 2023
- 2023Recent Advances of Deep Robotic Affordance Learning: A Reinforcement Learning Perspective9 Mar 2023
- 2023A Comprehensive Survey of AI-Generated Content (AIGC): A History of Generative AI from GAN to ChatGPT7 Mar 2023
- 2023MOSO: Decomposing MOtion, Scene and Object for Video Prediction7 Mar 2023
- 2023Open-World Object Manipulation using Pre-trained Vision-Language Models2 Mar 2023
- 2023Weakly-supervised Representation Learning for Video Alignment and Analysis8 Feb 2023
- 2023Blockwise Self-Supervised Learning at Scale3 Feb 2023
- 2023Anti-aliasing Predictive Coding Network for Future Video Frame Prediction13 Jan 2023
- 2022Long-horizon video prediction using a dynamic latent hierarchy29 Dec 2022
- 2022Predictive Coding Based Multiscale Network with Encoder-Decoder LSTM for Video Prediction22 Dec 2022
- 2022WALDO: Future Video Synthesis using Object Layer Decomposition and Parametric Flow Prediction25 Nov 2022
- 2022Exploiting Spatial-temporal Correlations for Video Anomaly Detection2 Nov 2022
- 2022STAP: Sequencing Task-Agnostic Policies21 Oct 2022
- 2022Rethinking Learning Approaches for Long-Term Action Anticipation20 Oct 2022
- 2022A unified model for continuous conditional video prediction11 Oct 2022
- 2022See, Plan, Predict: Language-guided Cognitive Planning with Video Prediction7 Oct 2022
- 2022Phi-DVAE: Physics-Informed Dynamical Variational Autoencoders for Unstructured Data Assimilation30 Sep 2022
- 2022HARP: Autoregressive Latent Video Prediction with High-Fidelity Image Generator15 Sep 2022
- 2022Adversarial robustness of VAEs through the lens of local geometry8 Aug 2022
50 newest of 75 All 75 (JSON)
- 2022MCVD: Masked Conditional Video Diffusion for Prediction, Generation, and Interpolationcited by 307
- 2022Diffusion Probabilistic Modeling for Video Generationcited by 237
- 2022Diffusion Models for Video Prediction and Infillingcited by 134
- 2023VDT: General-purpose Video Diffusion Transformers via Mask Modelingcited by 102
- 2023Video Diffusion Models with Local-Global Context Guidancecited by 15
- 2025Dynamical Diffusion: Learning Temporal Dynamics with Diffusion Modelscited by 8
- 2025Foresight Diffusion: Improving Sampling Consistency in Predictive Diffusion Modelscited by 1
- 2025Autoregressive Flow Matching for Motion Predictioncited by 1
- 2026HMPDM: A Diffusion Model for Driving Video Prediction with Historical Motion Priorscited by 0
- 2026HMPDM: A Diffusion Model for Driving Video Prediction with Historical Motion Priors28 Mar 2026
- 2025Autoregressive Flow Matching for Motion Prediction27 Dec 2025
- 2025Foresight Diffusion: Improving Sampling Consistency in Predictive Diffusion Models22 May 2025
- 2025Dynamical Diffusion: Learning Temporal Dynamics with Diffusion Models2 Mar 2025
- 2023Video Diffusion Models with Local-Global Context Guidance5 Jun 2023
- 2023VDT: General-purpose Video Diffusion Transformers via Mask Modeling22 May 2023
- 2022Diffusion Models for Video Prediction and Infilling15 Jun 2022
- 2022MCVD: Masked Conditional Video Diffusion for Prediction, Generation, and Interpolation19 May 2022
- 2022Diffusion Probabilistic Modeling for Video Generation16 Mar 2022
- 2023Driving into the Future: Multiview Visual Forecasting and Planning with World Model for Autonomous Drivingcited by 309
- 2024iVideoGPT: Interactive VideoGPTs are Scalable World Modelscited by 136
- 2022Iso-Dream: Isolating and Leveraging Noncontrollable Visual Dynamics in World Modelscited by 53
- 2025World Models That Know When They Don't Know - Controllable Video Generation with Calibrated Uncertaintycited by 16
- 2026Latent Particle World Models: Self-supervised Object-centric Stochastic Dynamics Modelingcited by 11
- 2022Continual Predictive Learning from Videoscited by 8
- 2024A Survey on Future Frame Synthesis: Bridging Deterministic and Generative Approachescited by 7
- 2023Model-Based Reinforcement Learning with Isolated Imaginationscited by 4
- 2025SAMPO:Scale-wise Autoregression with Motion PrOmpt for generative world modelscited by 4
- 2026Latent Particle World Models: Self-supervised Object-centric Stochastic Dynamics Modeling4 Mar 2026
- 2025World Models That Know When They Don't Know - Controllable Video Generation with Calibrated Uncertainty5 Dec 2025
- 2025SAMPO:Scale-wise Autoregression with Motion PrOmpt for generative world models19 Sep 2025
- 2024iVideoGPT: Interactive VideoGPTs are Scalable World Models24 May 2024
- 2024A Survey on Future Frame Synthesis: Bridging Deterministic and Generative Approaches26 Jan 2024
- 2023Driving into the Future: Multiview Visual Forecasting and Planning with World Model for Autonomous Driving29 Nov 2023
- 2023Model-Based Reinforcement Learning with Isolated Imaginations27 Mar 2023
- 2022Iso-Dream: Isolating and Leveraging Noncontrollable Visual Dynamics in World Models27 May 2022
- 2022Continual Predictive Learning from Videos12 Apr 2022
- 2023A Comprehensive Survey of AI-Generated Content (AIGC): A History of Generative AI from GAN to ChatGPTcited by 221
- 2021PredRNN: A Recurrent Neural Network for Spatiotemporal Predictive Learningcited by 181
- 2021Physion: Evaluating Physical Prediction from Vision in Humans and Machinescited by 145
- 2021Learning Language-Conditioned Robot Behavior from Offline Data and Crowd-Sourced Annotationcited by 117
- 2022MaskViT: Masked Visual Pre-Training for Video Predictioncited by 112
- 2021GENESIS-V2: Inferring Unordered Object Representations without Iterative Refinementcited by 109
- 2021FitVid: Overfitting in Pixel-Level Video Predictioncited by 86
- 2021CCVS: Context-aware Controllable Video Synthesiscited by 68
- 2022Transframer: Arbitrary Frame Prediction with Generative Modelscited by 37
- 2022Do better ImageNet classifiers assess perceptual similarity better?cited by 34
- 2022Rethinking Learning Approaches for Long-Term Action Anticipationcited by 29
- 2022The Free Energy Principle for Perception and Action: A Deep Learning Perspectivecited by 24
12 most cited of 55 All 55 (JSON)
- 2025STARFlow-V: End-to-End Video Generative Modeling with Normalizing Flows25 Nov 2025
- 2025MAN++: Scaling Momentum Auxiliary Network for Supervised Local Learning in Vision Tasks22 Jul 2025
- 2025One Step Closer: Creating the Future to Boost Monocular Semantic Scene Completion18 Jul 2025
- 2024Advances in Transformers for Robotic Applications: A Review13 Dec 2024
- 2024PredBench: Benchmarking Spatio-Temporal Prediction across Diverse Disciplines11 Jul 2024
- 2024Momentum Auxiliary Network for Supervised Local Learning8 Jul 2024
- 2024MLAAN: Scaling Supervised Local Learning with Multilaminar Leap Augmented Auxiliary Network24 Jun 2024
- 2024CAVACHON: a hierarchical variational autoencoder to integrate multi-modal single-cell data28 May 2024
- 2024Scaling Supervised Local Learning with Augmented Auxiliary Networks27 Feb 2024
- 2023STDiff: Spatio-temporal Diffusion for Continuous Stochastic Video Prediction11 Dec 2023
- 2023Slot-VAE: Object-Centric Scene Generation with Slot Attention12 Jun 2023
- 2023DDLP: Unsupervised Object-Centric Video Prediction with Deep Dynamic Latent Particles9 Jun 2023
12 newest of 55 All 55 (JSON)