一个模型搞定十多种图像视频生成与编辑任务,效果媲美商业引擎。
Many-for-Many: Unify the Training of Multiple Video and Image Generation and Manipulation Tasks
- 用轻量适配器统一不同任务的输入条件,实现多任务联合训练。
- 80亿参数模型在视频生成上表现接近商业级水平,支持10+任务。
- 引入深度图增强3D空间理解,适合多模态视觉生成研究者使用。
扩散模型在多种视觉生成与操作任务中表现优异。现有方法通常针对特定任务(如文本到视频)训练模型,或微调预训练的文本到视频模型以支持图像到视频、视频到视频、图像与视频编辑等任务。但训练强性能的文本到视频基础模型需要大量高质量标注数据,成本高昂。此外,多数模型仅能完成一至几项任务。本文提出统一框架Many-for-Many,利用来自多种视觉生成与操作任务的训练数据,训练单一模型以支持这些任务。我们设计轻量适配器统一不同任务的条件输入,并采用图像-视频联合学习策略从零开始渐进式训练。该联合学习使模型具备更强的视频生成能力。同时引入深度图作为条件,提升对3D空间的感知。训练了两个版本模型(8B和2B),均可执行超过10种任务。其中80亿参数模型在视频生成任务上表现媲美开源及商用引擎。模型与代码已开源:https://github.com/leeruibin/MfM.git。
原文摘要 · Abstract (English)
Diffusion models have shown impressive performance in many visual generation and manipulation tasks. Many existing methods focus on training a model for a specific task, especially, text-to-video (T2V) generation, while many other works focus on finetuning the pretrained T2V model for image-to-video (I2V), video-to-video (V2V), image and video manipulation tasks, etc. However, training a strong T2V foundation model requires a large amount of high-quality annotations, which is very costly. In addition, many existing models can perform only one or several tasks. In this work, we introduce a unified framework, namely many-for-many, which leverages the available training data from many different visual generation and manipulation tasks to train a single model for those different tasks. Specifically, we design a lightweight adapter to unify the different conditions in different tasks, then employ a joint image-video learning strategy to progressively train the model from scratch. Our joint learning leads to a unified visual generation and manipulation model with improved video generation performance. In addition, we introduce depth maps as a condition to help our model better perceive the 3D space in visual generation. Two versions of our model are trained with different model sizes (8B and 2B), each of which can perform more than 10 different tasks. In particular, our 8B model demonstrates highly competitive performance in video generation tasks compared to open-source and even commercial engines. Our models and source codes are available at https://github.com/leeruibin/MfM.git.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。