Post training via GRPO has demonstrated remarkable effectiveness in improving the generation quality of flow-matching models. However, GRPO suffers from inherently low sample efficiency due to its on-policy training paradigm. To address this limitation, we present OP-GRPO, the first Off-Policy GRPO framework tailored for flow-matching models. First, we actively select high-quality trajectories and adaptively incorporate them into a replay buffer for reuse in subsequent training iterations. Second, to mitigate the distribution shift introduced by off-policy samples, we propose a sequence-level importance sampling correction that preserves the integrity of GRPO's clipping mechanism while ensuring stable policy updates. Third, we theoretically and empirically show that late denoising steps yield ill-conditioned off-policy ratios, and mitigate this by truncating trajectories at late steps. Across image and video generation benchmarks, OP-GRPO achieves comparable or superior performance to Flow-GRPO with only 34.2% of the training steps on average, yielding substantial gains in training efficiency while maintaining generation quality.
Figures & tables
Figure 1: Overall framework of OP-GRPO, including (a) OP-GRPO rollout and (b) OP-GRPO training. Blue regions represent samples from the replay buffer, green regions represent samples from the dataset.
Figure 2: Log-probability values of on-policy and off-policy samples across denoising steps, where the dashed line indicates the truncation starting step.
Figure 3: Training Curves of OP-GRPO and Flow GRPO.
Model
Step
Task Metric
Image Quality
Preference Score
Aesthetic
DeQA
ImgRwd
PickScore
UniRwd
SD3.5-M
—
0.63 / 0.59 / 21.72
5.31
4.07
0.92
22.18
3.29
Compositional Image Generation
Flow-GRPO
1020
0.68
5.27
4.05
1.37
22.24
3.35
OP-GRPO
1020
0.88
5.22
4.02
1.21
22.35
3.49
Flow-GRPO
Best
0.95
5.18
4.00
1.12
22.37
3.51
Table 1: Algorithm Performance on the task metrics including Compositional Image Generation, Visual Text Rendering, and Human Preference benchmarks. For the unseen metrics across image quality and preference score, ImgRwd denotes ImageReward and UniRwd denotes UnifiedReward. The best score is in blue
Figure 4: Visual Results of OP-GRPO and Flow GRPO on three image generation tasks using SD3.5-M.
Model
Color
Shape
2D-Spatial
3D-Spatial
Numeracy
Non-Spatial
Average
FLUX.1 Dev [ 13 ]
0.738
0.574
0.289
0.391
0.621
0.315
0.488
SD3.5-M [ 5 ]
0.796
0.569
0.283
0.376
0.595
0.312
0.489
Intermediate checkpoints
SD3.5-M+Flow-GRPO
0.769
0.578
0.412
0.405
0.644
0.312
0.520
SD3.5-M+OP-GRPO
0.824
0.603
0.498
0.437
0.660
0.318
0.567
Best checkpoints
Table 2: T2I-CompBench++ Result. We report results using the Best model trained on the GenEval. The best score is in blue .
Figure 5: Visual Results of Buffer-based GRPO and Flow GRPO on OCR task on video generation model Wan2.1-1.4B. Refer to Appendix for more results.
Deploying GRPO on Flow Matching models has proven effective for text-to-image generation. However, existing paradigms typically propagate an outcome-based reward to all preceding denoising steps without distinguishing the local effect of each step. Moreover, current group-wise ranking mainly compares trajectories at matched timesteps and ignores within-trajectory dependencies, where certain early denoising actions can affect later states via delayed, implicit interactions. We propose TurningPoint-GRPO (TP-GRPO), a GRPO framework that alleviates step-wise reward sparsity and explicitly models long-term effects within the denoising trajectory. TP-GRPO makes two key innovations: (i) it replaces outcome-based rewards with step-level incremental rewards, providing a dense, step-aware learning signal that better isolates each denoising action's "pure" effect, and (ii) it identifies turning points-steps that flip the local reward trend and make subsequent reward evolution consistent with the overall trajectory trend-and assigns these actions an aggregated long-term reward to capture their delayed impact. Turning points are detected solely via sign changes in incremental rewards, making TP-GRPO efficient and hyperparameter-free. Extensive experiments also demonstrate that TP-GRPO exploits reward signals more effectively and consistently improves generation. Demo code is available at https://github.com/YunzeTong/TurningPoint-GRPO.
Yunze Tong, Mushui Liu, Canyu Zhao +7
1Zhejiang University, Hangzhou, China · 2Taobao & Tmall Group of Alibaba, Hangzhou, China · 3Tsinghua University, Beijing, China
Although GRPO substantially enhances flow matching models in human preference alignment of image generation, methods such as FlowGRPO and DanceGRPO still exhibit inefficiency due to the necessity of sampling and optimizing over all denoising steps specified by the Markov Decision Process (MDP). In this paper, we propose MixGRPO, a novel framework that leverages the flexibility of mixed sampling strategies through the integration of stochastic differential equations (SDE) and ordinary differential equations (ODE). This streamlines the optimization process within the MDP to improve efficiency and boost performance. Specifically, MixGRPO introduces a sliding window mechanism, using SDE sampling and GRPO-guided optimization only within the window, while applying ODE sampling outside. This design confines sampling randomness to the time-steps within the window, thereby reducing the optimization overhead, and allowing for more focused gradient updates to accelerate convergence. Additionally, as time-steps beyond the sliding window are not involved in optimization, higher-order solvers are supported for faster sampling. So we present a faster variant, termed MixGRPO-Flash, which further improves training efficiency while achieving comparable performance. MixGRPO exhibits substantial gains across multiple dimensions of human preference alignment, outperforming DanceGRPO in both effectiveness and efficiency, with nearly 50% lower training time. Notably, MixGRPO-Flash further reduces training time by 71%.
Junzhe Li, Yutao Cui, Tao Huang +8
Hunyuan, Tencent, China · School of Computer Science, Peking University, China · Computer Center, Peking University, China +2
Recent work has demonstrated that online reinforcement learning (RL) can substantially improve the quality and alignment of flow matching models for image and video generation. Methods such as Flow-GRPO and CPS cast the denoising process as a Markov Decision Process and apply PPO-style ratio clipping to enforce a trust region. However, we argue that ratio clipping is structurally ill-suited for flow models: the probability ratio between new and old policies is a noisy, single-sample estimate of the true policy divergence, leading to over-constraining in some regions of the trajectory and under-constraining in others. We propose Flow-DPPO (Flow Divergence Proximal Policy Optimization), which replaces ratio clipping with a divergence proximal constraint. A key observation is that the per-step policy in flow models is Gaussian, enabling exact and cheap computation of the KL divergence between old and new policies. Flow-DPPO employs an asymmetric divergence mask that blocks gradient updates only when they simultaneously move away from the trusted region and violate the divergence threshold. Experiments show that Flow-DPPO achieves higher rewards with better KL-proximal efficiency, alleviates catastrophic forgetting, promotes balanced multi-objective optimization, and enables stable multi-epoch training where ratio clipping degrades. Code and models are available at https://github.com/Tencent-Hunyuan/UniRL/tree/main/FlowDPPO.
Bowen Ping, Xiangxin Zhou, Penghui Qi +3
Xi’an Jiaotong University · Tencent Hunyuan · National University of Singapore