ImageGen-CoT: Enhancing Text-to-Image In-context Learning with Chain-of-Thought Reasoning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Liao, Jiaqi, Yang, Zhengyuan, Li, Linjie, Li, Dianqi, Lin, Kevin, Cheng, Yu, Wang, Lijuan
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913756916744192
author Liao, Jiaqi
Yang, Zhengyuan
Li, Linjie
Li, Dianqi
Lin, Kevin
Cheng, Yu
Wang, Lijuan
author_facet Liao, Jiaqi
Yang, Zhengyuan
Li, Linjie
Li, Dianqi
Lin, Kevin
Cheng, Yu
Wang, Lijuan
contents In this work, we study the problem of Text-to-Image In-Context Learning (T2I-ICL). While Unified Multimodal LLMs (MLLMs) have advanced rapidly in recent years, they struggle with contextual reasoning in T2I-ICL scenarios. To address this limitation, we propose a novel framework that incorporates a thought process called ImageGen-CoT prior to image generation. To avoid generating unstructured ineffective reasoning steps, we develop an automatic pipeline to curate a high-quality ImageGen-CoT dataset. We then fine-tune MLLMs using this dataset to enhance their contextual reasoning capabilities. To further enhance performance, we explore test-time scale-up strategies and propose a novel hybrid scaling approach. This approach first generates multiple ImageGen-CoT chains and then produces multiple images for each chain via sampling. Extensive experiments demonstrate the effectiveness of our proposed method. Notably, fine-tuning with the ImageGen-CoT dataset leads to a substantial 80\% performance gain for SEED-X on T2I-ICL tasks. See our project page at https://ImageGen-CoT.github.io/. Code and model weights will be open-sourced.
format Preprint
id arxiv_https___arxiv_org_abs_2503_19312
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle ImageGen-CoT: Enhancing Text-to-Image In-context Learning with Chain-of-Thought Reasoning
Liao, Jiaqi
Yang, Zhengyuan
Li, Linjie
Li, Dianqi
Lin, Kevin
Cheng, Yu
Wang, Lijuan
Computer Vision and Pattern Recognition
In this work, we study the problem of Text-to-Image In-Context Learning (T2I-ICL). While Unified Multimodal LLMs (MLLMs) have advanced rapidly in recent years, they struggle with contextual reasoning in T2I-ICL scenarios. To address this limitation, we propose a novel framework that incorporates a thought process called ImageGen-CoT prior to image generation. To avoid generating unstructured ineffective reasoning steps, we develop an automatic pipeline to curate a high-quality ImageGen-CoT dataset. We then fine-tune MLLMs using this dataset to enhance their contextual reasoning capabilities. To further enhance performance, we explore test-time scale-up strategies and propose a novel hybrid scaling approach. This approach first generates multiple ImageGen-CoT chains and then produces multiple images for each chain via sampling. Extensive experiments demonstrate the effectiveness of our proposed method. Notably, fine-tuning with the ImageGen-CoT dataset leads to a substantial 80\% performance gain for SEED-X on T2I-ICL tasks. See our project page at https://ImageGen-CoT.github.io/. Code and model weights will be open-sourced.
title ImageGen-CoT: Enhancing Text-to-Image In-context Learning with Chain-of-Thought Reasoning
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2503.19312