cs.CVJun 5, 2026

AdaGRPO: A Capability-Aware Adaptive Enhancement for Flow-based GRPO

Authors: Jiazi BuPengyang LingYujie ZhouYibin WangYuhang ZangTianyi WeiXiaohang ZhanJiaqi Wang+3 more

Organizations: 1Shanghai Jiao Tong University · 2S-Lab, Nanyang Technological University · 3Shanghai AI Laboratory · University of Science and Technology of China · 8Fudan University · 6Shanghai Innovation Institute · 10Adobe Research · 5Stanford University · 7The Chinese University of Hong Kong · 9CPII under InnoHK

Abstract

Group Relative Policy Optimization (GRPO) has demonstrated remarkable success in aligning text-to-image (T2I) flow models with human preferences. However, we have identified that the learning loop of current flow-based GRPO is fundamentally decoupled from the learner's current capability, suffering from critical blind spots at both prompt selection and advantage estimation: (i) Existing methods sample prompts randomly, overlooking the substantial impact of data selection on reinforcement learning (RL) efficacy--a factor proven crucial in GRPO for large language models; (ii) They evaluate sample quality solely relying on intra-group statistics, lacking a global perspective to accurately measure true policy improvement. To address these issues, we propose Adaptive GRPO (AdaGRPO), a novel capability-aware RL algorithm tailored for flow models. Specifically, AdaGRPO consists of two principal components: (i) Online Curriculum Filtering Strategy: Dynamically tracks the model's proficiency and adaptively selects prompts that best match its current learning boundary; (ii) Cross-Level Advantage Fusion: Synergistically integrates fine-grained intra-group advantages with macro-level global advantages, providing a comprehensive and unbiased policy evaluation. As a lightweight, plug-and-play module, AdaGRPO can be seamlessly integrated with existing frameworks such as Flow-GRPO, DanceGRPO, and Flow-CPS. Extensive experiments demonstrate that AdaGRPO consistently drives performance gains while significantly stabilizes GRPO training for flow models.

Explore similar work

CardsList