Vchitect-2.0: Parallel Transformer for Scaling Up Video Diffusion Models

📅 2025-01-14
📈 Citations: 0
Influential: 0
📄 PDF
🤖 AI Summary
Existing text-to-video (T2V) diffusion models suffer from three key challenges: discontinuous long-video generation, severe memory bottlenecks during training, and scarcity of high-quality paired data. To address these, we propose a parallel Transformer architecture for T2V generation, introducing the first multimodal diffusion module that jointly optimizes text-video alignment and temporal coherence. We further design a hybrid-parallel, memory-efficient training framework to overcome GPU memory constraints in long-sequence modeling. Additionally, we construct Vchitect T2V DataVerse—a million-scale, high-fidelity T2V dataset—augmented with an automated aesthetic evaluation and annotation pipeline. Experiments demonstrate state-of-the-art performance across video fidelity, temporal consistency, training throughput, and model scalability. Our method achieves, for the first time, end-to-end generation of high-fidelity, minute-long videos.

Technology Category

Application Category

📝 Abstract
We present Vchitect-2.0, a parallel transformer architecture designed to scale up video diffusion models for large-scale text-to-video generation. The overall Vchitect-2.0 system has several key designs. (1) By introducing a novel Multimodal Diffusion Block, our approach achieves consistent alignment between text descriptions and generated video frames, while maintaining temporal coherence across sequences. (2) To overcome memory and computational bottlenecks, we propose a Memory-efficient Training framework that incorporates hybrid parallelism and other memory reduction techniques, enabling efficient training of long video sequences on distributed systems. (3) Additionally, our enhanced data processing pipeline ensures the creation of Vchitect T2V DataVerse, a high-quality million-scale training dataset through rigorous annotation and aesthetic evaluation. Extensive benchmarking demonstrates that Vchitect-2.0 outperforms existing methods in video quality, training efficiency, and scalability, serving as a suitable base for high-fidelity video generation.
Problem

Research questions and friction points this paper is trying to address.

Video Generation
Performance Bottleneck
Dataset Scalability
Innovation

Methods, ideas, or system contributions that make the work stand out.

Text-Video Alignment
Efficient Training Methodology
High-Quality Dataset
🔎 Similar Papers
No similar papers found.
Weichen Fan
Weichen Fan
Ph.D. Student, MMLab@NTU, Nanyang Technological University
Computer Vision
C
Chenyang Si
S-Lab, Nanyang Technological University, Singapore, 639798
J
Junhao Song
Shanghai Artificial Intelligence Laboratory, China
Z
Zhenyu Yang
Shanghai Artificial Intelligence Laboratory, China
Yinan He
Yinan He
Shanghai Al Laboratory
L
Long Zhuo
Shanghai Artificial Intelligence Laboratory, China
Ziqi Huang
Ziqi Huang
Ph.D. Student, MMLab@NTU, Nanyang Technological University
Computer Vision
Z
Ziyue Dong
Shanghai Artificial Intelligence Laboratory, China
J
Jingwen He
Shanghai Artificial Intelligence Laboratory, China
D
Dongwei Pan
Shanghai Artificial Intelligence Laboratory, China
Y
Yi Wang
Shanghai Artificial Intelligence Laboratory, China
Y
Yuming Jiang
S-Lab, Nanyang Technological University, Singapore, 639798
Yaohui Wang
Yaohui Wang
Research Scientist, Shanghai AI Laboratory | Inria
Machine LearningDeep Generative ModelsVideo Generation
P
Peng Gao
Shanghai Artificial Intelligence Laboratory, China
X
Xinyuan Chen
Shanghai Artificial Intelligence Laboratory, China
H
Hengjie Li
Shanghai Artificial Intelligence Laboratory, China
Dahua Lin
Dahua Lin
The Chinese University of Hong Kong
computer visionmachine learningprobabilistic inferencebayesian nonparametrics
Y
Yu Qiao
Shanghai Artificial Intelligence Laboratory, China
Ziwei Liu
Ziwei Liu
Associate Professor, Nanyang Technological University
Computer VisionMachine LearningComputer Graphics