pith. sign in

arxiv: 2606.18195 · v2 · pith:YQJ4BCEZnew · submitted 2026-06-16 · 💻 cs.CL

Learning from the Self-future: On-policy Self-distillation for dLLMs

Pith reviewed 2026-06-27 01:17 UTC · model grok-4.3

classification 💻 cs.CL
keywords on-policy self-distillationdiffusion LLMsdLLMsreasoning benchmarkspost-trainingstep-level supervisionsuffix conditioning
0
0 comments X

The pith

d-OPSD lets diffusion LLMs distill from their own future generations via suffix conditioning and step-level supervision.

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

The paper establishes that standard on-policy self-distillation clashes with diffusion LLMs because it relies on left-to-right prefix conditioning and token-level losses. d-OPSD instead builds the self-teacher from the model's own generated answers used as suffix conditioning and switches supervision to the step level. This change aligns the objective with the iterative denoising process that defines dLLMs. The result is consistent gains over RLVR and SFT on four reasoning benchmarks while using roughly one-tenth the optimization steps of RLVR.

Core claim

By reframing self-teacher construction around self-generated answers as suffix conditioning and moving supervision from token level to step level, d-OPSD produces an on-policy self-distillation procedure that matches the arbitrary-order, iterative nature of dLLMs and delivers higher reasoning performance than RLVR or SFT baselines with far fewer training steps.

What carries the argument

Suffix conditioning drawn from the model's own self-generated answers, combined with step-level supervision that matches the iterative denoising schedule.

If this is right

  • d-OPSD outperforms both RLVR and SFT baselines across four reasoning benchmarks.
  • The method reaches those results with only around 10 percent of the optimization steps needed by RLVR.
  • It supplies a concrete route for efficient post-training of diffusion language models.

Where Pith is reading between the lines

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

  • The same suffix-plus-step-level pattern may transfer to other iterative non-autoregressive generators.
  • Lower step counts could make post-training feasible on larger dLLM sizes under fixed compute budgets.
  • Combining d-OPSD with existing RLVR schedules remains an open direction the paper leaves unexplored.

Load-bearing premise

That suffix conditioning from self-generated answers plus step-level supervision will align self-distillation with dLLM denoising without creating new training conflicts.

What would settle it

Running the same four reasoning benchmarks and finding that d-OPSD requires as many or more optimization steps as RLVR or fails to exceed the RLVR and SFT scores would falsify the central claim.

Figures

Figures reproduced from arXiv: 2606.18195 by Haoyu Wang, Shiwei Liu, Xinhao Hu, Yifu Luo, Yuxuan Zhang, Zeyu Chen, Zhizhou Sha.

Figure 1
Figure 1. Figure 1: The reasoning performance and sample efficiency [PITH_FULL_IMAGE:figures/full_fig_p001_1.png] view at source ↗
Figure 2
Figure 2. Figure 2: The framework of our approach, d-OPSD. It leverages self-generated answers as suffix [PITH_FULL_IMAGE:figures/full_fig_p003_2.png] view at source ↗
Figure 3
Figure 3. Figure 3: The Overlap Top-K compari￾son between d-OPSD and the AR-style counterpart. We further investigate the mechanism behind this perfor￾mance gap. We define the metric of Overlap Top-Kt. At each denoising step t, it measures the proportion of to￾kens that appear simultaneously in both the student’s and teacher’s Top-K vocabulary distributions over the top-k subset Kt masked positions. Note that Top-K and top-k … view at source ↗
Figure 4
Figure 4. Figure 4: A question from GSM8K training set. First, we sample an on-policy trajectory 5 from the student model and obtain the final clean answer as the self-generated future [PITH_FULL_IMAGE:figures/full_fig_p014_4.png] view at source ↗
Figure 5
Figure 5. Figure 5: The self-generated future answer. 5Using pass@k, it keeps sampling until a correct final answer appears or it reaches the iteration threshold. 14 [PITH_FULL_IMAGE:figures/full_fig_p014_5.png] view at source ↗
Figure 6
Figure 6. Figure 6: Current student decoding status. We then construct the self-teacher at step t = 20 as follows [PITH_FULL_IMAGE:figures/full_fig_p015_6.png] view at source ↗
Figure 7
Figure 7. Figure 7: Self-teacher construction at t = 20. For comparison, we also illustrate the AR-style construction, which appends a reference solution to the prompt, as shown in [PITH_FULL_IMAGE:figures/full_fig_p015_7.png] view at source ↗
Figure 8
Figure 8. Figure 8: AR-style teacher construction. all status tensors across all steps of this trajectory to form a “batch” tensor of shape (bsz×steps, seq-length). Since all inputs share the same model, the gradient remains constant for each input and no longer needs to be stored as previously. C.3 Compute only on Correct Generations By default, we compute the loss objective Equation (12) only on correct generations 6 . Alth… view at source ↗
Figure 9
Figure 9. Figure 9: A question from GSM8K training set. First, we sample a generation 7 from the student model and obtain the final clean answer [PITH_FULL_IMAGE:figures/full_fig_p017_9.png] view at source ↗
Figure 10
Figure 10. Figure 10: The self-generated answer. We then construct self-teacher by partially revealing the final generation, as shown in [PITH_FULL_IMAGE:figures/full_fig_p017_10.png] view at source ↗
Figure 11
Figure 11. Figure 11: Self-teacher in the toy experiment [PITH_FULL_IMAGE:figures/full_fig_p018_11.png] view at source ↗
Figure 12
Figure 12. Figure 12: presents the failure mode mentioned in Section 4.5 [PITH_FULL_IMAGE:figures/full_fig_p018_12.png] view at source ↗
Figure 13
Figure 13. Figure 13: Qualitative Examples on GSM8k 19 [PITH_FULL_IMAGE:figures/full_fig_p019_13.png] view at source ↗
read the original abstract

On-policy self-distillation (OPSD) has proven effective for post-training large language models (LLMs), yet its application to diffusion LLMs (dLLMs) remains unexplored. Existing OPSD methods are inherently autoregressive-centric. They inject privileged information via left-to-right prefix conditioning with token-level divergence supervision, a design that fundamentally conflicts with the arbitraryorder generation of dLLMs. We introduce d-OPSD, the first OPSD framework tailored for dLLMs. Our approach makes two core contributions. First, we reframe self-teacher construction by using self-generated answers as suffix conditioning, enabling the student model to learn from "self future-experience" rather than privileged prefixes. Second, we shift supervision from token-level to step-level, aligning training with the iterative denoising process of dLLMs. Experiments across four reasoning benchmarks show that d-OPSD consistently outperforms RLVR and SFT baselines with superior sample efficiency, requiring only around 10% of the optimization steps by RLVR and opening a promising pathway for dLLM posttraining. The code is available at https://github.com/xingzhejun/d-OPSD.

Editorial analysis

A structured set of objections, weighed in public.

Desk editor's note, referee report, simulated authors' rebuttal, and a circularity audit. Tearing a paper down is the easy half of reading it; the pith above is the substance, this is the friction.

Referee Report

2 major / 0 minor

Summary. The paper introduces d-OPSD as the first on-policy self-distillation framework for diffusion LLMs (dLLMs). It reframes self-teacher construction to use self-generated answers as suffix conditioning (learning from 'self-future') instead of left-to-right prefixes, and shifts from token-level to step-level supervision to align with dLLM iterative denoising. Experiments on four reasoning benchmarks claim consistent outperformance over RLVR and SFT baselines with superior sample efficiency (approximately 10% of RLVR optimization steps). Code is released at the cited GitHub repository.

Significance. If the empirical claims hold after verification, the work fills a gap in adapting OPSD to non-autoregressive dLLMs and provides a concrete pathway for their post-training. The public code release is a clear strength that enables direct reproducibility and follow-up work.

major comments (2)
  1. [Abstract] Abstract: The central claim that d-OPSD 'consistently outperforms RLVR and SFT baselines with superior sample efficiency, requiring only around 10% of the optimization steps by RLVR' is load-bearing yet unsupported by any reported metrics, benchmark names, variance estimates, or ablation results in the visible text. Without these, it is impossible to assess whether the efficiency gain is robust or an artifact of particular runs.
  2. [Abstract] Abstract (method description): The assertion that suffix conditioning plus step-level supervision 'aligns training with the iterative denoising process of dLLMs' without new conflicts is not accompanied by any derivation showing that (a) suffix masking commutes with the diffusion noise schedule or (b) the resulting step-level KL objective produces gradients compatible with arbitrary-order reverse processes. This alignment is required for the on-policy property to hold and for the reported gains to generalize beyond the tested dLLM architecture.

Simulated Author's Rebuttal

2 responses · 0 unresolved

We thank the referee for highlighting the need for greater self-containment in the abstract and for formal justification of the alignment claims. We address both points below and will revise the manuscript accordingly.

read point-by-point responses
  1. Referee: [Abstract] Abstract: The central claim that d-OPSD 'consistently outperforms RLVR and SFT baselines with superior sample efficiency, requiring only around 10% of the optimization steps by RLVR' is load-bearing yet unsupported by any reported metrics, benchmark names, variance estimates, or ablation results in the visible text. Without these, it is impossible to assess whether the efficiency gain is robust or an artifact of particular runs.

    Authors: We agree the abstract is too terse. The full manuscript (Sections 4–5) reports results on four benchmarks (GSM8K, MATH, HumanEval, MBPP) with concrete metrics, standard deviations across runs, and ablations comparing optimization steps. We will revise the abstract to name the benchmarks, include key quantitative gains with variance, and reference the sample-efficiency comparison explicitly. revision: yes

  2. Referee: [Abstract] Abstract (method description): The assertion that suffix conditioning plus step-level supervision 'aligns training with the iterative denoising process of dLLMs' without new conflicts is not accompanied by any derivation showing that (a) suffix masking commutes with the diffusion noise schedule or (b) the resulting step-level KL objective produces gradients compatible with arbitrary-order reverse processes. This alignment is required for the on-policy property to hold and for the reported gains to generalize beyond the tested dLLM architecture.

    Authors: Section 3 motivates the design by showing that suffix conditioning preserves the arbitrary-order property of dLLMs and that step-level supervision matches the denoising trajectory, avoiding the prefix conflicts of autoregressive OPSD. We acknowledge the absence of an explicit commutativity derivation or gradient-compatibility proof. We will add a short appendix providing a derivation sketch based on the diffusion forward process and the step-wise KL objective. revision: partial

Circularity Check

0 steps flagged

No circularity: empirical method contribution with no derivation chain

full rationale

The paper presents d-OPSD as a new empirical framework for on-policy self-distillation on diffusion LLMs, with two described changes (suffix conditioning from self-generated answers and step-level supervision) justified by alignment to dLLM iterative denoising. No equations, derivations, fitted parameters renamed as predictions, or self-citation load-bearing steps appear in the provided text. The central claims rest on benchmark experiments rather than any reduction of outputs to inputs by construction. This is a standard non-circular empirical contribution.

Axiom & Free-Parameter Ledger

0 free parameters · 0 axioms · 0 invented entities

No free parameters, axioms, or invented entities are identifiable from the abstract alone; the contribution is framed as an empirical adaptation rather than a theoretical derivation.

pith-pipeline@v0.9.1-grok · 5752 in / 1024 out tokens · 32823 ms · 2026-06-27T01:17:21.615568+00:00 · methodology

discussion (0)

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

Reference graph

Works this paper leans on

41 extracted references · 1 canonical work pages · 1 internal anchor

  1. [1]

    d1: Scaling reasoning in diffusion large language models via reinforcement learning.arXiv preprint arXiv:2504.12216,

    Siyan Zhao, Devaansh Gupta, Qinqing Zheng, and Aditya Grover. d1: Scaling reasoning in diffusion large language models via reinforcement learning.arXiv preprint arXiv:2504.12216,

  2. [2]

    Qwen3 technical report.arXiv preprint arXiv:2505.09388,

    An Yang, Anfeng Li, Baosong Yang, Beichen Zhang, Binyuan Hui, Bo Zheng, Bowen Yu, Chang Gao, Chengen Huang, Chenxu Lv, et al. Qwen3 technical report.arXiv preprint arXiv:2505.09388,

  3. [3]

    Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipe

    doi: 10.64434/tml.20251026. https://thinkingmachines.ai/blog/on-policy-distillation. Yaxuan Li, Yuxin Zuo, Bingxiang He, Jinqian Zhang, Chaojun Xiao, Cheng Qian, Tianyu Yu, Huan- ang Gao, Wenkai Yang, Zhiyuan Liu, et al. Rethinking on-policy distillation of large language models: Phenomenology, mechanism, and recipe.arXiv preprint arXiv:2604.13016,

  4. [4]

    Deepseek-r1: Incentivizing reasoning capability in llms via reinforcement learning.arXiv preprint arXiv:2501.12948,

    Daya Guo, Dejian Yang, Haowei Zhang, Junxiao Song, Peiyi Wang, Qihao Zhu, Runxin Xu, Ruoyu Zhang, Shirong Ma, Xiao Bi, et al. Deepseek-r1: Incentivizing reasoning capability in llms via reinforcement learning.arXiv preprint arXiv:2501.12948,

  5. [5]

    Self-distilled reasoner: On-policy self-distillation for large language models.arXiv preprint arXiv:2601.18734,

    Siyan Zhao, Zhihui Xie, Mengchen Liu, Jing Huang, Guan Pang, Feiyu Chen, and Aditya Grover. Self-distilled reasoner: On-policy self-distillation for large language models.arXiv preprint arXiv:2601.18734,

  6. [6]

    Reinforcement learning via self-distillation.arXiv preprint arXiv:2601.20802,

    Jonas Hübotter, Frederike Lübeck, Lejs Behric, Anton Baumann, Marco Bagatella, Daniel Marta, Ido Hakimi, Idan Shenfeld, Thomas Kleine Buening, Carlos Guestrin, et al. Reinforcement learning via self-distillation.arXiv preprint arXiv:2601.20802,

  7. [7]

    Self-distillation enables continual learning.arXiv preprint arXiv:2601.19897,

    Idan Shenfeld, Mehul Damani, Jonas Hübotter, and Pulkit Agrawal. Self-distillation enables continual learning.arXiv preprint arXiv:2601.19897,

  8. [8]

    Your absorbing discrete diffusion secretly models the conditional distributions of clean data.arXiv preprint arXiv:2406.03736,

    Jingyang Ou, Shen Nie, Kaiwen Xue, Fengqi Zhu, Jiacheng Sun, Zhenguo Li, and Chongxuan Li. Your absorbing discrete diffusion secretly models the conditional distributions of clean data.arXiv preprint arXiv:2406.03736,

  9. [9]

    Large language diffusion models.arXiv preprint arXiv:2502.09992,

    Shen Nie, Fengqi Zhu, Zebin You, Xiaolu Zhang, Jingyang Ou, Jun Hu, Jun Zhou, Yankai Lin, Ji- Rong Wen, and Chongxuan Li. Large language diffusion models.arXiv preprint arXiv:2502.09992,

  10. [10]

    Dream 7b: Diffusion large language models.arXiv preprint arXiv:2508.15487,

    10 Jiacheng Ye, Zhihui Xie, Lin Zheng, Jiahui Gao, Zirui Wu, Xin Jiang, Zhenguo Li, and Lingpeng Kong. Dream 7b: Diffusion large language models.arXiv preprint arXiv:2508.15487,

  11. [11]

    Sdar: A synergistic diffusion-autoregression paradigm for scalable sequence generation.arXiv preprint arXiv:2510.06303,

    Shuang Cheng, Yihan Bian, Dawei Liu, Linfeng Zhang, Qian Yao, Zhongbo Tian, Wenhai Wang, Qipeng Guo, Kai Chen, Biqing Qi, et al. Sdar: A synergistic diffusion-autoregression paradigm for scalable sequence generation.arXiv preprint arXiv:2510.06303,

  12. [12]

    Tiwei Bie, Maosong Cao, Kun Chen, Lun Du, Mingliang Gong, Zhuochen Gong, Yanmei Gu, Jiaqi Hu, Zenan Huang, Zhenzhong Lan, et al. Llada2. 0: Scaling up diffusion language models to 100b. arXiv preprint arXiv:2512.15745,

  13. [13]

    Openai o1 system card.arXiv preprint arXiv:2412.16720,

    Aaron Jaech, Adam Kalai, Adam Lerer, Adam Richardson, Ahmed El-Kishky, Aiden Low, Alec Helyar, Aleksander Madry, Alex Beutel, Alex Carney, et al. Openai o1 system card.arXiv preprint arXiv:2412.16720,

  14. [14]

    Mimo-v2-flash technical report.arXiv preprint arXiv:2601.02780,

    Bangjun Xiao, Bingquan Xia, Bo Yang, Bofei Gao, Bowen Shen, Chen Zhang, Chenhong He, Chiheng Lou, Fuli Luo, Gang Wang, et al. Mimo-v2-flash technical report.arXiv preprint arXiv:2601.02780,

  15. [15]

    Seed diffusion: A large-scale diffusion language model with high-speed inference.arXiv preprint arXiv:2508.02193,

    Yuxuan Song, Zheng Zhang, Cheng Luo, Pengyang Gao, Fan Xia, Hao Luo, Zheng Li, Yuehang Yang, Hongli Yu, Xingwei Qu, et al. Seed diffusion: A large-scale diffusion language model with high-speed inference.arXiv preprint arXiv:2508.02193,

  16. [16]

    Fast-dllm: Training-free acceleration of diffusion llm by enabling kv cache and parallel decoding.arXiv preprint arXiv:2505.22618,

    Chengyue Wu, Hao Zhang, Shuchen Xue, Zhijian Liu, Shizhe Diao, Ligeng Zhu, Ping Luo, Song Han, and Enze Xie. Fast-dllm: Training-free acceleration of diffusion llm by enabling kv cache and parallel decoding.arXiv preprint arXiv:2505.22618,

  17. [17]

    wd1: Weighted policy optimization for reasoning in diffusion language models.arXiv preprint arXiv:2507.08838,

    Xiaohang Tang, Rares Dolga, Sangwoong Yoon, and Ilija Bogunovic. wd1: Weighted policy optimization for reasoning in diffusion language models.arXiv preprint arXiv:2507.08838,

  18. [18]

    Step-aware policy optimization for reasoning in diffusion large language models.arXiv preprint arXiv:2510.01544,

    Shaoan Xie, Lingjing Kong, Xiangchen Song, Xinshuai Dong, Guangyi Chen, Eric P Xing, and Kun Zhang. Step-aware policy optimization for reasoning in diffusion large language models.arXiv preprint arXiv:2510.01544,

  19. [19]

    Revolutionizing reinforce- ment learning framework for diffusion large language models.arXiv preprint arXiv:2509.06949,

    Yinjie Wang, Ling Yang, Bowen Li, Ye Tian, Ke Shen, and Mengdi Wang. Revolutionizing reinforce- ment learning framework for diffusion large language models.arXiv preprint arXiv:2509.06949,

  20. [20]

    d3llm: Ultra-fast diffusion llm using pseudo-trajectory distillation.arXiv preprint arXiv:2601.07568,

    Yu-Yang Qian, Junda Su, Lanxiang Hu, Peiyuan Zhang, Zhijie Deng, Peng Zhao, and Hao Zhang. d3llm: Ultra-fast diffusion llm using pseudo-trajectory distillation.arXiv preprint arXiv:2601.07568,

  21. [21]

    Cd4lm: Consistency distillation and adaptive decoding for diffusion language models.arXiv preprint arXiv:2601.02236,

    Yihao Liang, Ze Wang, Hao Chen, Ximeng Sun, Jialian Wu, Xiaodong Yu, Jiang Liu, Emad Barsoum, Zicheng Liu, and Niraj K Jha. Cd4lm: Consistency distillation and adaptive decoding for diffusion language models.arXiv preprint arXiv:2601.02236,

  22. [22]

    d-treerpo: Towards more reliable policy optimization for diffusion language models.arXiv preprint arXiv:2512.09675,

    Leyi Pan, Shuchang Tao, Yunpeng Zhai, Zheyu Fu, Liancheng Fang, Minghua He, Lingzhe Zhang, Zhaoyang Liu, Bolin Ding, Aiwei Liu, et al. d-treerpo: Towards more reliable policy optimization for diffusion language models.arXiv preprint arXiv:2512.09675,

  23. [23]

    Training verifiers to solve math word problems.arXiv preprint arXiv:2110.14168,

    Karl Cobbe, Vineet Kosaraju, Mohammad Bavarian, Mark Chen, Heewoo Jun, Lukasz Kaiser, Matthias Plappert, Jerry Tworek, Jacob Hilton, Reiichiro Nakano, et al. Training verifiers to solve math word problems.arXiv preprint arXiv:2110.14168,

  24. [24]

    Llada 1.5: Variance-reduced preference optimization for large language diffusion models.arXiv preprint arXiv:2505.19223,

    Fengqi Zhu, Rongzhen Wang, Shen Nie, Xiaolu Zhang, Chunwei Wu, Jun Hu, Jun Zhou, Jianfei Chen, Yankai Lin, Ji-Rong Wen, et al. Llada 1.5: Variance-reduced preference optimization for large language diffusion models.arXiv preprint arXiv:2505.19223,

  25. [25]

    Block diffusion: Interpolating between autoregres- sive and diffusion language models.arXiv preprint arXiv:2503.09573,

    Marianne Arriola, Aaron Gokaslan, Justin T Chiu, Zhihan Yang, Zhixuan Qi, Jiaqi Han, Sub- ham Sekhar Sahoo, and V olodymyr Kuleshov. Block diffusion: Interpolating between autoregres- sive and diffusion language models.arXiv preprint arXiv:2503.09573,

  26. [26]

    On grpo collapse in search-r1: The lazy likelihood-displacement death spiral.arXiv preprint arXiv:2512.04220,

    Wenlong Deng, Yushu Li, Boying Gong, Yi Ren, Christos Thrampoulidis, and Xiaoxiao Li. On grpo collapse in search-r1: The lazy likelihood-displacement death spiral.arXiv preprint arXiv:2512.04220,

  27. [27]

    M-grpo: Stabilizing self-supervised reinforcement learning for large language models with momentum-anchored policy optimization.arXiv preprint arXiv:2512.13070,

    Bizhe Bai, Hongming Wu, Peng Ye, and Tao Chen. M-grpo: Stabilizing self-supervised reinforcement learning for large language models with momentum-anchored policy optimization.arXiv preprint arXiv:2512.13070,

  28. [28]

    Distilling the knowledge in a neural network.arXiv preprint arXiv:1503.02531,

    Geoffrey Hinton, Oriol Vinyals, and Jeff Dean. Distilling the knowledge in a neural network.arXiv preprint arXiv:1503.02531,

  29. [29]

    Sequence-level knowledge distillation

    Yoon Kim and Alexander M Rush. Sequence-level knowledge distillation. InProceedings of the 2016 conference on empirical methods in natural language processing, pages 1317–1327,

  30. [30]

    Tinybert: Distilling bert for natural language understanding

    Xiaoqi Jiao, Yichun Yin, Lifeng Shang, Xin Jiang, Xiao Chen, Linlin Li, Fang Wang, and Qun Liu. Tinybert: Distilling bert for natural language understanding. InFindings of the association for computational linguistics: EMNLP 2020, pages 4163–4174,

  31. [31]

    Learning beyond teacher: Generalized on-policy distillation with reward extrapolation.arXiv preprint arXiv:2602.12125,

    Wenkai Yang, Weijie Liu, Ruobing Xie, Kai Yang, Saiyong Yang, and Yankai Lin. Learning beyond teacher: Generalized on-policy distillation with reward extrapolation.arXiv preprint arXiv:2602.12125,

  32. [32]

    Unifying autoregressive and diffusion-based sequence generation.arXiv preprint arXiv:2504.06416,

    Nima Fathi, Torsten Scholak, and Pierre-André Noël. Unifying autoregressive and diffusion-based sequence generation.arXiv preprint arXiv:2504.06416,

  33. [33]

    Reinforcing the diffusion chain of lateral thought with diffusion language models.arXiv preprint arXiv:2505.10446,

    Zemin Huang, Zhiyang Chen, Zijun Wang, Tiancheng Li, and Guo-Jun Qi. Reinforcing the diffusion chain of lateral thought with diffusion language models.arXiv preprint arXiv:2505.10446,

  34. [34]

    Diffucoder: Understanding and improving masked diffusion models for code generation.arXiv preprint arXiv:2506.20639,

    Shansan Gong, Ruixiang Zhang, Huangjie Zheng, Jiatao Gu, Navdeep Jaitly, Lingpeng Kong, and Yizhe Zhang. Diffucoder: Understanding and improving masked diffusion models for code generation.arXiv preprint arXiv:2506.20639,

  35. [35]

    Improving reasoning for diffusion language models via group diffusion policy optimization

    Kevin Rojas, Jiahe Lin, Kashif Rasul, Anderson Schneider, Yuriy Nevmyvaka, Molei Tao, and Wei Deng. Improving reasoning for diffusion language models via group diffusion policy optimization. arXiv preprint arXiv:2510.08554,

  36. [36]

    Principled rl for diffusion llms emerges from a sequence-level perspective.arXiv preprint arXiv:2512.03759,

    Jingyang Ou, Jiaqi Han, Minkai Xu, Shaoxuan Xu, Jianwen Xie, Stefano Ermon, Yi Wu, and Chongxuan Li. Principled rl for diffusion llms emerges from a sequence-level perspective.arXiv preprint arXiv:2512.03759,

  37. [37]

    Ilya Loshchilov and Frank Hutter

    URLhttps://github.com/huggingface/trl. Ilya Loshchilov and Frank Hutter. Decoupled weight decay regularization.arXiv preprint arXiv:1711.05101,

  38. [38]

    Flashattention-2: Faster attention with better parallelism and work partitioning.arXiv preprint arXiv:2307.08691,

    12 Tri Dao. Flashattention-2: Faster attention with better parallelism and work partitioning.arXiv preprint arXiv:2307.08691,

  39. [39]

    end-of-text

    13 A Additional Preliminaries and Related Works A.1 Additional Preliminaries Block-diffusion.In practice, the block-diffusion inference strategy [Han et al., 2023, Arriola et al., 2025, Fathi et al., 2025] is commonly used in current dLLMs. This hybrid approach partitions a response y into B contiguous, non-overlapping blocks {block1,block 2,· · ·,block B...

  40. [40]

    C Additional Implementation Details C.1 Per-Token pointwise clipping Following [Zhao et al., 2026], we apply pointwise clipping to the vocabulary level divergence contributions. The reason is that token-level divergence is highly skewed across vocabulary entries, and our ablation study in Section 4.4 empirically validates that pointwise clipping stabilize...

  41. [41]

    right” or “wrong

    Although computing on all generations also improves the model’s reasoning performance, our default setting achieves superior results. Detailed experimental results are provided in Section E.1. D Additional Experiment Details D.1 Training Details We used the TRL library [von Werra et al., 2020] to implement d-OPSD. We employed Low-Rank Adaptation (LoRA) wi...