MARS: Mixture of Auto-Regressive Models for Fine-grained Text-to-image Synthesis

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: He, Wanggui, Fu, Siming, Liu, Mushui, Wang, Xierui, Xiao, Wenyi, Shu, Fangxun, Wang, Yi, Zhang, Lei, Yu, Zhelun, Li, Haoyuan, Huang, Ziwei, Gan, LeiLei, Jiang, Hao
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866911952153870336
author He, Wanggui
Fu, Siming
Liu, Mushui
Wang, Xierui
Xiao, Wenyi
Shu, Fangxun
Wang, Yi
Zhang, Lei
Yu, Zhelun
Li, Haoyuan
Huang, Ziwei
Gan, LeiLei
Jiang, Hao
author_facet He, Wanggui
Fu, Siming
Liu, Mushui
Wang, Xierui
Xiao, Wenyi
Shu, Fangxun
Wang, Yi
Zhang, Lei
Yu, Zhelun
Li, Haoyuan
Huang, Ziwei
Gan, LeiLei
Jiang, Hao
contents Auto-regressive models have made significant progress in the realm of language generation, yet they do not perform on par with diffusion models in the domain of image synthesis. In this work, we introduce MARS, a novel framework for T2I generation that incorporates a specially designed Semantic Vision-Language Integration Expert (SemVIE). This innovative component integrates pre-trained LLMs by independently processing linguistic and visual information, freezing the textual component while fine-tuning the visual component. This methodology preserves the NLP capabilities of LLMs while imbuing them with exceptional visual understanding. Building upon the powerful base of the pre-trained Qwen-7B, MARS stands out with its bilingual generative capabilities corresponding to both English and Chinese language prompts and the capacity for joint image and text generation. The flexibility of this framework lends itself to migration towards any-to-any task adaptability. Furthermore, MARS employs a multi-stage training strategy that first establishes robust image-text alignment through complementary bidirectional tasks and subsequently concentrates on refining the T2I generation process, significantly augmenting text-image synchrony and the granularity of image details. Notably, MARS requires only 9% of the GPU days needed by SD1.5, yet it achieves remarkable results across a variety of benchmarks, illustrating the training efficiency and the potential for swift deployment in various applications.
format Preprint
id arxiv_https___arxiv_org_abs_2407_07614
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle MARS: Mixture of Auto-Regressive Models for Fine-grained Text-to-image Synthesis
He, Wanggui
Fu, Siming
Liu, Mushui
Wang, Xierui
Xiao, Wenyi
Shu, Fangxun
Wang, Yi
Zhang, Lei
Yu, Zhelun
Li, Haoyuan
Huang, Ziwei
Gan, LeiLei
Jiang, Hao
Computer Vision and Pattern Recognition
Auto-regressive models have made significant progress in the realm of language generation, yet they do not perform on par with diffusion models in the domain of image synthesis. In this work, we introduce MARS, a novel framework for T2I generation that incorporates a specially designed Semantic Vision-Language Integration Expert (SemVIE). This innovative component integrates pre-trained LLMs by independently processing linguistic and visual information, freezing the textual component while fine-tuning the visual component. This methodology preserves the NLP capabilities of LLMs while imbuing them with exceptional visual understanding. Building upon the powerful base of the pre-trained Qwen-7B, MARS stands out with its bilingual generative capabilities corresponding to both English and Chinese language prompts and the capacity for joint image and text generation. The flexibility of this framework lends itself to migration towards any-to-any task adaptability. Furthermore, MARS employs a multi-stage training strategy that first establishes robust image-text alignment through complementary bidirectional tasks and subsequently concentrates on refining the T2I generation process, significantly augmenting text-image synchrony and the granularity of image details. Notably, MARS requires only 9% of the GPU days needed by SD1.5, yet it achieves remarkable results across a variety of benchmarks, illustrating the training efficiency and the potential for swift deployment in various applications.
title MARS: Mixture of Auto-Regressive Models for Fine-grained Text-to-image Synthesis
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2407.07614