Versatile Diffusion: Text, Images and Variations All in One Diffusion Model

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Xu, Xingqian, Wang, Zhangyang, Zhang, Eric, Wang, Kai, Shi, Humphrey
Formato: Preprint
Publicado: 2022
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866911755451498496
author Xu, Xingqian
Wang, Zhangyang
Zhang, Eric
Wang, Kai
Shi, Humphrey
author_facet Xu, Xingqian
Wang, Zhangyang
Zhang, Eric
Wang, Kai
Shi, Humphrey
contents Recent advances in diffusion models have set an impressive milestone in many generation tasks, and trending works such as DALL-E2, Imagen, and Stable Diffusion have attracted great interest. Despite the rapid landscape changes, recent new approaches focus on extensions and performance rather than capacity, thus requiring separate models for separate tasks. In this work, we expand the existing single-flow diffusion pipeline into a multi-task multimodal network, dubbed Versatile Diffusion (VD), that handles multiple flows of text-to-image, image-to-text, and variations in one unified model. The pipeline design of VD instantiates a unified multi-flow diffusion framework, consisting of sharable and swappable layer modules that enable the crossmodal generality beyond images and text. Through extensive experiments, we demonstrate that VD successfully achieves the following: a) VD outperforms the baseline approaches and handles all its base tasks with competitive quality; b) VD enables novel extensions such as disentanglement of style and semantics, dual- and multi-context blending, etc.; c) The success of our multi-flow multimodal framework over images and text may inspire further diffusion-based universal AI research. Our code and models are open-sourced at https://github.com/SHI-Labs/Versatile-Diffusion.
format Preprint
id arxiv_https___arxiv_org_abs_2211_08332
institution arXiv
publishDate 2022
record_format arxiv
spellingShingle Versatile Diffusion: Text, Images and Variations All in One Diffusion Model
Xu, Xingqian
Wang, Zhangyang
Zhang, Eric
Wang, Kai
Shi, Humphrey
Computer Vision and Pattern Recognition
Recent advances in diffusion models have set an impressive milestone in many generation tasks, and trending works such as DALL-E2, Imagen, and Stable Diffusion have attracted great interest. Despite the rapid landscape changes, recent new approaches focus on extensions and performance rather than capacity, thus requiring separate models for separate tasks. In this work, we expand the existing single-flow diffusion pipeline into a multi-task multimodal network, dubbed Versatile Diffusion (VD), that handles multiple flows of text-to-image, image-to-text, and variations in one unified model. The pipeline design of VD instantiates a unified multi-flow diffusion framework, consisting of sharable and swappable layer modules that enable the crossmodal generality beyond images and text. Through extensive experiments, we demonstrate that VD successfully achieves the following: a) VD outperforms the baseline approaches and handles all its base tasks with competitive quality; b) VD enables novel extensions such as disentanglement of style and semantics, dual- and multi-context blending, etc.; c) The success of our multi-flow multimodal framework over images and text may inspire further diffusion-based universal AI research. Our code and models are open-sourced at https://github.com/SHI-Labs/Versatile-Diffusion.
title Versatile Diffusion: Text, Images and Variations All in One Diffusion Model
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2211.08332