论文概要
研究领域: Computer Vision (CV) Authors: Dian Zheng, Harry Lee, Manyuan Zhang Published: 2025-06-13 arXiv: 2506.10669
Overview
Recent image generators have demonstrated impressive photorealism and instruction-following capabilities in single-image generation and editing. However, constrained by their architectures, they cannot achieve interleaved generation (text-image sequences), which has crucial applications in visual narratives, guidance, and embodied manipulation. Even the latest open-source Unified Multimodal Models (UMMs) exhibit limited performance in this regard.
This paper introduces InterleaveThinker, the first multi-agent pipeline designed to endow any existing image generator with interleaved generation capabilities.
Key Points
- Planner agent: organizes the image-text input sequence, instructing the image generator on the required execution at each step.
- Critic agent: evaluates generator outputs, identifies samples that deviate from planned instructions, and optimizes instructions for regeneration.
- Cold-start SFT datasets: Interleave-Planner-SFT-80k and Interleave-Critic-SFT-112k are built for format initialization.
- RL stage: Interleave-Critic-RL-13k is developed with GRPO to enhance step-wise instruction correction across generation trajectories.
- Reward design: since a single interleaved generation trajectory may involve over 25 generator calls, optimizing the entire trajectory is computationally impractical. The paper proposes an accuracy reward and a step-wise reward, enabling single-step RL to effectively guide the entire generation trajectory.
- InterleaveThinker improves the performance of various image generators.
- On interleaved generation benchmarks, it achieves performance comparable to Nano Banana and GPT-5.
- Surprisingly, it also significantly enhances base model performance on reasoning benchmarks; for example, notable gains are observed on WISE and RISE with 4-step FLUX.2-klein.
- arXiv paper: https://arxiv.org/abs/2506.10669