CogVideoX: 带专家 Transformer 的文本到视频扩散模型

CogVideoX: Text-to-Video Diffusion Models with An Expert Transformer

唐杰 Jie Tang · · 2024-08-12 · arXiv:2408.06072 ↗ · 被引 2079

打开互动全文版(逐段中英对照 + 图/公式 + 论文问答)→

摘要 · Abstract

我们提出了 CogVideoX,一个基于扩散 Transformer 的大规模文本到视频生成模型,能够生成与文本提示对齐的 10 秒连续视频,帧率为 16 fps,分辨率为 768×1360 像素。以前的视频生成模型通常运动有限、持续时间短,且难以基于文本生成具有连贯叙事的视频。我们提出了几种设计来应对这些问题。首先,我们提出了一种 3D 变分自编码器(VAE),在空间和时间两个维度上压缩视频,以提高压缩率和视频保真度。其次,为了改善文本-视频对齐,我们提出了一种专家 Transformer,带有专家自适应层归一化,以促进两种模态之间的深度融合。第三,通过采用渐进式训练和多分辨率帧打包技术,CogVideoX 擅长生成连贯、长时间、不同形状且具有显著运动的视频。此外,我们开发了一个有效的文本-视频数据处理流程,包括多种数据预处理策略和一种视频描述方法,大大提高了生成质量和语义对齐。结果表明,CogVideoX 在多个机器指标和人工评估中均展现出最先进的性能。3D 因果 VAE、视频描述模型和 CogVideoX 的模型权重已在 https://github.com/THUDM/CogVideo 公开。

We present CogVideoX, a large-scale text-to-video generation model based on diffusion transformer, which can generate 10-second continuous videos aligned with text prompt, with a frame rate of 16 fps and resolution of 768 * 1360 pixels. Previous video generation models often had limited movement and short durations, and is difficult to generate videos with coherent narratives based on text. We propose several designs to address these issues. First, we propose a 3D Variational Autoencoder (VAE) to compress videos along both spatial and temporal dimensions, to improve both compression rate and video fidelity. Second, to improve the text-video alignment, we propose an expert transformer with the expert adaptive LayerNorm to facilitate the deep fusion between the two modalities. Third, by employing a progressive training and multi-resolution frame pack technique, CogVideoX is adept at producing coherent, long-duration, different shape videos characterized by significant motions. In addition, we develop an effective text-video data processing pipeline that includes various data preprocessing strategies and a video captioning method, greatly contributing to the generation quality and semantic alignment. Results show that CogVideoX demonstrates state-of-the-art performance across both multiple machine metrics and human evaluations. The model weight of both 3D Causal VAE, Video caption model and CogVideoX are publicly available at https://github.com/THUDM/CogVideo.

核心贡献 · Key contributions

局限 · Limitations

论文章节 · Sections(共 14)

阅读逐段中英对照全文 →