Abstract
Recent advancements in large language models have demonstrated howchain-of-thought (CoT) and reinforcement learning (RL) can improve performance.However, applying such reasoning strategies to the visual generation domainremains largely unexplored. In this paper, we present T2I-R1, a novelreasoning-enhanced text-to-image generation model, powered by RL with abi-level CoT reasoning process. Specifically, we identify two levels of CoTthat can be utilized to enhance different stages of generation: (1) thesemantic-level CoT for high-level planning of the prompt and (2) thetoken-level CoT for low-level pixel processing during patch-by-patchgeneration. To better coordinate these two levels of CoT, we introduceBiCoT-GRPO with an ensemble of generation rewards, which seamlessly optimizesboth generation CoTs within the same training step. By applying our reasoningstrategies to the baseline model, Janus-Pro, we achieve superior performancewith 13% improvement on T2I-CompBench and 19% improvement on the WISEbenchmark, even surpassing the state-of-the-art model FLUX.1. Code is availableat: https://github.com/CaraJ7/T2I-R1