Alleviating Sparse Rewards by Modeling Step-Wise and Long-Term Sampling Effects in Flow-Based GRPO

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Tong, Yunze, Liu, Mushui, Zhao, Canyu, He, Wanggui, Zhang, Shiyi, Zhang, Hongwei, Zhang, Peng, Liu, Jinlong, Huang, Ju, Wang, Jiamang, Jiang, Hao, Huang, Pipei
Format: Preprint
Veröffentlicht: 2026
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866918325786771456
author Tong, Yunze
Liu, Mushui
Zhao, Canyu
He, Wanggui
Zhang, Shiyi
Zhang, Hongwei
Zhang, Peng
Liu, Jinlong
Huang, Ju
Wang, Jiamang
Jiang, Hao
Huang, Pipei
author_facet Tong, Yunze
Liu, Mushui
Zhao, Canyu
He, Wanggui
Zhang, Shiyi
Zhang, Hongwei
Zhang, Peng
Liu, Jinlong
Huang, Ju
Wang, Jiamang
Jiang, Hao
Huang, Pipei
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2602_06422
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Alleviating Sparse Rewards by Modeling Step-Wise and Long-Term Sampling Effects in Flow-Based GRPO
Tong, Yunze
Liu, Mushui
Zhao, Canyu
He, Wanggui
Zhang, Shiyi
Zhang, Hongwei
Zhang, Peng
Liu, Jinlong
Huang, Ju
Wang, Jiamang
Jiang, Hao
Huang, Pipei
Computer Vision and Pattern Recognition
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.
title Alleviating Sparse Rewards by Modeling Step-Wise and Long-Term Sampling Effects in Flow-Based GRPO
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2602.06422