REVIEW 4 major objections 5 minor 30 references
The learned partition function in GFlowNet-style RL can be replaced by a Monte Carlo estimate from the existing rollout batch, preserving the reward-proportional fixed point while stabilizing training.
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 · deepseek-v4-flash
2026-08-02 05:18 UTC pith:E7GJIDC2
load-bearing objection A serious, mostly well-executed scaling recipe for GFlowNet-style RL, but the paper's own theory does not cover the length-normalized loss actually trained, so the distribution-matching claim is weaker than it looks. the 4 major comments →
GFlowRL: Scaling Distribution-Matching RL to Large Language Models
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
Core claim
The central claim is that log Z(x), the per-prompt normalizer that trajectory-balance objectives learn as an auxiliary network, can be replaced by Zt(x): the mean over a prompt's rollout group of β r(x,y) + log π_ref(y|x) − log π_old(y|x), inserted as a stop-gradient baseline. With this substitution, a zero-loss self-consistent fixed point of the GFlowRL loss satisfies π_θ(y|x) ∝ π_ref(y|x) exp(β r(x,y)), the same target as trajectory balance, so no distribution-matching fidelity is lost. Two stabilizers—importance-sampling weights that correct rollout/trainer drift and asymmetric clipping of the flow gap—keep outlier rollouts from destabilizing updates. The paper reports that this design tr
What carries the argument
The load-bearing object is the in-batch Monte Carlo estimator Zt(x), which replaces the learned partition network. It is the average over the G rollouts of βr + log π_ref − log π_old; because it enters the loss only through a stop-gradient operator, it acts as a centering baseline with no parameters and no gradient. The other machinery is the flow-gap clip: the per-rollout residual is split into a flow gap evaluated at the rollout policy plus a policy-update term, and the gap is clipped asymmetrically so positive corrections have more room than negative ones.
Load-bearing premise
The fixed-point proof is stated for the loss without length normalization and with clipping inactive, while the actual loss divides every term by response length; the load-bearing premise is that response lengths within a rollout group are nearly equal, otherwise the policy converges to a length-weighted distribution different from the intended reward-proportional target.
What would settle it
Run a trained GFlowRL model on prompts with verifiable rewards and compare the empirical response distribution to π_ref(y|x) exp(β r(x,y))/Z(x). If long responses are sampled more often than that target predicts, and the discrepancy tracks exp(|y| β r), the length-normalized loss is not reproducing the claimed fixed point. A simpler check: compute within-group response-length variance on the training data; the theory predicts the distortion grows with it.
If this is right
- GFlowNet-style RL can run on the same infrastructure as standard group-based policy optimization; the auxiliary partition network and its optimizer state disappear.
- At convergence the trained policy should sample responses with probability proportional to the reference-policy likelihood times exp(β r), so multiple high-reward reasoning paths are retained instead of collapsing to one mode.
- Gradient norms during training stay at the same scale as reward-maximizing baselines, so the divergence pattern seen in prior distribution-matching methods is avoidable.
- The recipe transfers unchanged from dense models to MoE models up to 235B parameters, opening distribution-matching RL to sparse architectures.
Where Pith is reading between the lines
- Not tested in the paper: if the learned partition term is as useless as the random-replacement diagnostic suggests, then a constant or group-mean baseline might capture most of the method's benefit; an ablation that replaces Zt(x) with the group mean of rewards would separate the estimator's centering effect from its distribution-matching effect.
- The proof's dependence on equal response lengths makes a concrete prediction: on tasks with high within-group length variance, the trained policy should be length-tilted (weighted by exp(|y| β r)); measuring this tilt would tell whether the length-normalized loss is actually matching the claimed target.
- The estimator's variance is O(1/G), so reducing the group size G should degrade stability and accuracy; this is a testable scaling law for deployment in low-throughput settings.
- The same estimator-centric design could generalize to agentic or multimodal RL, where rollouts are more heterogeneous and rewards noisier, but the paper does not claim that and leaves it open.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper proposes GFlowRL, a GFlowNet-style RL algorithm for LLM post-training that replaces the learned prompt-conditional partition function Z_phi with an in-batch Monte Carlo estimate Z_t computed from the rollout group already sampled for policy gradient training. The algorithm adds two stabilizers, an importance-sampling weight and an asymmetric flow-gap clip, and is claimed to preserve the reward-distribution-matching objective of trajectory balance while removing the auxiliary partition network and its optimizer/synchronization overhead. Empirical results are reported for dense models (Qwen2.5-7B/32B, DeepSeek-R1-Distill-Qwen-7B/14B) on math, code, and adversarial red-teaming benchmarks, and for MoE models up to Qwen3-235B-A22B, where the paper reports stable training and gains over GRPO/PPO/FlowRL, including a Codeforces rating of 2048 at 14B.
Significance. If the central claim holds, this is a practically significant contribution: it would be the first GFlowNet-style RL method to scale stably to large MoE models, while simplifying the training infrastructure to be GRPO-compatible. The paper has real strengths: a clear diagnostic for the learned partition function, gradient-norm statistics, a synthetic distribution-matching benchmark, extensive benchmark tables at multiple scales, and qualitative Codeforces examples. However, the theoretical guarantee in Proposition B.1 is proved for a population, unnormalized, unclipped loss, whereas the deployed loss in Eq. (8) is finite-batch, length-normalized, and clipped; Remark B.4 concedes a length-dependent fixed-point distortion. Because the distribution-matching claim is the paper's central narrative, this gap must be addressed before the result can be regarded as supported.
major comments (4)
- [Appendix B, Prop. B.1 / Remark B.4, and Eq. (8)] The central claim that GFlowRL 'preserves the reward-distribution-matching objective' is proved only for the population, unnormalized, unclipped loss in Eq. (9), while every experiment uses Eq. (8), which divides the log-ratio terms by |y|. Remark B.4 explicitly derives the resulting fixed point as pi_theta(y|x) proportional to pi_ref(y|x) exp(|y|(beta r(x,y) - Z_t(x))), which equals the claimed p*(y|x) only when all responses in a rollout group have equal length. The paper asserts without measurement that 'response lengths vary only modestly within rollout groups on reasoning benchmarks.' This is load-bearing: if within-group length variation is non-negligible, the trained policy optimizes a length-weighted distribution, not the softmax target. Please report within-group length statistics for the actual benchmark groups, quantify the induced fixed-point distortion, or modify the estimat
- [Sec. 3.1 and Table 15] The diagnostic claim that 'replacing Z_phi with random noise yields comparable performance' is contradicted by the paper's own 32B results. In Table 2 (7B), FlowRL-RandomLogZ averages 36.19 vs. FlowRL 35.63, supporting the claim; but in Table 15 (32B), FlowRL-RandomLogZ averages 44.73 vs. FlowRL 48.39, with large drops on AIME25, Minerva, and Olympiad. This inconsistency weakens the motivation that the learned partition function 'contributes little.' Please reconcile the two scales or temper the claim.
- [Sec. 4.3 and Table 13] The text states that for Qwen3-235B-A22B, GFlowRL 'is trained for only 30 steps vs. 100 for GRPO,' but Table 13 lists a training schedule '[0,100]' for that model and for Qwen3-30B-A3B. Please clarify the actual number of update steps and the schedule range. If the 235B run is only 30 steps, the claim that 'the recipe transfers' rests on a very short run; provide training curves or early-plateau evidence to support the scaling conclusion.
- [Appendix B, Prop. B.1 proof] The proof of Proposition B.1 imposes normalization over all y in Step 2, but the zero-loss condition is only established for y in the support of pi_theta*. If the target p* has support outside the current policy's support, equality on the support does not imply equality as distributions over Y. Please state the support assumption explicitly (e.g., full support of the autoregressive policy) or revise the proof to show that the fixed point is p* on the relevant support.
minor comments (5)
- [Eqs. (5) and (6)] Notation is inconsistent: Eq. (5) uses pi_theta_ref while Eq. (6) uses pi_ref for the same reference policy. Please unify.
- [Sec. 3.1 and Table 2] The text reports the learned partition function result as 35.61, while Table 2 lists FlowRL as 35.63. Also, the claim that random replacement 'slightly improves' should be qualified by the 32B results in Table 15, where random replacement is worse.
- [Table 9] The hyperparameter table lists 'Clip ratio epsilon_low 0.2, clip ratio epsilon_high 0.28' alongside an 'IS threshold 2.0.' It is unclear whether the clip ratios refer to the flow-gap clip in Eq. (7) or to a policy clip ratio. Please clarify the naming and distinguish the two clipping mechanisms.
- [Appendix I] The qualitative comparison states 'All evaluations use Pass@16 with temperature 0.6 and top-p=0.95,' but the main code tables report both Pass@16 and Avg@16. Please specify which sampling parameters apply to which table and metric.
- [References] The reference list contains a typo: 'HugginfaceH4' should be 'HuggingFaceH4.'
Circularity Check
No significant circularity: GFlowRL's fixed-point claim is derived from a stated objective, and the main empirical results are external and falsifiable.
full rationale
The derivation chain is self-contained. Eq. (4) defines Z_t(x) as an in-batch mean of per-sample trajectory-balance targets; Proposition B.1 then derives the fixed point of the population loss (Eq. 9) from the zero-residual condition, and Step 3 verifies that the definition of Z_t is consistent with log Z. This is a fixed-point characterization of a stated objective, not a fitted parameter relabeled as a prediction; the estimator is not tuned to any benchmark. The empirical claims (Tables 2–5) are tested against external math, code, and safety benchmarks, and against FlowRL/GRPO/PPO baselines, so they are externally falsifiable. The chief self-citations (FlowRL, SEMA) are prior work used as baselines; the load-bearing instability claim is supported by in-paper diagnostics (random-logZ ablation, gradient norm statistics), not by those citations alone. The main acknowledged gap — Remark B.4's unmeasured assertion that response lengths vary only modestly within rollout groups — is a limitation of the theoretical claim (the actual Eq. 8 length-normalizes and therefore has a length-weighted fixed point unless lengths are equal), but this is a correctness/assumption issue, not circularity. No circular step is identified.
Axiom & Free-Parameter Ledger
free parameters (4)
- beta (inverse temperature) =
8
- flow-gap clip bounds (eps_low, eps_high) =
0.2, 0.28
- importance-sampling cap (IS threshold 2.0; RS thresholds 1.01/0.99) =
2.0; 1.01/0.99
- length-normalization divisor 1/|y|
axioms (5)
- standard math Trajectory-balance optimum characterization (Malkin et al. 2022)
- domain assumption Policy class expressivity: p*(y|x) proportional to pi_ref(y|x) exp(beta r(x,y)) is representable
- domain assumption Response lengths vary only modestly within rollout groups on reasoning benchmarks
- ad hoc to paper Flow-gap clip is inactive in a neighborhood of the optimum
- domain assumption Reward verifier r(x,y) is a faithful oracle
read the original abstract
Generative Flow Networks (GFlowNets) offer a promising alternative to reward-maximizing reinforcement learning (RL) for large reasoning models, encouraging diverse reasoning paths by matching reward distributions rather than collapsing to dominant modes. Recent work shows promise on math and code, but scaling GFlowNet-style RL to modern post-training pipelines remains difficult: as model size, rollout horizon, reward noise, and distributed-systems complexity grow together, a learned prompt-conditional partition function becomes a source of gradient instability and engineering overhead rather than a useful normalizer. Through systematic analysis, we find that the learned partition function, previously treated as essential, can be replaced by an in-batch Monte Carlo estimate computed from the rollout group already required for training. We propose GFlowRL, a streamlined GFlowNet-style RL algorithm that removes the auxiliary partition network entirely while preserving the reward-distribution-matching objective, completed by two stabilizers: importance-sampling correction for rollout/trainer drift and asymmetric flow-gap clipping for outlier residuals. GFlowRL exceeds all counterparts on math, code, and adversarial red-teaming benchmarks, reaching a Codeforces rating of 2048 at the 14B scale (within 25 Elo of o3-mini) and attaining the highest average ASR@1 on AdvBench and HarmBench, outperforming the previous SOTA multi-turn attacker in a regime where FlowRL, a prior GFlowNet-style method, diverges. The same recipe transfers to all evaluated MoE configurations up to 235B parameters, where FlowRL again fails to converge. To our knowledge, GFlowRL is the first GFlowNet-style RL algorithm to scale stably across both dense and sparse architectures. Code will be at: https://github.com/microsoft/gflowrl
Figures
Reference graph
Works this paper leans on
-
[1]
Josh Achiam, Steven Adler, Sandhini Agarwal, Lama Ahmad, Ilge Akkaya, Florencia Leoni Aleman, Diogo Almeida, Janko Altenschmidt, Sam Altman, Shyamal Anadkat, et al. Gpt-4 technical report. arXiv preprint arXiv:2303.08774,
-
[4]
Jian Hu, Jason Klein Liu, Haotian Xu, and Wei Shen
URL https://openreview.net /forum?id=Ouj6p4ca60. Jian Hu, Jason Klein Liu, Haotian Xu, and Wei Shen. Reinforce++: Stabilizing critic-free policy optimization with global advantage normalization.arXiv preprint arXiv:2501.03262,
-
[5]
Math-500
HugginfaceH4. Math-500. URL https://huggingface.co/datasets/HuggingFaceH4/MATH -500. Naman Jain, Han, Alex Gu, Wen-Ding Li, Fanjia Yan, Tianjun Zhang, Sida Wang, Armando Solar- Lezama, Koushik Sen, and Ion Stoica. Livecodebench: Holistic and contamination free evaluation of large language models for code. In Y . Yue, A. Garg, N. Peng, F. Sha, and R. Yu, e...
2025
-
[6]
Koray Kavukcuoglu
URL https://proceedings.iclr.cc/paper_files/paper/2025/file/94074dd5a072 d28ff75a76dabed43767-Paper-Conference.pdf. Koray Kavukcuoglu. Gemini 2.5: Our most intelligent AI model,
2025
-
[7]
Google Blog (The Keyword), Published Mar
URL https://blog.goo gle/technology/google-deepmind/gemini-model-thinking-updates-march-2025/ . Google Blog (The Keyword), Published Mar. 25,
2025
-
[8]
Kimi k1.5: Scaling reinforcement learning with llms
Kimi-Team, Angang Du, Bofei Gao, Bowei Xing, Changjiu Jiang, Cheng Chen, Cheng Li, Chenjun Xiao, Chenzhuang Du, Chonghua Liao, et al. Kimi k1.5: Scaling reinforcement learning with llms. arXiv preprint arXiv:2501.12599,
-
[11]
Automated red teaming with GOAT: the generative offensive agent tester
Maya Pavlova, Erik Brinkman, Krithika Iyer, Vítor Albiero, Joanna Bitton, Hailey Nguyen, Cris- tian Canton Ferrer, Ivan Evtimov, and Aaron Grattafiori. Automated red teaming with GOAT: the generative offensive agent tester. InICLR 2025 Workshop on Building Trust in Language Models and Applications,
2025
-
[12]
URLhttps://aclanthology.org/2025.acl-long.1207/
Association for Computational Linguistics. URLhttps://aclanthology.org/2025.acl-long.1207/. Lorenz Richter, Ayman Boustati, Nikolas Nüsken, Francisco Ruiz, and Omer Deniz Akyildiz. Vargrad: A low-variance gradient estimator for variational inference. In H. Larochelle, M. Ranzato, R. Hadsell, M.F. Balcan, and H. Lin, editors,Advances in Neural Information ...
2025
-
[13]
D.6 Diagnostic and Synthetic Setup Random-logZ replacement.For the diagnostic in Sec. 4.2, we replace logZ ϕ(x) at every training step with an independent sample logZ∼ N(µ, σ2), where (µ, σ2) are matched to the empirical distribution of logZ ϕ(x) observed under standard FlowRL training [Zhu et al., 2026]. We use µ= 0.5 and σ= 1 . All other settings (optim...
2026
-
[14]
John Schulman, Filip Wolski, Prafulla Dhariwal, Alec Radford, and Oleg Klimov
USENIX Association. John Schulman, Filip Wolski, Prafulla Dhariwal, Alec Radford, and Oleg Klimov. Proximal policy optimization algorithms.arXiv preprint arXiv:1707.06347,
-
[16]
URLhttps://arxiv.org/abs/2410.18469. Thinking Machines. Defeating nondeterminism in llm inference. https://thinkingmachines.a i/blog/defeating-nondeterminism-in-llm-inference/,
-
[17]
URL https://proceedings.neurips.cc/paper_files/paper/2017/file/ebd6d2f5d60ff 9afaeda1a81fc53e2d0-Paper.pdf. Siddarth Venkatraman, Minsu Kim, Luke Rowe, Moksh Jain, Marcin Sendera, Sarthak Mittal, Luca Scimeca, Mohsin Hasan, Pablo Lemos, Emmanuel Bengio, Alexandre Adam, Jarrid Rector-Brooks, Yoshua Bengio, Glen Berseth, and Nikolay Malkin. Amortizing intra...
2017
-
[19]
Ronald J Williams
URL https://aclanthology.org/2025.emnlp-main.100/. Ronald J Williams. Simple statistical gradient-following algorithms for connectionist reinforcement learning.Machine Learning, 8(3):229–256,
2025
-
[20]
13 Hao Yang, Lizhen Qu, Ehsan Shareghi, and Gholamreza Haffari
URLhttps://www.pnas.org/doi/abs/10.1073/pnas.2504966122. 13 Hao Yang, Lizhen Qu, Ehsan Shareghi, and Gholamreza Haffari. Jigsaw puzzles: Splitting harmful questions to jailbreak large language models, 2024a. URL https://arxiv.org/abs/2410.1
-
[22]
Fangxu Yu, Lai Jiang, Haoqiang Kang, Shibo Hao, and Lianhui Qin
Work in progress. Fangxu Yu, Lai Jiang, Haoqiang Kang, Shibo Hao, and Lianhui Qin. Flow of reasoning: Training LLMs for divergent reasoning with minimal examples. InForty-second International Conference on Machine Learning, 2025a. URLhttps://openreview.net/forum?id=qyMxunrR2j. Qiying Yu, Zheng Zhang, Ruofei Zhu, Yufeng Yuan, Xiaochen Zuo, Yu Yue, Weinan D...
2025
-
[23]
mlr.press/v162/zhang22v.html
URL https://proceedings. mlr.press/v162/zhang22v.html. Yifan Zhang and Team Math-AI. Minerva math. URL https://huggingface.co/datasets/ma th-ai/minervamath. Yifan Zhang and Team Math-AI. American mathematics competitions (amc) 2023,
2023
-
[25]
American invitational mathematics examination (aime) 2025,
Yifan Zhang and Team Math-AI. American invitational mathematics examination (aime) 2025,
2025
-
[26]
14 A Limitations and Broader Impacts GFlowRL’s in-batch Monte Carlo estimate of logZ can have higher variance in principle, especially when the group size is small. In practice, our two stabilizers, importance-sampling correction and flow-gap clipping, help control this variance, and the resulting streamlined estimator consistently outperforms the learned...
2022
-
[27]
17 Table 8:Dataset statistics.Training and evaluation datasets used in our experiments, spanning mathematical reasoning, code generation, and AI security. Domain Split Dataset # Examples License Math Train DAPO-Math-17K [BytedTsinghua-SIA, 2025] 17,398 Apache 2.0 Test AIME 2024 [Zhang and Math-AI, 2024] 30 Apache 2.0 AIME 2025 [Zhang and Math-AI, 2025] 30...
2025
-
[29]
Instruct
with Qwen2.5-7B as the backbone. Method Backbone ConstantLogZ FOR GFlowRL Average Acc 23.02 25.34 29.53 40.92 F.2 Dense 32B Math Results Table 15:Results on math reasoning benchmarks for dense models.We report Avg@16 accuracy with relative improvements over the backbone shown as subscripts. Positive gains are shown in green and negative changes in red. GF...
2023
-
[30]
to Maximum Diversity (score = 5), asEVALUATION CRITERIA inthe prompt. 22 H Distribution Matching Comparison To evaluate distribution matching performance, we construct a synthetic target distribution defined as a mixture of three Gaussian components with distinct centers. We compare four algorithms: GRPO (a maximum-reward method), FlowRL, FlowRL-RandomLog...
2025
-
[1459]
Chain of attack: a semantic-driven contextual multi-turn attacker for llm, 2024b
Xikang Yang, Xuehai Tang, Songlin Hu, and Jizhong Han. Chain of attack: a semantic-driven contextual multi-turn attacker for llm, 2024b. URLhttps://arxiv.org/abs/2405.05610. Feng Yao, Liyuan Liu, Dinghuai Zhang, Chengyu Dong, Jingbo Shang, and Jianfeng Gao. Your efficient rl framework secretly brings you off-policy rl training. https://fengyao.notion.sit ...
-
[2001]
Foot-in-the-door: A multi-turn jailbreak for LLMs
Zixuan Weng, Xiaolong Jin, Jinyuan Jia, and Xiangyu Zhang. Foot-in-the-door: A multi-turn jailbreak for LLMs. In Christos Christodoulopoulos, Tanmoy Chakraborty, Carolyn Rose, and Violet Peng, editors,Proceedings of the 2025 Conference on Empirical Methods in Natural Language Processing, pages 1939–1950. Association for Computational Linguistics,
2025
-
[2017]
Zhihong Shao, Peiyi Wang, Qihao Zhu, Runxin Xu, Junxiao Song, Xiao Bi, Haowei Zhang, Mingchuan Zhang, YK Li, Y Wu, et al. Deepseekmath: Pushing the limits of mathematical reasoning in open language models.arXiv preprint arXiv:2402.03300,
-
[2020]
12 Mark Russinovich, Ahmed Salem, and Ronen Eldan
URL https://proceedings.neurips.cc/paper_fil es/paper/2020/file/9c22c0b51b3202246463e986c7e205df-Paper.pdf. 12 Mark Russinovich, Ahmed Salem, and Ronen Eldan. Great, now write an article about that: the crescendo multi-turn llm jailbreak attack. InProceedings of the 34th USENIX Conference on Security Symposium, SEC ’25, USA,
2020
-
[2021]
URL https://doi.org/10.1007/s4 2985-021-00102-x. OpenAI. Introducing openai o1. https://openai.com/o1/ ,
-
[2022]
URL https://proceedings.neurips.cc/paper _files/paper/2022/file/27b51baca8377a0cf109f6ecc15a0f70-Paper-Conference. pdf. 11 Nikolay Malkin, Salem Lahlou, Tristan Deleu, Xu Ji, Edward J Hu, Katie E Everett, Dinghuai Zhang, and Yoshua Bengio. GFlownets and variational inference. InThe Eleventh International Conference on Learning Representations,
2022
-
[2023]
Yifan Zhang and Team Math-AI
URL https://huggingface.co/datasets/math-ai/amc23. Yifan Zhang and Team Math-AI. American invitational mathematics examination (aime) 2024,
2024
-
[2024]
URL https://arxiv.org/abs/2407.21783. Daya Guo, Dejian Yang, Haowei Zhang, Junxiao Song, Ruoyu Zhang, Runxin Xu, Qihao Zhu, Shirong Ma, Peiyi Wang, Xiao Bi, et al. Deepseek-r1: Incentivizing reasoning capability in llms via reinforcement learning.Nature, 645:633–638, 2025a. Weiyang Guo, Zesheng Shi, Zhuo Li, Yequan Wang, Xuebo Liu, Wenya Wang, Fangming Li...
-
[2025]
URL https://huggingface.co/datasets/Byte dTsinghua-SIA/DAPO-Math-17k. Mark Chen, Jerry Tworek, Heewoo Jun, Qiming Yuan, Henrique Ponde de Oliveira Pinto, Jared Kaplan, Harri Edwards, Yuri Burda, Nicholas Joseph, Greg Brockman, Alex Ray, Raul Puri, Gretchen Krueger, Michael Petrov, Heidy Khlaaf, Girish Sastry, Pamela Mishkin, Brooke Chan, Scott Gray, Nick ...
discussion (0)
Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.