Power Reinforcement Post-Training of Text-to-Image Models with Super-Linear Advantage Shaping
On the map: LLM reasoning → Reinforcement learning for reasoning
Abstract (arXiv)
Recently, post-training methods based on reinforcement learning, with a particular focus on Group Relative Policy Optimization (GRPO), have emerged as the robust paradigm for further advancement of text-to-image (T2I) models. However, these methods are often prone to reward hacking, wherein models exploit biases in imperfect reward functions rather than yielding genuine performance gains. In this work, we identify that normalization could lead to miscalibration and directly removing the prompt-level standard deviation term yields an optimal policy ascent direction that is linear in the advantage but still limits the separation of genuine signals from noise. To mitigate the above issues, we propose Super-Linear Advantage Shaping (SLAS) by revisiting the functional update from an information geometry perspective. By extending the Fisher-Rao information metric with advantage-dependent weighting, SLAS introduces a non-linear geometric structure that reshapes the local policy space. This design relaxes constraints along high-advantage directions to amplify informative updates, while tightening those in low-advantage regions to suppress illusory gradients. In addition, batch-level normalization is applied to stabilize training under varying reward scales. Extensive evaluations demonstrate that SLAS consistently surpasses the DanceGRPO baseline across multiple backbones and benchmarks. In particular, it yields faster training dynamics, improved out-of-domain performance on GenEval and UniGenBench++, and enhanced robustness to model scaling, while mitigating reward hacking and preserving semantic and compositional fidelity in generations.
Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Placed in Reinforcement learning for reasoning, one of 17 directions in LLM reasoning. Builds on 113 of its 124 references that are in the map. First posted on arXiv 11 May 2026.
Builds on
113 of 124 references.
Closest ancestors · 26
Same direction, newest first
- 2026DenseGRPO: From Sparse to Dense Reward for Flow Matching Model Alignment28 Jan
- 2026E-GRPO: High Entropy Steps Drive Effective Reinforcement Learning for Flow Models1 Jan
- 2025Co-GRPO: Co-Optimized Group Relative Policy Optimization for Masked Diffusion Model25 Dec
- 2025DiverseGRPO: Mitigating Mode Collapse in Image Generation via Diversity-Aware GRPO25 Dec
- 2025TreeGRPO: Tree-Advantage GRPO for Online RL Post-Training of Diffusion Models9 Dec
- 2025Multi-GRPO: Multi-Group Advantage Estimation for Text-to-Image Generation with Tree-Based Trajectories and Multiple Rewards30 Nov
- 2025GRPO-Guard: Mitigating Implicit Over-Optimization in Flow Matching via Regulated Clipping25 Oct
- 2025Principled RL for Flow Matching Emerges from the Chunk-level Policy Optimization24 Oct
- 2025Reinforcement Learning Meets Masked Generative Models: Mask-GRPO for Text-to-Image Generation15 Oct
- 2025Smart-GRPO: Smartly Sampling Noise for Efficient RL of Flow-Matching Models3 Oct
- 2025STAGE: Stable and Generalizable GRPO for Autoregressive Image Generation29 Sep
- 2025A Survey of Reinforcement Learning for Large Reasoning Models10 Sep
14 more
- 2025Pref-GRPO: Pairwise Preference Reward-based GRPO for Stable Text-to-Image Reinforcement Learning28 Aug
- 2025Uncalibrated Reasoning: GRPO Induces Overconfidence for Stochastic Outcomes15 Aug
- 2025AR-GRPO: Training Autoregressive Image Generation Models via Reinforcement Learning9 Aug
- 2025TempFlow-GRPO: When Timing Matters for GRPO in Flow Models6 Aug
- 2025Geometric-Mean Policy Optimization28 Jul
- 2025DanceGRPO: Unleashing GRPO on Visual Generation12 May
- 2025Flow-GRPO: Training Flow Matching Models via Online RL8 May
- 2025VAPO: Efficient and Reliable Reinforcement Learning for Advanced Reasoning Tasks7 Apr
- 2025CPPO: Accelerating the Training of Group Relative Policy Optimization-Based Reasoning Models28 Mar
- 2025Understanding R1-Zero-Like Training: A Critical Perspective26 Mar
- 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scale18 Mar
- 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learning22 Jan
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Model29 May
- 2017Proximal Policy Optimization Algorithms20 Jul
Other references · 87
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,694
- 2023GPT-4 Technical Reportcited by 18,132
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,865
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2022Flow Matching for Generative Modelingcited by 5,426
- 2023SDXL: Improving Latent Diffusion Models for High-Resolution Image Synthesiscited by 4,290
- 2022LAION-5B: An open large-scale dataset for training next generation image-text modelscited by 4,131
- 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesiscited by 4,094
- 2017Deep reinforcement learning from human preferencescited by 3,846
- 2025Gemini 2.5: Pushing the Frontier with Advanced Reasoning, Multimodality, Long Context, and Next Generation Agentic Capabilitiescited by 3,655
- 2019Fine-Tuning Language Models from Human Preferencescited by 2,124
38 more
- 2025Gemma 3 Technical Reportcited by 1,686
- 2023ImageReward: Learning and Evaluating Human Preferences for Text-to-Image Generationcited by 1,010
- 2025Qwen-Image Technical Reportcited by 973
- 2024SimPO: Simple Preference Optimization with a Reference-Free Rewardcited by 951
- 2023Pick-a-Pic: An Open Dataset of User Preferences for Text-to-Image Generationcited by 950
- 2023GenEval: An Object-Focused Framework for Evaluating Text-to-Image Alignmentcited by 946
- 2023Training Diffusion Models with Reinforcement Learningcited by 939
- 2023Human Preference Score v2: A Solid Benchmark for Evaluating Human Preferences of Text-to-Image Synthesiscited by 847
- 2025Group Sequence Policy Optimizationcited by 788
- 2021Vector Quantized Diffusion Model for Text-to-Image Synthesiscited by 773
- 2023Diffusion Model Alignment Using Direct Preference Optimizationcited by 767
- 2023Open Problems and Fundamental Limitations of Reinforcement Learning from Human Feedbackcited by 743
- 2016Generative Adversarial Text to Image Synthesiscited by 735
- 2024Show-o: One Single Transformer to Unify Multimodal Understanding and Generationcited by 734
- 2021CogView: Mastering Text-to-Image Generation via Transformerscited by 691
- 2024ORPO: Monolithic Preference Optimization without Reference Modelcited by 556
- 2023DPOK: Reinforcement Learning for Fine-tuning Text-to-Image Diffusion Modelscited by 431
- 2023Directly Fine-Tuning Diffusion Models on Differentiable Rewardscited by 431
- 2023Aligning Text-to-Image Models using Human Feedbackcited by 418
- 2018Shampoo: Preconditioned Stochastic Tensor Optimizationcited by 413
- 2024Contrastive Preference Optimization: Pushing the Boundaries of LLM Performance in Machine Translationcited by 390
- 2022CogView2: Faster and Better Text-to-Image Generation via Hierarchical Transformerscited by 305
- 2023Text-to-image Diffusion Models in Generative AI: A Surveycited by 264
- 2025MiniMax-M1: Scaling Test-Time Compute Efficiently with Lightning Attentioncited by 248
- 2023Using Human Feedback to Fine-tune Diffusion Models without Any Reward Modelcited by 246
- 2024Iterative Reasoning Preference Optimizationcited by 244
- 2025Show-o2: Improved Native Unified Multimodal Modelscited by 242
- 2025DiffusionNFT: Online Diffusion Reinforcement with Forward Processcited by 204
- 2025MixGRPO: Unlocking Flow-based GRPO Efficiency with Mixed ODE-SDEcited by 168
- 2015Generating Images from Captions with Attentioncited by 135
- 2024Direct Preference Optimization with an Offsetcited by 122
- 2024ReNO: Enhancing One-step Text-to-Image Models through Reward-based Noise Optimizationcited by 115
- 2023End-to-End Diffusion Latent Optimization Improves Classifier Guidancecited by 107
- 2024Aligning Diffusion Models by Optimizing Human Utilitycited by 99
- 2024LiPO: Listwise Preference Optimization through Learning-to-Rankcited by 94
- 2021GAN-Control: Explicitly Controllable GANscited by 91
- 2024Self-Play Fine-Tuning of Diffusion Models for Text-to-Image Generationcited by 85
- 2024Aesthetic Post-Training Diffusion Models from Generic Preferences with Step-by-step Preference Optimizationcited by 81
50 most cited of 87 All 87 (JSON)
Built on by
Citing papers per year: 2026: 3.
3 papers.
- 2026Balancing Performance and Diversity in GRPO Autoregressive Text-to-Image Post-Trainingcited by 0
- 2026Advantage Scale Calibration Imbalance in Group-Relative Optimization under Low-Variance Rewards: Diagnosis and Bounded Recoverycited by 0
- 2026Elucidating the Design Space of Regression-based Diffusion Reinforcement Learningcited by 0
- 2026Elucidating the Design Space of Regression-based Diffusion Reinforcement Learning27 Sep 2026
- 2026Advantage Scale Calibration Imbalance in Group-Relative Optimization under Low-Variance Rewards: Diagnosis and Bounded Recovery27 Aug 2026
- 2026Balancing Performance and Diversity in GRPO Autoregressive Text-to-Image Post-Training19 Jun 2026
Similar, not cited
Close in wording, no citation either way
- 2026Why GRPO Needs Normalization: A Local-Curvature Perspective on Adaptive Gradients
- 2026AdaGRPO: A Capability-Aware Adaptive Enhancement for Flow-based GRPO
- 2025REINFORCE++: Stabilizing Critic-Free Policy Optimization with Global Advantage Normalization
- 2025CoCA: Step-level Reward for Free in RL-based T2I Diffusion Model Fine-tuning
- 2026Curriculum Group Policy Optimization: Adaptive Sampling for Unleashing the Potential of Text-to-Image Generation
- 2026Arena as Offline Reward: Efficient Fine-Grained Preference Optimization for Diffusion Models
- 2026S-GRPO: Unified Post-Training for Large Vision-Language Models
- 2026Finite Difference Flow Optimization for RL Post-Training of Text-to-Image Models