LFPO: Likelihood-Free Policy Optimization for Masked Diffusion Models
On the map: LLM reasoning → Diffusion reasoning and test-time scaling
also in 1 more
Abstract (arXiv)
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.
Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Placed in Diffusion reasoning and test-time scaling, one of 17 directions in LLM reasoning. Builds on 43 of its 49 references that are in the map. First posted on arXiv 2 Mar 2026.
Builds on
43 of 49 references.
Closest ancestors · 3
Same direction, newest first
- 2025Improving Reasoning for Diffusion Language Models via Group Diffusion Policy Optimization9 Oct
- 2025d1: Scaling Reasoning in Diffusion Large Language Models via Reinforcement Learning16 Apr
- 2024Beyond Autoregression: Discrete Diffusion for Complex Reasoning and Planning18 Oct
Other references · 40
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2021Training Verifiers to Solve Math Word Problemscited by 9,743
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2021Evaluating Large Language Models Trained on Codecited by 8,053
- 2022Flow Matching for Generative Modelingcited by 5,426
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,018
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,550
- 2021Program Synthesis with Large Language Modelscited by 3,509
- 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,818
- 2023GPQA: A Graduate-Level Google-Proof Q&A Benchmarkcited by 2,748
28 more
- 2015Trust Region Policy Optimizationcited by 2,503
- 2021Structured Denoising Diffusion Models in Discrete State-Spacescited by 1,665
- 2023Is Your Code Generated by ChatGPT Really Correct? Rigorous Evaluation of Large Language Models for Code Generationcited by 1,332
- 2024Qwen2.5-Coder Technical Reportcited by 1,151
- 2024Visual Autoregressive Modeling: Scalable Image Generation via Next-Scale Predictioncited by 1,005
- 2025Large Language Diffusion Modelscited by 872
- 2023Discrete Diffusion Modeling by Estimating the Ratios of the Data Distributioncited by 712
- 2025Flow-GRPO: Training Flow Matching Models via Online RLcited by 605
- 2024BigCodeBench: Benchmarking Code Generation with Diverse Function Calls and Complex Instructionscited by 527
- 2025Dream 7B: Diffusion Large Language Modelscited by 515
- 2025Fast-dLLM: Training-free Acceleration of Diffusion LLM by Enabling KV Cache and Parallel Decodingcited by 404
- 2025LLaDA 1.5: Variance-Reduced Preference Optimization for Large Language Diffusion Modelscited by 273
- 2025MMaDA: Multimodal Large Diffusion Language Modelscited by 241
- 2025DiffusionNFT: Online Diffusion Reinforcement with Forward Processcited by 204
- 2025DiffuCoder: Understanding and Improving Masked Diffusion Models for Code Generationcited by 184
- 2023The probability flow ODE is provably fastcited by 145
- 2020OT-Flow: Fast and Accurate Continuous Normalizing Flows via Optimal Transportcited by 143
- 2025Diffusion LLMs Can Do Faster-Than-AR Inference via Discrete Diffusion Forcingcited by 111
- 2025ACECODER: Acing Coder RL via Automated Test-Case Synthesiscited by 89
- 2025Revolutionizing Reinforcement Learning Framework for Diffusion Large Language Modelscited by 80
- 2025Discrete Diffusion in Large Language and Multimodal Models: A Surveycited by 61
- 2023PLANNER: Generating Diversified Paragraph via Latent Language Diffusion Modelcited by 56
- 2025SPG: Sandwiched Policy Gradient for Masked Diffusion Language Modelscited by 38
- 2025Training Optimal Large Diffusion Language Modelscited by 21
- 2026Stable-DiffCoder: Pushing the Frontier of Code Diffusion Large Language Modelcited by 12
- 2025ReDit: Reward Dithering for Improved LLM Policy Optimizationcited by 8
- 2025DeepSeek: Paradigm Shifts and Technical Evolution in Large AI Modelscited by 3
- 2025Likelihood Training of Cascaded Diffusion Models via Hierarchical Volume-preserving Mapscited by 2
Built on by
Citing papers per year: 2026: 2.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2025MDPO: Overcoming the Training-Inference Divide of Masked Diffusion Language Models
- 2025wd1: Weighted Policy Optimization for Reasoning in Diffusion Language Models
- 2026Reinforcement Learning for Diffusion LLMs with Entropy-Guided Step Selection and Stepwise Advantages
- 2025Learning Unmasking Policies for Diffusion Language Models
- 2025Principled RL for Diffusion LLMs Emerges from a Sequence-Level Perspective
- 2026Relative Score Policy Optimization for Diffusion Language Models
- 2025Inpainting-Guided Policy Optimization for Diffusion Large Language Models
- 2026Mask-Aware Policy Gradients for Diffusion Language Models