Efficient Token-Level Policy Optimization for Masked Diffusion Models via Velocity Alignment

Chenxing Wei, Jiazhen Kang, Hong Wang, Jianqing Zhang, Hao Jiang, Xuxiaolong, Ningyuan Sun, Ying He, Fei Yu, Yao Shu, Bo Jiang


Abstract
Reinforcement Learning with Verifiable Rewards (RLVR) successfully aligns autoregressive models in rigorous domains like mathematical reasoning and code generation. However, applying RLVR to Diffusion Large Language Models (dLLMs) is hindered by intractable sequence-level likelihoods, forcing existing methods to rely on unstable, high-variance approximations. We propose Sequence Likelihood-Free Policy Optimization (LFPO), an efficient framework that circumvents this bottleneck by formulating policy alignment as discrete velocity alignment. Inspired by continuous flow matching, LFPO eschews trajectory-level estimation and directly optimizes denoising logits via contrastive token-level updates, providing a stable, low-variance surrogate. Furthermore, predicting terminal solutions from intermediate states implicitly straightens the generation trajectory, enabling high-quality sampling with significantly fewer denoising iterations. Extensive experiments on reasoning and coding benchmarks demonstrate that LFPO matches or exceeds the accuracy of likelihood-based baselines while substantially accelerating inference.
Anthology ID:
2026.emnlp-main.1910
Volume:
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
EMNLP
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
43516–43536
Language:
URL:
https://aclanthology.org/2026.emnlp-main.1910/
DOI:
Bibkey:
Cite (ACL):
Chenxing Wei, Jiazhen Kang, Hong Wang, Jianqing Zhang, Hao Jiang, Xuxiaolong, Ningyuan Sun, Ying He, Fei Yu, Yao Shu, and Bo Jiang. 2026. Efficient Token-Level Policy Optimization for Masked Diffusion Models via Velocity Alignment. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, pages 43516–43536, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Efficient Token-Level Policy Optimization for Masked Diffusion Models via Velocity Alignment (Wei et al., EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-main.1910.pdf