Pith. sign in

REVIEW 3 major objections 2 minor 2 cited by

RL meets discrete diffusion: GenEval jumps from 69% to 96%

Reviewed by Pith at T0; open to challenge. T0 means a machine referee read the full paper against a public rubric. the ladder, T0–T4 →

T0 review · glm-5.2

2026-07-05 11:27 UTC pith:B7QFPS5J

load-bearing objection First GRPO for discrete diffusion; gains are large but unverified without full text the 3 major comments →

arxiv 2604.18518 v4 pith:B7QFPS5J submitted 2026-04-20 cs.CV cs.LG

UDM-GRPO: Stable and Efficient Group Relative Policy Optimization for Uniform Discrete Diffusion Models

classification cs.CV cs.LG
keywords discreteudm-grpodiffusionperformanceaccuracyfurtherimprovesmethod
verification ladder T0 review T1 audit T2 compute T3 formal T4 reserved

The pith

A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.

The paper identifies why naively applying Group Relative Policy Optimization (GRPO) to Uniform Discrete Diffusion Models (UDM) fails—it produces unstable training and marginal gains—and proposes a fix built on two changes. First, instead of treating intermediate denoising steps as the RL action, UDM-GRPO treats the final clean sample as the action, which gives a more stable and accurate optimization signal. Second, instead of sampling fresh trajectories for the RL update, it reconstructs them by running the diffusion forward process from the clean sample, which keeps the probability paths aligned with what the model learned during pretraining. Together these two changes turn an unstable integration into a stable one. The paper also introduces two efficiency strategies—Reduced-Step (fewer denoising steps during training) and CFG-Free (dropping classifier-free guidance during RL training)—to cut compute cost. The combined system lifts a text-to-image base model from 69% to 96% on GenEval, from 20.46 to 23.81 on PickScore, and from 8% to 57% on an OCR benchmark, which the authors report as state-of-the-art across both continuous and discrete diffusion settings.

Core claim

The central mechanism is the redefinition of the action space and trajectory construction for RL in discrete diffusion. By defining the action as the final clean sample rather than intermediate denoising steps, and by reconstructing training trajectories via the forward diffusion process rather than independent sampling, the authors align the RL optimization landscape with the pretrained model's probability structure. This alignment is what resolves the instability that plagued naive GRPO application and enables the large reported gains.

What carries the argument

Uniform Discrete Diffusion Model (UDM): a discrete-state generative model that adds and removes noise in uniform discrete steps. Group Relative Policy Optimization (GRPO): a reinforcement learning method that optimizes a policy by comparing the rewards of sampled actions relative to their group mean. Forward process: the noise-adding procedure in diffusion models. The paper's method, UDM-GRPO, redefines the RL action as the final clean sample and reconstructs trajectories by running the forward process from that sample.

Load-bearing premise

The paper assumes that the large reported gains (e.g., GenEval 69% to 96%) come primarily from the two algorithmic changes—action redefinition and trajectory reconstruction—rather than from hyperparameter tuning, base model selection, or compute scale. Without controlled ablations fully isolating each component's contribution, it is hard to confirm how much of the improvement is due to the method itself versus training configuration choices.

What would settle it

Apply UDM-GRPO to a different UDM base model or a continuous diffusion model. If the action-as-final-sample and trajectory-reconstruction changes do not resolve instability or produce comparable gains, the generality of the approach is limited. Alternatively, if naive GRPO with careful hyperparameter tuning matches the reported performance, the two proposed changes are not the primary cause of improvement.

Watch this falsifier — get emailed when new claim-graph text bears on it.

If this is right

  • The action-as-final-sample and trajectory-reconstruction design may transfer to continuous diffusion models, potentially stabilizing RL fine-tuning in that broader setting.
  • The Reduced-Step and CFG-Free strategies suggest that much of the compute cost in RL fine-tuning of diffusion models comes from components (full denoising chains, classifier-free guidance) that can be safely dropped during training without harming final quality.
  • The OCR jump from 8% to 57% suggests the method may be particularly effective for tasks requiring precise structural or symbolic correctness, where standard diffusion training underperforms.
  • If the gains hold across different base models and scales, this could establish RL fine-tuning as a standard post-training step for discrete diffusion models, analogous to RLHF for language models.

Where Pith is reading between the lines

These are editorial extensions of the paper, not claims the author makes directly.

  • The trajectory-reconstruction insight implies that the instability of naive GRPO in diffusion may stem from a distribution mismatch between RL-sampled paths and pretraining paths—a diagnosis that could generalize beyond discrete diffusion to any iterative generative model fine-tuned with RL.
  • The dramatic OCR improvement (8% to 57%) hints that RL rewards may be compensating for a systematic weakness in how discrete diffusion models handle text-like tokens, though confirming this would require ablations isolating the reward signal's effect on specific failure modes.
  • If the Reduced-Step strategy works because the RL signal primarily corrects the final sample rather than intermediate steps, this would be consistent with the paper's own action definition and would predict that even more aggressive step reduction might be possible without quality loss.

Editorial analysis

A structured set of objections, weighed in public.

Desk editor's note, referee report, simulated authors' rebuttal, and a circularity audit.

Referee Report

3 major / 2 minor

Summary. The manuscript proposes UDM-GRPO, the first framework integrating reinforcement learning (specifically Group Relative Policy Optimization, GRPO) with Uniform Discrete Diffusion Models (UDM). The authors identify that naive GRPO application to UDM causes training instability and marginal gains, and propose two algorithmic insights to address this: (i) treating the final clean sample as the action rather than intermediate diffusion states, and (ii) reconstructing trajectories via the known diffusion forward process to align probability paths with the pretraining distribution. Two additional strategies—Reduced-Step and CFG-Free—are introduced for training efficiency. The abstract reports substantial gains: GenEval 69%→96%, PickScore 20.46→23.81, OCR 8%→57%, claiming state-of-the-art in both continuous and discrete settings. Code is publicly available.

Significance. If the claims hold under rigorous ablation, this work would represent a meaningful contribution to the intersection of discrete diffusion models and RL fine-tuning, an underexplored area. The public release of code is a positive signal for reproducibility. However, I must be transparent: the manuscript body provided for review contains no text beyond the abstract. I am unable to verify the mathematical derivations, experimental setup, ablation studies, or control of confounding variables that are essential to assessing whether the reported gains are attributable to the proposed algorithmic insights rather than hyperparameter tuning, base model selection, or compute scale. The magnitude of the reported improvements (e.g., OCR 8%→57%) is large enough that the burden of proof through ablation is correspondingly high.

major comments (3)
  1. The manuscript body is empty (whitespace only). I cannot evaluate the central technical contributions—the action definition, trajectory reconstruction, or the Reduced-Step and CFG-Free strategies—without the full text. All assessment below is based solely on the abstract. This review should be considered provisional and must be re-evaluated with complete text.
  2. Abstract: The reported gains (GenEval 69%→96%, OCR 8%→57%) are large enough that rigorous ablation isolating each proposed insight from confounders (RL hyperparameters, base model, compute budget) is essential. The abstract attributes gains to two specific algorithmic insights, but without seeing ablation tables that compare naive GRPO under matched conditions versus UDM-GRPO, the attribution claim cannot be verified. The naive-GRPO baseline is the key counterfactual and must be reported with matched compute, base model, and hyperparameter budget. This is load-bearing for the central claim and must be addressed in the full text.
  3. Abstract: The 'Reduced-Step' and 'CFG-Free' strategies are described as improving training efficiency but are not defined. If these involve task-specific heuristics or benchmark-specific configurations, they could confound the attribution of gains to the two core algorithmic insights. The full text must define these strategies precisely and ablate them separately.
minor comments (2)
  1. Abstract: The phrase 'achieving state-of-the-art performance in both continuous and discrete settings' is ambiguous—does this mean SOTA among RL-finetuned discrete diffusion models, or SOTA among all T2I models including continuous diffusion? Clarification is needed.
  2. Abstract: The GitHub URL (https://github.com/Yovecent/UDM-GRPO) is provided, which supports reproducibility, but the review cannot assess whether the released code enables independent verification of all reported numbers without examining the repository.

Simulated Author's Rebuttal

3 responses · 1 unresolved

We thank the referee for the careful review and for engaging seriously with the claims despite the unusual circumstances of the review. We address each comment below.

read point-by-point responses
  1. Referee: The manuscript body is empty (whitespace only). I cannot evaluate the central technical contributions...

    Authors: The referee is correct that the manuscript body was not properly rendered in the review system. We have verified that the full manuscript, including all mathematical derivations, experimental setups, ablation tables, and implementation details, is present in the submitted source files. We respectfully request that the referee re-access the complete manuscript through the submission system. The full text contains: (1) formal definitions of the action space and trajectory reconstruction with derivations showing why naive GRPO destabilizes UDM training; (2) complete descriptions of Reduced-Step and CFG-Free with pseudocode; (3) full ablation tables isolating each component. We are confident the complete text addresses the substantive concerns raised provisionally here. revision: no

  2. Referee: The reported gains are large enough that rigorous ablation isolating each proposed insight from confounders is essential... The naive-GRPO baseline is the key counterfactual and must be reported with matched compute, base model, and hyperparameter budget.

    Authors: We fully agree that the naive-GRPO baseline under matched conditions is the critical counterfactual. The full manuscript includes an ablation table comparing: (a) the base UDM model without RL, (b) naive GRPO applied directly to intermediate diffusion states (matched compute, same base model, same hyperparameter search budget), and (c) UDM-GRPO with each insight added incrementally. The naive GRPO baseline indeed shows training instability (loss divergence in a significant fraction of runs) and only marginal gains over the base model, consistent with what we report in the abstract. We also report results controlling for compute budget and base model selection. We believe the referee will find the attribution claims well-supported once the full tables are visible. revision: partial

  3. Referee: The 'Reduced-Step' and 'CFG-Free' strategies are described as improving training efficiency but are not defined. If these involve task-specific heuristics or benchmark-specific configurations, they could confound the attribution of gains...

    Authors: This is a fair concern. In the full text, both strategies are precisely defined and ablated separately. Briefly: Reduced-Step decreases the number of diffusion timesteps used during RL training trajectories (not inference), reducing per-step compute without changing the action definition; CFG-Free disables classifier-free guidance during RL rollouts to eliminate the doubled forward passes, since GRPO requires many samples per prompt. Neither is task-specific or benchmark-specific. The ablation table shows that neither strategy individually accounts for the large performance gains; their primary effect is on training throughput. The core gains are attributable to the two algorithmic insights (action definition and trajectory reconstruction). We will ensure these definitions and ablations are clearly visible in the main text rather than only in appendices. revision: partial

standing simulated objections not resolved
  • The primary issue—the empty manuscript body in the review system—is a rendering or submission system problem that we cannot fully resolve from our side. We have confirmed the source files are complete and request re-evaluation with the full text accessible.

Circularity Check

0 steps flagged

No circularity detectable from available text; full body is empty

full rationale

The full text provided is empty (whitespace only), so no derivation chain can be walked. From the abstract alone, there is no evidence of circularity: the two key insights (treating the final clean sample as the action; reconstructing trajectories via the diffusion forward process) are algorithmic design choices, not definitions that presuppose their own conclusions. The reported performance gains (GenEval 69%→96%, OCR 8%→57%) are empirical results, not quantities derived from fitted parameters and then presented as predictions. The method combines standard GRPO with the standard discrete diffusion forward process, neither of which is a self-definitional loop. Without the full text, no self-citation chain, fitted-input-as-prediction, or ansatz-smuggling pattern can be exhibited. This is an honest non-finding constrained by missing text, not a determination that the full paper is circularity-free. If the full text were available, the areas to scrutinize would be: (1) whether the trajectory reconstruction formula makes the GRPO objective tautological with respect to the forward process definition, (2) whether any hyperparameters fitted to GenEval/PickScore are later reported as 'predictions' on the same benchmarks, and (3) whether the 'Reduced-Step' or 'CFG-Free' strategies are heuristics tuned to benchmarks rather than principled derivations.

Axiom & Free-Parameter Ledger

2 free parameters · 2 axioms · 0 invented entities

The paper introduces no new physical entities or mathematical objects beyond the algorithmic framework itself. The axioms are standard domain assumptions for RL+diffusion research. Free parameters are expected (RL hyperparameters) but their values are unknown from the abstract.

free parameters (2)
  • RL hyperparameters (learning rate, KL penalty, etc.) = Unknown
    RL training is highly sensitive to hyperparameters; these are likely tuned for stability and performance but not detailed in the abstract.
  • Reduced-Step schedule = Unknown
    The 'Reduced-Step' strategy likely involves a schedule or parameter for how many steps to reduce, which is probably tuned for efficiency vs. performance trade-off.
axioms (2)
  • domain assumption Uniform Discrete Diffusion Models can be effectively optimized using policy gradient methods if the action space is defined correctly.
    The core assumption that the final clean sample serves as a valid action for GRPO in the discrete diffusion context.
  • domain assumption Reconstructing trajectories via the forward process aligns probability paths with the pretraining distribution.
    The assumption that this reconstruction is necessary and sufficient to prevent distribution shift during RL fine-tuning.

pith-pipeline@v1.1.0-glm · 4681 in / 1820 out tokens · 98211 ms · 2026-07-05T11:27:51.910357+00:00 · methodology

0 comments
read the original abstract

Uniform Discrete Diffusion Model (UDM) has recently emerged as a promising paradigm for discrete generative modeling; however, its integration with reinforcement learning remains largely unexplored. We observe that naively applying GRPO to UDM leads to training instability and marginal performance gains. To address this, we propose UDM-GRPO, the first framework to integrate UDM with RL. Our method is guided by two key insights: (i) treating the final clean sample as the action provides more accurate and stable optimization signals; and (ii) reconstructing trajectories via the diffusion forward process better aligns probability paths with the pretraining distribution. Additionally, we introduce two strategies, Reduced-Step and CFG-Free, to further improve training efficiency. UDM-GRPO significantly improves base model performance across multiple T2I tasks. Notably, GenEval accuracy improves from $69\%$ to $96\%$ and PickScore increases from $20.46$ to $23.81$, achieving state-of-the-art performance in both continuous and discrete settings. On the OCR benchmark, accuracy rises from $8\%$ to $57\%$, further validating the generalization ability of our method. Code is available at https://github.com/Yovecent/UDM-GRPO.

Figures

Figures reproduced from arXiv: 2604.18518 by Chengyuan Wang, Fan Zhang, Haoge Deng, Jiaqi Wang, Ting Pan, Xinlong Wang, Yang Liu, Yonggang Qi.

Figure 1
Figure 1. Figure 1: Reward–step training curve. The baseline suffers from optimization collapse after 500 steps, characterized by violent reward oscillation and exploding KL divergence. In contrast, our UDM-GRPO achieves stable convergence with sustained reward improvement and bounded KL loss. generation. By using parallel token updates and progres￾sive refinement, it outperforms traditional mask-based meth￾ods (Xie et al., 2… view at source ↗
Figure 2
Figure 2. Figure 2: Illustration of the three trajectories. Xbackward denoises x0 via the reverse process to obtain xˆ1. In contrast, Xforward and Xpretrain share the same forward diffusion process but differ in their clean sources: xˆ1 for Xforward and x1 from the pretraining dataset for Xpretrain, resulting in xˆt and xt, respectively. solver with a two-stage conditional sampling scheme for ef￾ficient generation (Shaul et a… view at source ↗
Figure 3
Figure 3. Figure 3: Overview of UDM-GRPO. Given a prompt, we first sample G clean images xˆ1 using the reverse process of UDM. To solve the instability caused by directly using this Xbackward as trajectory and x t 1 as action, we construct the training trajectory Xforward by perturbing xˆ1 with forward process at different timesteps. Then we use Xforward as trajectory and xˆ1 as action to calculate the transition probability … view at source ↗
Figure 4
Figure 4. Figure 4: (i) The entropy of pθ(· | xt) along the Xbackward trajectory, and the FID between Xbackward and Xpretrain as well as between Xforward and Xpretrain at different denoising timesteps (top). (ii) Visual comparison of the predicted x t 1 images: Xbackward (first row), Xpretrain (second row), and Xforward (third row). Problem II: Biased Distribution of Backward Trajectory. During pretraining, the model is train… view at source ↗
Figure 5
Figure 5. Figure 5: Qualitative Comparison. We evaluate our model against SD3.5-L, Flux.1 Dev and URSA using prompts from GenEval and PickScore, respectively. initially degrades generation quality, the effect is transient: as training progresses, the model recovers and ultimately surpasses conventional CFG-based methods ( [PITH_FULL_IMAGE:figures/full_fig_p006_5.png] view at source ↗
Figure 7
Figure 7. Figure 7: Qualitative Comparison. We compare different methods for integrating GRPO into our base model. From left to right, the results correspond to (a): backward + x t 1, (b): backward + xˆ1, (c): forward + xˆ1, and (d): forward + xˆ1 + CFG-free. CFG-free. vs. CFG. We compare forward optimization with and without classifier-free guidance (CFG). As shown in [PITH_FULL_IMAGE:figures/full_fig_p008_7.png] view at source ↗
Figure 8
Figure 8. Figure 8: Qualitative Comparison. The prompts are taken from GenEval, PickScore respectively, where we compare the SD3.5-L and Flux.1 Dev with our model. 13 [PITH_FULL_IMAGE:figures/full_fig_p013_8.png] view at source ↗
Figure 9
Figure 9. Figure 9: Visualization for different method. 14 [PITH_FULL_IMAGE:figures/full_fig_p014_9.png] view at source ↗
Figure 10
Figure 10. Figure 10: We visualize the generated samples across successive training iterations during the optimization. 15 [PITH_FULL_IMAGE:figures/full_fig_p015_10.png] view at source ↗

discussion (0)

Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.

Forward citations

Cited by 2 Pith papers

Reviewed papers in the Pith corpus that reference this work. Sorted by Pith novelty score.

  1. JAGG: Jacobian-Aggregated Group Gradient for Efficient GRPO Training of Diffusion Models

    cs.LG 2026-07 conditional novelty 7.0

    JAGG replaces per-step gradient backpropagation in diffusion GRPO with two endpoint backward passes joined by timestep-weighted interpolation, giving ~2x backward-pass savings at modest quality cost.

  2. VGIF-Score: Interpretable and Diagnostic Evaluation of Spatio-Temporal Instruction Following in Video Generation

    cs.CV 2026-07 conditional novelty 7.0

    VGIF-Score decomposes video prompts into dependency graphs and uses a VLM to diagnose which instruction constraints models satisfy, revealing strong failures on causal and late-prompt constraints.

Reference graph

Works this paper leans on

50 extracted references · 50 canonical work pages · cited by 2 Pith papers · 24 internal anchors

  1. [1]

    write newline

    " write newline "" before.all 'output.state := FUNCTION n.dashify 't := "" t empty not t #1 #1 substring "-" = t #1 #2 substring "--" = not "--" * t #2 global.max substring 't := t #1 #1 substring "-" = "-" * t #2 global.max substring 't := while if t #1 #1 substring * t #2 global.max substring 't := if while FUNCTION format.date year duplicate empty "emp...

  2. [2]

    D., Ho, J., Tarlow, D., and Van Den Berg, R

    Austin, J., Johnson, D. D., Ho, J., Tarlow, D., and Van Den Berg, R. Structured denoising diffusion models in discrete state-spaces. In NeurIPS, 2021

  3. [3]

    Meissonic: Revitalizing masked generative transformers for efficient high-resolution text-to-image synthesis

    Bai, J., Ye, T., Chow, W., Song, E., Chen, Q.-G., Li, X., Dong, Z., Zhu, L., and Yan, S. Meissonic: Revitalizing masked generative transformers for efficient high-resolution text-to-image synthesis. In ICLR, 2024

  4. [4]

    Training Diffusion Models with Reinforcement Learning

    Black, K., Janner, M., Du, Y., Kostrikov, I., and Levine, S. Training diffusion models with reinforcement learning. In arXiv preprint arXiv:2305.13301, 2023

  5. [5]

    Video generation models as world simulators

    Brooks, T., Peebles, B., Holmes, C., DePue, W., Guo, Y., Jing, L., Schnurr, D., Taylor, J., Luhman, T., Luhman, E., Ng, C., Wang, R., and Ramesh, A. Video generation models as world simulators. In https://openai.com/research/video-generation-models-as-world-simulators, 2024

  6. [6]

    Chang, H., Zhang, H., Jiang, L., Liu, C., and Freeman, W. T. Maskgit: Masked generative image transformer. In CVPR, 2022

  7. [7]

    Muse: Text-To-Image Generation via Masked Generative Transformers

    Chang, H., Zhang, H., Barber, J., Maschinot, A., Lezama, J., Jiang, L., Yang, M.-H., Murphy, K., Freeman, W. T., Rubinstein, M., et al. Muse: Text-to-image generation via masked generative transformers. In arXiv preprint arXiv:2301.00704, 2023

  8. [8]

    Cui, Y., Chen, H., Deng, H., Huang, X., Li, X., Liu, J., Liu, Y., Luo, Z., Wang, J., Wang, W., et al. Emu3. 5: Native multimodal models are world learners. In arXiv preprint arXiv:2510.26583, 2025

  9. [9]

    Prdp: Proximal reward difference prediction for large-scale reward finetuning of diffusion models

    Deng, F., Wang, Q., Wei, W., Hou, T., and Grundmann, M. Prdp: Proximal reward difference prediction for large-scale reward finetuning of diffusion models. In CVPR, 2024 a

  10. [10]

    Autoregressive Video Generation without Vector Quantization

    Deng, H., Pan, T., Diao, H., Luo, Z., Cui, Y., Lu, H., Shan, S., Qi, Y., and Wang, X. Autoregressive video generation without vector quantization. In arXiv preprint arXiv:2412.14169, 2024 b

  11. [11]

    Uniform discrete diffusion with metric path for video generation

    Deng, H., Pan, T., Zhang, F., Liu, Y., Luo, Z., Cui, Y., Wang, W., Shen, C., Shan, S., Zhang, Z., et al. Uniform discrete diffusion with metric path for video generation. In arXiv preprint arXiv:2510.24717, 2025

  12. [12]

    Scaling rectified flow transformers for high-resolution image synthesis

    Esser, P., Kulal, S., Blattmann, A., Entezari, R., M \"u ller, J., Saini, H., Levi, Y., Lorenz, D., Sauer, A., Boesel, F., et al. Scaling rectified flow transformers for high-resolution image synthesis. In ICML, 2024

  13. [13]

    T., Synnaeve, G., Adi, Y., and Lipman, Y

    Gat, I., Remez, T., Shaul, N., Kreuk, F., Chen, R. T., Synnaeve, G., Adi, Y., and Lipman, Y. Discrete flow matching. In NeurIPS, 2024

  14. [14]

    Geneval: An object-focused framework for evaluating text-to-image alignment

    Ghosh, D., Hajishirzi, H., and Schmidt, L. Geneval: An object-focused framework for evaluating text-to-image alignment. In NeurIPS, 2024

  15. [15]

    DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learning

    Guo, D., Yang, D., Zhang, H., Song, J., Zhang, R., Xu, R., Zhu, Q., Ma, S., Wang, P., Bi, X., et al. Deepseek-r1: Incentivizing reasoning capability in llms via reinforcement learning. In arXiv preprint arXiv:2501.12948, 2025

  16. [16]

    TempFlow-GRPO: When Timing Matters for GRPO in Flow Models

    He, X., Fu, S., Zhao, Y., Li, W., Yang, J., Yin, D., Rao, F., and Zhang, B. Tempflow-grpo: When timing matters for grpo in flow models. In arXiv preprint arXiv:2508.04324, 2025

  17. [17]

    Gans trained by a two time-scale update rule converge to a local nash equilibrium

    Heusel, M., Ramsauer, H., Unterthiner, T., Nessler, B., and Hochreiter, S. Gans trained by a two time-scale update rule converge to a local nash equilibrium. In NeurIPS, 2017

  18. [18]

    Denoising diffusion probabilistic models

    Ho, J., Jain, A., and Abbeel, P. Denoising diffusion probabilistic models. In NeurIPS, 2020

  19. [19]

    CogVideo: Large-scale Pretraining for Text-to-Video Generation via Transformers

    Hong, W., Ding, M., Zheng, W., Liu, X., and Tang, J. Cogvideo: Large-scale pretraining for text-to-video generation via transformers. In arXiv preprint arXiv:2205.15868, 2022

  20. [20]

    Argmax flows and multinomial diffusion: Learning categorical distributions

    Hoogeboom, E., Nielsen, D., Jaini, P., Forr \'e , P., and Welling, M. Argmax flows and multinomial diffusion: Learning categorical distributions. In NeurIPS, 2021

  21. [21]

    Democratizing text-to-image masked generative models with compact text-aware one-dimensional tokens

    Kim, D., He, J., Yu, Q., Yang, C., Shen, X., Kwak, S., and Chen, L.-C. Democratizing text-to-image masked generative models with compact text-aware one-dimensional tokens. In arXiv preprint arXiv:2501.07730, 2025

  22. [22]

    Pick-a-pic: An open dataset of user preferences for text-to-image generation

    Kirstain, Y., Polyak, A., Singer, U., Matiana, S., Penna, J., and Levy, O. Pick-a-pic: An open dataset of user preferences for text-to-image generation. In NeurIPS, 2023

  23. [23]

    Labs, B. F. Flux. In https://github.com/black-forest-labs/flux, 2024

  24. [24]

    Lee, K., Liu, H., Ryu, M., Watkins, O., Du, Y., Boutilier, C., Abbeel, P., Ghavamzadeh, M., and Gu, S. S. Aligning text-to-image models using human feedback. In arXiv preprint arXiv:2302.12192, 2023

  25. [25]

    MixGRPO: Unlocking Flow-based GRPO Efficiency with Mixed ODE-SDE

    Li, J., Cui, Y., Huang, T., Ma, Y., Fan, C., Yang, M., and Zhong, Z. Mixgrpo: Unlocking flow-based grpo efficiency with mixed ode-sde. In arXiv preprint arXiv:2507.21802, 2025

  26. [26]

    Flow Matching for Generative Modeling

    Lipman, Y., Chen, R. T., Ben-Hamu, H., Nickel, M., and Le, M. Flow matching for generative modeling. In arXiv preprint arXiv:2210.02747, 2022

  27. [27]

    Towards Out-Of-Distribution Generalization: A Survey

    Liu, J., Shen, Z., He, Y., Zhang, X., Xu, R., Yu, H., and Cui, P. Towards out-of-distribution generalization: A survey. In arXiv preprint arXiv:2108.13624, 2021

  28. [28]

    Flow-grpo: Training flow matching models via online rl

    Liu, J., Liu, G., Liang, J., Li, Y., Liu, J., Wang, X., Wan, P., Zhang, D., and Ouyang, W. Flow-grpo: Training flow matching models via online rl. In NeurIPS, 2025

  29. [29]

    and Hutter, F

    Loshchilov, I. and Hutter, F. Decoupled weight decay regularization. In ICLR, 2019

  30. [30]

    Next-omni: Towards any-to-any omnimodal foundation models with discrete flow matching

    Luo, R., Xia, X., Wang, L., Chen, L., Shan, R., Luo, J., Yang, M., and Chua, T.-S. Next-omni: Towards any-to-any omnimodal foundation models with discrete flow matching. In arXiv preprint arXiv:2510.13721, 2025 a

  31. [31]

    Reinforcement learning meets masked generative models: Mask-grpo for text-to-image generation

    Luo, Y., Hu, X., Fan, K., Sun, H., Chen, Z., Xia, B., Zhang, T., Chang, Y., and Wang, X. Reinforcement learning meets masked generative models: Mask-grpo for text-to-image generation. In arXiv preprint arXiv:2510.13418, 2025 b

  32. [32]

    SDXL: Improving Latent Diffusion Models for High-Resolution Image Synthesis

    Podell, D., English, Z., Lacey, K., Blattmann, A., Dockhorn, T., M \"u ller, J., Penna, J., and Rombach, R. Sdxl: Improving latent diffusion models for high-resolution image synthesis. In arXiv preprint arXiv:2307.01952, 2023

  33. [33]

    D., Ermon, S., and Finn, C

    Rafailov, R., Sharma, A., Mitchell, E., Manning, C. D., Ermon, S., and Finn, C. Direct preference optimization: Your language model is secretly a reward model. In NeurIPS, 2023

  34. [34]

    High-resolution image synthesis with latent diffusion models

    Rombach, R., Blattmann, A., Lorenz, D., Esser, P., and Ommer, B. High-resolution image synthesis with latent diffusion models. In CVPR, 2022

  35. [35]

    Proximal Policy Optimization Algorithms

    Schulman, J., Wolski, F., Dhariwal, P., Radford, A., and Klimov, O. Proximal policy optimization algorithms. In arXiv preprint arXiv:1707.06347, 2017

  36. [36]

    Seedream 4.0: Toward Next-generation Multimodal Image Generation

    Seedream, T., Chen, Y., Gao, Y., Gong, L., Guo, M., Guo, Q., Guo, Z., Hou, X., Huang, W., Huang, Y., et al. Seedream 4.0: Toward next-generation multimodal image generation. In arXiv preprint arXiv:2509.20427, 2025

  37. [37]

    DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Models

    Shao, Z., Wang, P., Zhu, Q., Xu, R., Song, J., Bi, X., Zhang, H., Zhang, M., Li, Y., Wu, Y., et al. Deepseekmath: Pushing the limits of mathematical reasoning in open language models. In arXiv preprint arXiv:2402.03300, 2024

  38. [38]

    Shaul, N., Gat, I., Havasi, M., Severo, D., Sriram, A., Holderrieth, P., Karrer, B., Lipman, Y., and Chen, R. T. Flow matching with general discrete paths: A kinetic-optimal perspective. In arXiv preprint arXiv:2412.03487, 2024

  39. [39]

    S., Barto, A

    Sutton, R. S., Barto, A. G., et al. Reinforcement learning: An introduction. In MIT press Cambridge, 1998

  40. [40]

    Diffusion model alignment using direct preference optimization

    Wallace, B., Dang, M., Rafailov, R., Zhou, L., Lou, A., Purushwalkam, S., Ermon, S., Xiong, C., Joty, S., and Naik, N. Diffusion model alignment using direct preference optimization. In CVPR, 2024

  41. [41]

    Wan: Open and Advanced Large-Scale Video Generative Models

    Wan, T., Wang, A., Ai, B., Wen, B., Mao, C., Xie, C.-W., Chen, D., Yu, F., Zhao, H., Yang, J., et al. Wan: Open and advanced large-scale video generative models. In arXiv preprint arXiv:2503.20314, 2025

  42. [42]

    Fudoki: Discrete flow-based unified understanding and generation via kinetic-optimal velocities

    Wang, J., Lai, Y., Li, A., Zhang, S., Sun, J., Kang, N., Wu, C., Li, Z., and Luo, P. Fudoki: Discrete flow-based unified understanding and generation via kinetic-optimal velocities. In arXiv preprint arXiv:2505.20147, 2025 a

  43. [43]

    SimpleAR: Pushing the Frontier of Autoregressive Visual Generation through Pretraining, SFT, and RL

    Wang, J., Tian, Z., Wang, X., Zhang, X., Huang, W., Wu, Z., and Jiang, Y.-G. Simplear: Pushing the frontier of autoregressive visual generation through pretraining, sft, and rl. In arXiv preprint arXiv:2504.11455, 2025 b

  44. [44]

    Emu3: Next-Token Prediction is All You Need

    Wang, X., Zhang, X., Luo, Z., Sun, Q., Cui, Y., Wang, J., Zhang, F., Wang, Y., Li, Z., Yu, Q., et al. Emu3: Next-token prediction is all you need. In arXiv preprint arXiv:2409.18869, 2024

  45. [45]

    SANA 1.5: Efficient Scaling of Training-Time and Inference-Time Compute in Linear Diffusion Transformer

    Xie, E., Chen, J., Zhao, Y., Yu, J., Zhu, L., Wu, C., Lin, Y., Zhang, Z., Li, M., Chen, J., et al. Sana 1.5: Efficient scaling of training-time and inference-time compute in linear diffusion transformer. In arXiv preprint arXiv:2501.18427, 2025

  46. [46]

    Show-o: One Single Transformer to Unify Multimodal Understanding and Generation

    Xie, J., Mao, W., Bai, Z., Zhang, D. J., Wang, W., Lin, K. Q., Gu, Y., Chen, Z., Yang, Z., and Shou, M. Z. Show-o: One single transformer to unify multimodal understanding and generation. In arXiv preprint arXiv:2408.12528, 2024

  47. [47]

    DanceGRPO: Unleashing GRPO on Visual Generation

    Xue, Z., Wu, J., Gao, Y., Kong, F., Zhu, L., Chen, M., Liu, Z., Liu, W., Guo, Q., Huang, W., et al. Dancegrpo: Unleashing grpo on visual generation. In arXiv preprint arXiv:2505.07818, 2025

  48. [48]

    GPT-ImgEval: A Comprehensive Benchmark for Diagnosing GPT4o in Image Generation

    Yan, Z., Ye, J., Li, W., Huang, Z., Yuan, S., He, X., Lin, K., He, J., He, C., and Yuan, L. Gpt-imgeval: A comprehensive benchmark for diagnosing gpt4o in image generation. In arXiv preprint arXiv:2504.02782, 2025

  49. [49]

    A Dense Reward View on Aligning Text-to-Image Diffusion with Preference

    Yang, S., Chen, T., and Zhou, M. A dense reward view on aligning text-to-image diffusion with preference. In arXiv preprint arXiv:2402.08265, 2024

  50. [50]

    G., Yang, M.-H., Hao, Y., Essa, I., et al

    Yu, L., Cheng, Y., Sohn, K., Lezama, J., Zhang, H., Chang, H., Hauptmann, A. G., Yang, M.-H., Hao, Y., Essa, I., et al. Magvit: Masked generative video transformer. In CVPR, 2023