Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

LFPO: Likelihood-Free Policy Optimization for Masked Diffusion Models

About

Reinforcement Learning with Verifiable Rewards (RLVR) has achieved remarkable success in improving autoregressive models, especially in domains requiring correctness like mathematical reasoning and code generation. However, directly applying such paradigms to Diffusion Large Language Models (dLLMs) is fundamentally hindered by the intractability of exact likelihood computation, which forces existing methods to rely on high-variance approximations. To bridge this gap, we propose Likelihood-Free Policy Optimization (LFPO), a native framework that maps the concept of vector field flow matching to the discrete token space. Specifically, LFPO formulates alignment as geometric velocity rectification, which directly optimizes denoising logits via contrastive updates. This design effectively bypasses the errors inherent in likelihood approximation, yielding the precise gradient estimation. Furthermore, LFPO enforce consistency by predicting final solutions from intermediate steps, effectively straightening the probability flow to enable high-quality generation with significantly fewer iterations. Extensive experiments demonstrate that LFPO not only outperforms state-of-the-art baselines on code and reasoning benchmarks but also accelerates inference by approximately 20% through reduced diffusion steps.

Chenxing Wei, Jiazhen Kang, Hong Wang, Jianqing Zhang, Hao Jiang, Xiaolong Xu, Ningyuan Sun, Ying He, F. Richard Yu, Yao Shu, Bo Jiang• 2026

Related benchmarks

TaskDatasetResultRank
Commonsense ReasoningWinoGrande
Accuracy86.9
1085
Code GenerationHumanEval
Pass@175.6
1036
Physical Commonsense ReasoningPIQA
Accuracy85.9
572
Code GenerationHumanEval+
Pass@170.1
383
Mathematical ReasoningMATH
Accuracy37.6
338
Science ReasoningGPQA
Accuracy27.1
243
Code GenerationMBPP+
Pass@171.3
216
Common Sense ReasoningHellaSwag
Accuracy85.7
213
Code GenerationMBPP
Pass@181.6
159
Code GenerationEvalPlus
Pass@169.5
61
Showing 10 of 12 rows

Other info

GitHub

Follow for update