MilliVid: Hierarchical Latents for Long-Range Consistency in Video Generation

· Source: Machine Learning · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Computer Vision & Pattern Recognition · Depth: Expert, quick

Summary

MilliVid is a novel video generation model introduced to overcome long-range consistency challenges in generative video models, which typically struggle with impractically long transformer sequence lengths for extended videos. This approach employs a coarse-to-fine rollout strategy within a multi-scale token space. First, an autoencoder is pre-trained to compress each video frame into a hierarchical token structure, ranging from standard latent resolutions to just a few tokens per frame. The coarsest tokens encode critical information like scene layout and semantics, while finer tokens capture high-frequency appearance and texture. Subsequently, a video diffusion model is trained to generate these tokens using the coarse-to-fine rollout. This method precisely manages the level of detail generated and used as context, ensuring long-range consistency in geometry and object permanence, while optimizing compute by focusing on perceptually relevant details. MilliVid was validated using a custom dataset of long Minecraft videos, demonstrating significantly more consistent rollouts compared to existing baselines.

Key takeaway

For Machine Learning Engineers developing video generative models, if you are struggling with long-range consistency or high computational demands for extended sequences, consider adopting a hierarchical latent approach. MilliVid's method of compressing frames into multi-scale tokens and using coarse-to-fine rollout can significantly improve object permanence and geometric consistency. You should explore implementing similar multi-scale tokenization to optimize compute while maintaining visual coherence in your long-form video outputs.

Key insights

Hierarchical latent token compression and coarse-to-fine rollout enable long-range consistency in video generation by managing detail levels efficiently.

Principles

Method

Pre-train an autoencoder to compress frames into hierarchical tokens. Then, train a video diffusion model to generate these tokens via coarse-to-fine rollout, controlling detail and context for consistency.

In practice

Topics

Best for: Research Scientist, AI Scientist, Machine Learning Engineer, Computer Vision Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Machine Learning.