Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Latte: Latent Diffusion Transformer for Video Generation

About

We propose Latte, a novel Latent Diffusion Transformer for video generation. Latte first extracts spatio-temporal tokens from input videos and then adopts a series of Transformer blocks to model video distribution in the latent space. In order to model a substantial number of tokens extracted from videos, four efficient variants are introduced from the perspective of decomposing the spatial and temporal dimensions of input videos. To improve the quality of generated videos, we determine the best practices of Latte through rigorous experimental analysis, including video clip patch embedding, model variants, timestep-class information injection, temporal positional embedding, and learning strategies. Our comprehensive evaluation demonstrates that Latte achieves state-of-the-art performance across four standard video generation datasets, i.e., FaceForensics, SkyTimelapse, UCF101, and Taichi-HD. In addition, we extend Latte to the text-to-video generation (T2V) task, where Latte achieves results that are competitive with recent T2V models. We strongly believe that Latte provides valuable insights for future research on incorporating Transformers into diffusion models for video generation.

Xin Ma, Yaohui Wang, Xinyuan Chen, Gengyun Jia, Ziwei Liu, Yuan-Fang Li, Cunjian Chen, Yu Qiao• 2024

Related benchmarks

TaskDatasetResultRank
Video GenerationUCF-101 (test)--
105
Video GenerationUCF-101
FVD478
17
Video GenerationSkyTimelapse (test)
FVD1659.82
16
Class-to-video generationUCF-101
FVD478
13
Unconditional video generationUCF-101 256x256
FVD (256x256, 2048)477.9
12
Time Series ForecastingGreenEarthNet 1.0 (test)
PSNR (NDVI)18.02
9
Video GenerationUCF-101 128-frame, 128x128 resolution (test)
FVD1.90e+3
9
Class-Conditional Video GenerationBenchmark 17x256x256 resolution (test)
gFVD478
9
Time Series ForecastingTS-S12 S2-Sentinel-2 (full-band)
PSNR18.33
7
Unconditional video generationUCF-101
FVD (2048 Dim)478
7
Showing 10 of 10 rows

Other info

Follow for update