mlsanityData to 7 Oct 2026

Power Reinforcement Post-Training of Text-to-Image Models with Super-Linear Advantage Shaping

arXiv 2605.10937 ↗First posted 11 May 2026cs.CVcited by 3

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

  1. 2026DenseGRPO: From Sparse to Dense Reward for Flow Matching Model Alignment28 Jan
  2. 2026E-GRPO: High Entropy Steps Drive Effective Reinforcement Learning for Flow Models1 Jan
  3. 2025Co-GRPO: Co-Optimized Group Relative Policy Optimization for Masked Diffusion Model25 Dec
  4. 2025DiverseGRPO: Mitigating Mode Collapse in Image Generation via Diversity-Aware GRPO25 Dec
  5. 2025TreeGRPO: Tree-Advantage GRPO for Online RL Post-Training of Diffusion Models9 Dec
  6. 2025Multi-GRPO: Multi-Group Advantage Estimation for Text-to-Image Generation with Tree-Based Trajectories and Multiple Rewards30 Nov
  7. 2025GRPO-Guard: Mitigating Implicit Over-Optimization in Flow Matching via Regulated Clipping25 Oct
  8. 2025Principled RL for Flow Matching Emerges from the Chunk-level Policy Optimization24 Oct
  9. 2025Reinforcement Learning Meets Masked Generative Models: Mask-GRPO for Text-to-Image Generation15 Oct
  10. 2025Smart-GRPO: Smartly Sampling Noise for Efficient RL of Flow-Matching Models3 Oct
  11. 2025STAGE: Stable and Generalizable GRPO for Autoregressive Image Generation29 Sep
  12. 2025A Survey of Reinforcement Learning for Large Reasoning Models10 Sep
14 more
  1. 2025Pref-GRPO: Pairwise Preference Reward-based GRPO for Stable Text-to-Image Reinforcement Learning28 Aug
  2. 2025Uncalibrated Reasoning: GRPO Induces Overconfidence for Stochastic Outcomes15 Aug
  3. 2025AR-GRPO: Training Autoregressive Image Generation Models via Reinforcement Learning9 Aug
  4. 2025TempFlow-GRPO: When Timing Matters for GRPO in Flow Models6 Aug
  5. 2025Geometric-Mean Policy Optimization28 Jul
  6. 2025DanceGRPO: Unleashing GRPO on Visual Generation12 May
  7. 2025Flow-GRPO: Training Flow Matching Models via Online RL8 May
  8. 2025VAPO: Efficient and Reliable Reinforcement Learning for Advanced Reasoning Tasks7 Apr
  9. 2025CPPO: Accelerating the Training of Group Relative Policy Optimization-Based Reasoning Models28 Mar
  10. 2025Understanding R1-Zero-Like Training: A Critical Perspective26 Mar
  11. 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scale18 Mar
  12. 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learning22 Jan
  13. 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Model29 May
  14. 2017Proximal Policy Optimization Algorithms20 Jul

Other references · 87

  1. 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
  2. 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,694
  3. 2023GPT-4 Technical Reportcited by 18,132
  4. 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,865
  5. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
  6. 2022Flow Matching for Generative Modelingcited by 5,426
  7. 2023SDXL: Improving Latent Diffusion Models for High-Resolution Image Synthesiscited by 4,290
  8. 2022LAION-5B: An open large-scale dataset for training next generation image-text modelscited by 4,131
  9. 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesiscited by 4,094
  10. 2017Deep reinforcement learning from human preferencescited by 3,846
  11. 2025Gemini 2.5: Pushing the Frontier with Advanced Reasoning, Multimodality, Long Context, and Next Generation Agentic Capabilitiescited by 3,655
  12. 2019Fine-Tuning Language Models from Human Preferencescited by 2,124
38 more
  1. 2025Gemma 3 Technical Reportcited by 1,686
  2. 2023ImageReward: Learning and Evaluating Human Preferences for Text-to-Image Generationcited by 1,010
  3. 2025Qwen-Image Technical Reportcited by 973
  4. 2024SimPO: Simple Preference Optimization with a Reference-Free Rewardcited by 951
  5. 2023Pick-a-Pic: An Open Dataset of User Preferences for Text-to-Image Generationcited by 950
  6. 2023GenEval: An Object-Focused Framework for Evaluating Text-to-Image Alignmentcited by 946
  7. 2023Training Diffusion Models with Reinforcement Learningcited by 939
  8. 2023Human Preference Score v2: A Solid Benchmark for Evaluating Human Preferences of Text-to-Image Synthesiscited by 847
  9. 2025Group Sequence Policy Optimizationcited by 788
  10. 2021Vector Quantized Diffusion Model for Text-to-Image Synthesiscited by 773
  11. 2023Diffusion Model Alignment Using Direct Preference Optimizationcited by 767
  12. 2023Open Problems and Fundamental Limitations of Reinforcement Learning from Human Feedbackcited by 743
  13. 2016Generative Adversarial Text to Image Synthesiscited by 735
  14. 2024Show-o: One Single Transformer to Unify Multimodal Understanding and Generationcited by 734
  15. 2021CogView: Mastering Text-to-Image Generation via Transformerscited by 691
  16. 2024ORPO: Monolithic Preference Optimization without Reference Modelcited by 556
  17. 2023DPOK: Reinforcement Learning for Fine-tuning Text-to-Image Diffusion Modelscited by 431
  18. 2023Directly Fine-Tuning Diffusion Models on Differentiable Rewardscited by 431
  19. 2023Aligning Text-to-Image Models using Human Feedbackcited by 418
  20. 2018Shampoo: Preconditioned Stochastic Tensor Optimizationcited by 413
  21. 2024Contrastive Preference Optimization: Pushing the Boundaries of LLM Performance in Machine Translationcited by 390
  22. 2022CogView2: Faster and Better Text-to-Image Generation via Hierarchical Transformerscited by 305
  23. 2023Text-to-image Diffusion Models in Generative AI: A Surveycited by 264
  24. 2025MiniMax-M1: Scaling Test-Time Compute Efficiently with Lightning Attentioncited by 248
  25. 2023Using Human Feedback to Fine-tune Diffusion Models without Any Reward Modelcited by 246
  26. 2024Iterative Reasoning Preference Optimizationcited by 244
  27. 2025Show-o2: Improved Native Unified Multimodal Modelscited by 242
  28. 2025DiffusionNFT: Online Diffusion Reinforcement with Forward Processcited by 204
  29. 2025MixGRPO: Unlocking Flow-based GRPO Efficiency with Mixed ODE-SDEcited by 168
  30. 2015Generating Images from Captions with Attentioncited by 135
  31. 2024Direct Preference Optimization with an Offsetcited by 122
  32. 2024ReNO: Enhancing One-step Text-to-Image Models through Reward-based Noise Optimizationcited by 115
  33. 2023End-to-End Diffusion Latent Optimization Improves Classifier Guidancecited by 107
  34. 2024Aligning Diffusion Models by Optimizing Human Utilitycited by 99
  35. 2024LiPO: Listwise Preference Optimization through Learning-to-Rankcited by 94
  36. 2021GAN-Control: Explicitly Controllable GANscited by 91
  37. 2024Self-Play Fine-Tuning of Diffusion Models for Text-to-Image Generationcited by 85
  38. 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

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 3.

3 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026Why GRPO Needs Normalization: A Local-Curvature Perspective on Adaptive Gradients
  2. 2026AdaGRPO: A Capability-Aware Adaptive Enhancement for Flow-based GRPO
  3. 2025REINFORCE++: Stabilizing Critic-Free Policy Optimization with Global Advantage Normalization
  4. 2025CoCA: Step-level Reward for Free in RL-based T2I Diffusion Model Fine-tuning
  5. 2026Curriculum Group Policy Optimization: Adaptive Sampling for Unleashing the Potential of Text-to-Image Generation
  6. 2026Arena as Offline Reward: Efficient Fine-Grained Preference Optimization for Diffusion Models
  7. 2026S-GRPO: Unified Post-Training for Large Vision-Language Models
  8. 2026Finite Difference Flow Optimization for RL Post-Training of Text-to-Image Models