DialogGen:
Multi-modal Interactive Dialogue System for Multi-turn Text-to-Image Generation

Minbin Huang1,*, Yanxin Long2,*, Xinchi Deng3, Ruihang Chu1, Jiangfeng Xiong2,
Xiaodan Liang3, Hong Cheng1, Qinglin Lu2,‡, Wei Liu2,
1The Chinese University of Hong Kong, 2Tencent Hunyuan, 3Shenzhen Campus of Sun Yat-sen University,
*Equal Contribution, ‡Corresponding author
huangminbin@link.cuhk.edu.hk, qinglinlu@tecent.com
🤗 Datasets DialogGen HunyuanDiT arXiv

Abstract

Text-to-image (T2I) generation models have significantly advanced in recent years. However, effective interaction with these models is challenging for average users due to the need for specialized prompt engineering knowledge and the inability to perform multi-turn image generation, hindering a dynamic and iterative creation process. Recent attempts have tried to equip Multi-modal Large Language Models (MLLMs) with T2I models to bring the user's natural language instructions into reality. Hence, the output modality of MLLMs is extended, and the multi-turn generation quality of T2I models is enhanced thanks to the strong multi-modal comprehension ability of MLLMs. However, many of these works face challenges in identifying correct output modalities and generating coherent images accordingly as the number of output modalities increases and the conversations go deeper. Therefore, we propose DialogGen, an effective pipeline to align off-the-shelf MLLMs and T2I models to build a Multi-modal Interactive Dialogue System (MIDS) for multi-turn Text-to-Image generation. It is composed of drawing prompt alignment, careful training data curation, and error correction. Moreover, as the field of MIDS flourishes, comprehensive benchmarks are urgently needed to evaluate MIDS fairly in terms of output modality correctness and multi-modal output coherence. To address this issue, we introduce the Multi-modal Dialogue Benchmark (DialogBen), a comprehensive bilingual benchmark designed to assess the ability of MLLMs to generate accurate and coherent multi-modal content that supports image editing. It contains two evaluation metrics to measure the model's ability to switch modalities and the coherence of the output images. Our extensive experiments on DialogBen and user study demonstrate the effectiveness of DialogGen in producing correct output modalities and coherent multi-modal outputs compared with other State-of-the-Art models. We hope that DialogBen can contribute to the community for building more powerful MIDS.

Overview

metamath

Figure 1: llustration of Multi-modal Interactive Dialogue System (MIDS) built by our proposed DialogGen that can perform multi-turn multi-modal tasks responding to user’s natural language instructions to meet the users’ needs for image generation, image editing, and chatting.

DialogBen

metamath

Figure 2: DialogBen encompasses 7 edit instruction types and 13 topic types.

DialogBen

metamath

Figure 3: Overview of MIDS and DialogBen. MIDS can respond to the multi-modal user instructions with either a text response or a drawing prompt to be sent to a T2I model for image generation. DialogBen consists of 9957 three-turn multi-modal dialogs and two evaluation metrics to assess the capability of MIDS.

DialogGen

metamath

Figure 4: The overall pipeline of DialogGen which consists of Drawing Prompt Alignment, Training Data Curation, and Error Correction. In Drawing Prompt Alignment, re-captioning on DG is performed to ensure the alignment between transformed prompts and the T2I model. Then we carefully curate the training data such as adding object consistency guarantee, bilingual data, and mixed instruction tuning data during training. Finally, we employ an error correction mechanism on student model Ms to make the model learn from its mistakes.

Visualizations

metamath metamath

Figure 5: Visualizations of the inference result of DialogGen-HunyuanDiT

BibTeX

@article{huang2024dialoggen,
  title={DialogGen: Multi-modal Interactive Dialogue System for Multi-turn Text-to-Image Generation},
  author={Huang, Minbin and Long, Yanxin and Deng, Xinchi and Chu, Ruihang and Xiong, Jiangfeng and Liang, Xiaodan and Cheng, Hong and Lu, Qinglin and Liu, Wei},
  journal={arXiv preprint arXiv:2403.08857},
  year={2024}
}