Delving into RL for Image Generation with CoT: A Study on DPO vs. GRPO
This study addresses the problem of improving image generation quality and consistency for researchers and practitioners in AI by providing a comprehensive analysis of RL methods in this domain, though it is incremental as it builds on existing RL techniques.
The paper investigates the application of Reinforcement Learning (RL) algorithms, specifically Direct Preference Optimization (DPO) and Group Relative Policy Optimization (GRPO), to autoregressive image generation, treating it as a Chain-of-Thought reasoning process. It finds that GRPO and DPO have distinct advantages, and reward models with strong generalization capabilities can enhance the algorithms' performance, with systematic scaling strategies improving both in-domain and out-of-domain proficiency.
Recent advancements underscore the significant role of Reinforcement Learning (RL) in enhancing the Chain-of-Thought (CoT) reasoning capabilities of large language models (LLMs). Two prominent RL algorithms, Direct Preference Optimization (DPO) and Group Relative Policy Optimization (GRPO), are central to these developments, showcasing different pros and cons. Autoregressive image generation, also interpretable as a sequential CoT reasoning process, presents unique challenges distinct from LLM-based CoT reasoning. These encompass ensuring text-image consistency, improving image aesthetic quality, and designing sophisticated reward models, rather than relying on simpler rule-based rewards. While recent efforts have extended RL to this domain, these explorations typically lack an in-depth analysis of the domain-specific challenges and the characteristics of different RL strategies. To bridge this gap, we provide the first comprehensive investigation of the GRPO and DPO algorithms in autoregressive image generation, evaluating their in-domain performance and out-of-domain generalization, while scrutinizing the impact of different reward models on their respective capabilities. Our findings reveal that GRPO and DPO exhibit distinct advantages, and crucially, that reward models possessing stronger intrinsic generalization capabilities potentially enhance the generalization potential of the applied RL algorithms. Furthermore, we systematically explore three prevalent scaling strategies to enhance both their in-domain and out-of-domain proficiency, deriving unique insights into efficiently scaling performance for each paradigm. We hope our study paves a new path for inspiring future work on developing more effective RL algorithms to achieve robust CoT reasoning in the realm of autoregressive image generation. Code is released at https://github.com/ZiyuGuo99/Image-Generation-CoT