OmniGen: Unified Image Generation

GenAI Diffusion TDIG Inpainting CV ODBO HPMR
2024年09月17日
在这项工作中,我们介绍了OmniGen,一种新的统一图像生成扩散模型。与流行的扩散模型(例如稳定扩散)不同,OmniGen不再需要额外的模块,如ControlNet或IP-Adapter来处理不同的控制条件。OmniGen具有以下特点:1)统一性:OmniGen不仅具有文本到图像生成能力,而且本质上支持其他下游任务,如图像编辑、主题驱动生成和视觉条件生成。此外,OmniGen可以通过将它们转化为图像生成任务来处理经典的计算机视觉任务,如边缘检测和人体姿态识别。2)简单性:OmniGen的架构非常简化,消除了额外的文本编码器的需要。此外,与现有的扩散模型相比,它更加用户友好,使得可以通过指令完成复杂任务而无需额外的预处理步骤(例如人体姿态估计),从而显著简化了图像生成的工作流程。3)知识转移:通过以统一格式进行学习,OmniGen有效地在不同任务之间转移知识,处理未见过的任务和领域,并展示新的能力。我们还探讨了模型的推理能力和链式思考机制的潜在应用。这项工作代表了通用图像生成模型的首次尝试,但仍存在一些未解决的问题。我们将在https://github.com/VectorSpaceLab/OmniGen开源相关资源,以促进这一领域的进步。
In this work, we introduce OmniGen, a new diffusion model for unified image generation. Unlike popular diffusion models (e.g., Stable Diffusion), OmniGen no longer requires additional modules such as ControlNet or IP-Adapter to process diverse control conditions. OmniGenis characterized by the following features: 1) Unification: OmniGen not only demonstrates text-to-image generation capabilities but also inherently supports other downstream tasks, such as image editing, subject-driven generation, and visual-conditional generation. Additionally, OmniGen can handle classical computer vision tasks by transforming them into image generation tasks, such as edge detection and human pose recognition. 2) Simplicity: The architecture of OmniGen is highly simplified, eliminating the need for additional text encoders. Moreover, it is more user-friendly compared to existing diffusion models, enabling complex tasks to be accomplished through instructions without the need for extra preprocessing steps (e.g., human pose estimation), thereby significantly simplifying the workflow of image generation. 3) Knowledge Transfer: Through learning in a unified format, OmniGen effectively transfers knowledge across different tasks, manages unseen tasks and domains, and exhibits novel capabilities. We also explore the model's reasoning capabilities and potential applications of chain-of-thought mechanism. This work represents the first attempt at a general-purpose image generation model, and there remain several unresolved issues. We will open-source the related resources at https://github.com/VectorSpaceLab/OmniGen to foster advancements in this field.
许愿