AR-GRPO: Training Autoregressive Image Generation Models via Reinforcement Learning
On the map: LLM reasoning → Reinforcement learning for reasoning
Abstract (arXiv)
Inspired by the success of reinforcement learning (RL) in refining large language models (LLMs), we propose AR-GRPO, an approach to integrate online RL training into autoregressive (AR) image generation models. We adapt the Group Relative Policy Optimization (GRPO) algorithm to refine the vanilla autoregressive models' outputs by carefully designed reward functions that evaluate generated images across multiple quality dimensions, including perceptual quality, realism, and semantic fidelity. We conduct comprehensive experiments on both class-conditional (i.e., class-to-image) and text-conditional (i.e., text-to-image) image generation tasks, demonstrating that our RL-enhanced framework significantly improves both the image quality and human preference of generated images compared to the standard AR baselines. Our results show consistent improvements across various evaluation metrics, establishing the viability of RL-based optimization for AR image generation and opening new avenues for controllable and high-quality image synthesis. The source codes and models are available at: https://github.com/Kwai-Klear/AR-GRPO.
Key facts. Cited by 22 papers in the map (data through 14 Oct 2026). Placed in Reinforcement learning for reasoning, one of 17 directions in LLM reasoning. Builds on 61 of its 72 references that are in the map. First posted on arXiv 9 Aug 2025.
Builds on
61 of 72 references.
Closest ancestors · 7
Same direction, newest first
- 2025Reasoning with Exploration: An Entropy Perspective17 Jun
- 2025Beyond the 80/20 Rule: High-Entropy Minority Tokens Drive Effective Reinforcement Learning for LLM Reasoning2 Jun
- 2025The Entropy Mechanism of Reinforcement Learning for Reasoning Language Models28 May
- 2025DanceGRPO: Unleashing GRPO on Visual Generation12 May
- 2025Flow-GRPO: Training Flow Matching Models via Online RL8 May
- 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scale18 Mar
- 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learning22 Jan
Other references · 54
- 2017Attention Is All You Needcited by 60,361
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
- 2022Training language models to follow instructions with human feedbackcited by 15,971
- 2023LLaMA: Open and Efficient Foundation Language Modelscited by 14,132
- 2020Denoising Diffusion Implicit Modelscited by 9,781
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 9,122
- 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,255
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,058
- 2022Photorealistic Text-to-Image Diffusion Models with Deep Language Understandingcited by 6,355
38 more
- 2022Scalable Diffusion Models with Transformerscited by 6,054
- 2022Classifier-Free Diffusion Guidancecited by 5,847
- 2017Neural Discrete Representation Learningcited by 4,712
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
- 2022Flamingo: a Visual Language Model for Few-Shot Learningcited by 4,386
- 2021Zero-Shot Text-to-Image Generationcited by 4,288
- 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,101
- 2019Analyzing and Improving the Image Quality of StyleGANcited by 3,588
- 2020Taming Transformers for High-Resolution Image Synthesiscited by 3,253
- 2022Scaling Instruction-Finetuned Language Modelscited by 3,024
- 2016Improved Techniques for Training GANscited by 2,970
- 2018Large Scale GAN Training for High Fidelity Natural Image Synthesiscited by 2,626
- 2020GLU Variants Improve Transformercited by 1,804
- 2019Root Mean Square Layer Normalizationcited by 1,480
- 2019Generating Diverse High-Fidelity Images with VQ-VAE-2cited by 1,387
- 2022Scaling Autoregressive Models for Content-Rich Text-to-Image Generationcited by 1,295
- 2023ImageReward: Learning and Evaluating Human Preferences for Text-to-Image Generationcited by 1,045
- 2017StarGAN: Unified Generative Adversarial Networks for Multi-Domain Image-to-Image Translationcited by 1,039
- 2024Visual Autoregressive Modeling: Scalable Image Generation via Next-Scale Predictioncited by 1,023
- 2023Pick-a-Pic: An Open Dataset of User Preferences for Text-to-Image Generationcited by 977
- 2023GenEval: An Object-Focused Framework for Evaluating Text-to-Image Alignmentcited by 972
- 2019Improved Precision and Recall Metric for Assessing Generative Modelscited by 944
- 2023Human Preference Score v2: A Solid Benchmark for Evaluating Human Preferences of Text-to-Image Synthesiscited by 890
- 2016Pixel Recurrent Neural Networkscited by 816
- 2025Janus-Pro: Unified Multimodal Understanding and Generation with Data and Model Scalingcited by 812
- 2024Autoregressive Model Beats Diffusion: Llama for Scalable Image Generationcited by 777
- 2022Autoregressive Image Generation using Residual Quantizationcited by 722
- 2021CogView: Mastering Text-to-Image Generation via Transformerscited by 694
- 2021Vector-quantized Image Modeling with Improved VQGANcited by 658
- 2022MANIQA: Multi-dimension Attention Network for No-Reference Image Quality Assessmentcited by 522
- 2024Janus: Decoupling Visual Encoding for Unified Multimodal Understanding and Generationcited by 457
- 2021Generating Images with Sparse Representationscited by 266
- 2018A Note on the Inception Scorecited by 226
- 2025Unified Reward Model for Multimodal Understanding and Generationcited by 199
- 2025T2I-R1: Reinforcing Image Generation with Collaborative Semantic-level and Token-level CoTcited by 166
- 2024JanusFlow: Harmonizing Autoregression and Rectified Flow for Unified Multimodal Understanding and Generationcited by 163
- 2025Teaching Large Language Models to Regress Accurate Image Quality Scores using Score Distributioncited by 141
- 2017PixelSNAIL: An Improved Autoregressive Generative Modelcited by 137
50 most cited of 54 All 54 (JSON)
Built on by
Citing papers per year: 2025: 10; 2026: 12.
22 papers.
Sort:
By area:
- 2025The Landscape of Agentic Reinforcement Learning for LLMs: A Surveycited by 205
- 2026Flow-OPD: On-Policy Distillation for Flow Matching Modelscited by 34
- 2026JavisDiT++: Unified Modeling and Optimization for Joint Audio-Video Generationcited by 25
- 2025Group Critical-token Policy Optimization for Autoregressive Image Generationcited by 17
- 2025STAGE: Stable and Generalizable GRPO for Autoregressive Image Generationcited by 15
- 2026NextFlow: Unified Sequential Modeling Activates Multimodal Understanding and Generationcited by 14
- 2025Unified Multimodal Models as Auto-Encoderscited by 13
- 2025COOPER: A Unified Model for Cooperative Perception and Reasoning in Spatial Intelligencecited by 9
- 2026Understanding Reward Hacking in Text-to-Image Reinforcement Learningcited by 9
- 2025RubricRL: Simple Generalizable Rewards for Text-to-Image Generationcited by 7
- 2025VA-π: Variational Policy Alignment for Pixel-Aware Autoregressive Generationcited by 7
- 2025IRG-MotionLLM: Interleaving Motion Generation, Assessment and Refinement for Text-to-Motion Generationcited by 6
10 more
- 2025Expand and Prune: Maximizing Trajectory Diversity for Effective GRPO in Generative Modelscited by 6
- 2026ConsistentRFT: Reducing Visual Hallucinations in Flow-based Reinforcement Fine-Tuningcited by 6
- 2026Power Reinforcement Post-Training of Text-to-Image Models with Super-Linear Advantage Shapingcited by 3
- 2026VAR RL Done Right: Tackling Asynchronous Policy Conflicts in Visual Autoregressive Generationcited by 2
- 2026Advances in GRPO for Generation Models: A Surveycited by 2
- 2026Sketch Then Paint: Hierarchical Reinforcement Learning for Diffusion Multi-Modal Large Language Modelscited by 1
- 2025Enhancing Blind Face Restoration through Online Reinforcement Learningcited by 0
- 2026Policy-based Tuning of Autoregressive Image Models with Instance- and Distribution-Level Rewardscited by 0
- 2026UniVR: Thinking in Visual Space for Unified Visual Reasoningcited by 0
- 2026JAGG: Jacobian-Aggregated Group Gradient for Efficient GRPO Training of Diffusion Modelscited by 0
- 2026JAGG: Jacobian-Aggregated Group Gradient for Efficient GRPO Training of Diffusion Models20 Jul 2026
- 2026UniVR: Thinking in Visual Space for Unified Visual Reasoning14 Jul 2026
- 2026Sketch Then Paint: Hierarchical Reinforcement Learning for Diffusion Multi-Modal Large Language Models16 May 2026
- 2026Power Reinforcement Post-Training of Text-to-Image Models with Super-Linear Advantage Shaping11 May 2026
- 2026Flow-OPD: On-Policy Distillation for Flow Matching Models8 May 2026
- 2026Policy-based Tuning of Autoregressive Image Models with Instance- and Distribution-Level Rewards24 Mar 2026
- 2026JavisDiT++: Unified Modeling and Optimization for Joint Audio-Video Generation22 Feb 2026
- 2026Advances in GRPO for Generation Models: A Survey21 Feb 2026
- 2026ConsistentRFT: Reducing Visual Hallucinations in Flow-based Reinforcement Fine-Tuning3 Feb 2026
- 2026Understanding Reward Hacking in Text-to-Image Reinforcement Learning6 Jan 2026
- 2026VAR RL Done Right: Tackling Asynchronous Policy Conflicts in Visual Autoregressive Generation5 Jan 2026
- 2026NextFlow: Unified Sequential Modeling Activates Multimodal Understanding and Generation5 Jan 2026
10 more
- 2025VA-π: Variational Policy Alignment for Pixel-Aware Autoregressive Generation22 Dec 2025
- 2025Expand and Prune: Maximizing Trajectory Diversity for Effective GRPO in Generative Models17 Dec 2025
- 2025IRG-MotionLLM: Interleaving Motion Generation, Assessment and Refinement for Text-to-Motion Generation11 Dec 2025
- 2025COOPER: A Unified Model for Cooperative Perception and Reasoning in Spatial Intelligence4 Dec 2025
- 2025RubricRL: Simple Generalizable Rewards for Text-to-Image Generation25 Nov 2025
- 2025STAGE: Stable and Generalizable GRPO for Autoregressive Image Generation29 Sep 2025
- 2025Enhancing Blind Face Restoration through Online Reinforcement Learning27 Sep 2025
- 2025Group Critical-token Policy Optimization for Autoregressive Image Generation26 Sep 2025
- 2025Unified Multimodal Models as Auto-Encoders11 Sep 2025
- 2025The Landscape of Agentic Reinforcement Learning for LLMs: A Survey2 Sep 2025
- 2025STAGE: Stable and Generalizable GRPO for Autoregressive Image Generationcited by 15
- 2025COOPER: A Unified Model for Cooperative Perception and Reasoning in Spatial Intelligencecited by 9
- 2025Expand and Prune: Maximizing Trajectory Diversity for Effective GRPO in Generative Modelscited by 6
- 2026Power Reinforcement Post-Training of Text-to-Image Models with Super-Linear Advantage Shapingcited by 3
- 2026Advances in GRPO for Generation Models: A Surveycited by 2
- 2026UniVR: Thinking in Visual Space for Unified Visual Reasoningcited by 0
- 2026JAGG: Jacobian-Aggregated Group Gradient for Efficient GRPO Training of Diffusion Modelscited by 0
- 2026JAGG: Jacobian-Aggregated Group Gradient for Efficient GRPO Training of Diffusion Models20 Jul 2026
- 2026UniVR: Thinking in Visual Space for Unified Visual Reasoning14 Jul 2026
- 2026Power Reinforcement Post-Training of Text-to-Image Models with Super-Linear Advantage Shaping11 May 2026
- 2026Advances in GRPO for Generation Models: A Survey21 Feb 2026
- 2025Expand and Prune: Maximizing Trajectory Diversity for Effective GRPO in Generative Models17 Dec 2025
- 2025COOPER: A Unified Model for Cooperative Perception and Reasoning in Spatial Intelligence4 Dec 2025
- 2025STAGE: Stable and Generalizable GRPO for Autoregressive Image Generation29 Sep 2025
- 2026Flow-OPD: On-Policy Distillation for Flow Matching Modelscited by 34
- 2026JavisDiT++: Unified Modeling and Optimization for Joint Audio-Video Generationcited by 25
- 2025Group Critical-token Policy Optimization for Autoregressive Image Generationcited by 17
- 2026NextFlow: Unified Sequential Modeling Activates Multimodal Understanding and Generationcited by 14
- 2025Unified Multimodal Models as Auto-Encoderscited by 13
- 2025IRG-MotionLLM: Interleaving Motion Generation, Assessment and Refinement for Text-to-Motion Generationcited by 6
- 2026VAR RL Done Right: Tackling Asynchronous Policy Conflicts in Visual Autoregressive Generationcited by 2
- 2026Policy-based Tuning of Autoregressive Image Models with Instance- and Distribution-Level Rewardscited by 0
- 2026Policy-based Tuning of Autoregressive Image Models with Instance- and Distribution-Level Rewards24 Mar 2026
- 2026JavisDiT++: Unified Modeling and Optimization for Joint Audio-Video Generation22 Feb 2026
- 2026VAR RL Done Right: Tackling Asynchronous Policy Conflicts in Visual Autoregressive Generation5 Jan 2026
- 2026NextFlow: Unified Sequential Modeling Activates Multimodal Understanding and Generation5 Jan 2026
- 2025IRG-MotionLLM: Interleaving Motion Generation, Assessment and Refinement for Text-to-Motion Generation11 Dec 2025
- 2025Group Critical-token Policy Optimization for Autoregressive Image Generation26 Sep 2025
- 2025Unified Multimodal Models as Auto-Encoders11 Sep 2025
Similar, not cited
Close in wording, no citation either way
- 2025Delving into RL for Image Generation with CoT: A Study on DPO vs. GRPO
- 2025Reinforcement Learning Meets Masked Generative Models: Mask-GRPO for Text-to-Image Generation
- 2025Fine-Tuning Next-Scale Visual Autoregressive Models with Group Relative Policy Optimization
- 2025X-Omni: Reinforcement Learning Makes Discrete Autoregressive Image Generative Models Great Again
- 2025Integrating Reinforcement Learning with Visual Generative Models: Foundations and Advances
- 2026Balancing Performance and Diversity in GRPO Autoregressive Text-to-Image Post-Training
- 2026UniGRPO: Unified Policy Optimization for Reasoning-Driven Visual Generation
- 2026From Broad Exploration to Stable Synthesis: Entropy-Guided Optimization for Autoregressive Image Generation