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

Yifu Luo, Zeyu Chen, Haoyu Wang, Xinhao Hu, Yuxuan Zhang +2 more
6/16/2026
cs.CL

No official code found. Best implementation: xingzhejun/d-opsd-code (19 stars, last commit 2026-06-15).

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.

View on arXivView PDF

Code Implementations(1)

The code for paper "Learning from the Self-future:On-policy Self-distillation for dLLMs"

195May 30, 20264 months ago

Similar Papers

Cite this paper

@article{luo2026learning,
  title  = {Learning from the Self-future: On-policy Self-distillation for dLLMs},
  author = {Yifu Luo and Zeyu Chen and Haoyu Wang and Xinhao Hu and Yuxuan Zhang and Zhizhou Sha and Shiwei Liu},
  year   = {2026},
  eprint = {2606.18195},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2606.18195v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2606.18195.svg)](https://codeofpaper.com/paper/2606.18195)

Discussion