TensorX
返回文献探索

Paper · arXiv 2403.03194

MAGID: An Automated Pipeline for Generating Synthetic Multi-modal Datasets

Hossein Aboutalebi, Hwanjun Song, Yusheng Xie, Arshit Gupta, Justin Sun, Hang Su, Igor Shalyminov, Nikolaos Pappas, Siffi Singh, Saab Mansour

14 upvotesMarch 5, 2024arXiv 预印本
AI 摘要

MAGID, a framework augmenting text-only dialogues with high-quality images using a diffusion model and feedback loops, outperforms existing multimodal dialogue generation methods.

diffusion modelimage-text matchingmultimodal dialoguesimage qualityfeedback looptextual LLMaesthetic assessmentsafety assessment

Abstract

Development of multimodal interactive systems is hindered by the lack of rich, multimodal (text, images) conversational data, which is needed in large quantities for LLMs. Previous approaches augment textual dialogues with retrieved images, posing privacy, diversity, and quality constraints. In this work, we introduce Multimodal Augmented Generative Images Dialogues (MAGID), a framework to augment text-only dialogues with diverse and high-quality images. Subsequently, a diffusion model is applied to craft corresponding images, ensuring alignment with the identified text. Finally, MAGID incorporates an innovative feedback loop between an image description generation module (textual LLM) and image quality modules (addressing aesthetics, image-text matching, and safety), that work in tandem to generate high-quality and multi-modal dialogues. We compare MAGID to other SOTA baselines on three dialogue datasets, using automated and human evaluation. Our results show that MAGID is comparable to or better than baselines, with significant improvements in human evaluation, especially against retrieval baselines where the image database is small.

北京市昌平区探索星信息技术及软件开发工作室

京ICP备2026059466号
MAGID: An Automated Pipeline for Generating Synthetic Multi-modal Datasets | TensorX