mlsanityData to 7 Oct 2026

DFPO: Scaling Value Modeling via Distributional Flow towards Robust and Generalizable LLM Post-Training

arXiv 2602.05890 ↗First posted 5 Feb 2026cs.LG cs.CLcited by 0

Not on the map. Why?

Abstract (arXiv)

Training reinforcement learning (RL) systems in real-world environments remains challenging due to noisy supervision and poor out-of-domain (OOD) generalization, especially in LLM post-training. Recent distributional RL methods improve robustness by modeling values with multiple quantile points, but they still learn each quantile independently as a scalar. This results in rough-grained value representations that lack fine-grained conditioning on state information, struggling under complex and OOD conditions. We propose DFPO (Distributional Value Flow Policy Optimization with Conditional Risk and Consistency Control), a robust distributional RL framework that models values as continuous flows across time steps. By scaling value modeling through learning of a value flow field instead of isolated quantile predictions, DFPO captures richer state information for more accurate advantage estimation. To stabilize training under noisy feedback, DFPO further integrates conditional risk control and consistency constraints along value flow trajectories. Experiments on dialogue, math reasoning, and scientific tasks show that DFPO outperforms PPO, FlowRL, and other robust baselines under noisy supervision, achieving improved training stability and generalization.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 32 of its 36 references that are in the map. First posted on arXiv 5 Feb 2026.

Builds on

32 of 36 references.

References · 32

  1. 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
  2. 2018Neural Ordinary Differential Equationscited by 3,733
  3. 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,550
  4. 2022Solving Quantitative Reasoning Problems with Language Modelscited by 1,359
  5. 2017A Distributional Perspective on Reinforcement Learningcited by 743
  6. 2017Distributional Reinforcement Learning with Quantile Regressioncited by 464
  7. 2019Augmented Neural ODEscited by 396
  8. 2018Implicit Quantile Networks for Distributional Reinforcement Learningcited by 362
  9. 2024Measuring short-form factuality in large language modelscited by 345
  10. 2025LLaDA 1.5: Variance-Reduced Preference Optimization for Large Language Diffusion Modelscited by 273
  11. 2024AgentGym: Evolving Large Language Model-based Agents across Diverse Environmentscited by 120
  12. 2025Flow Matching Policy Gradientscited by 86
20 more
  1. 2025ReinFlow: Fine-tuning Flow Matching Policy with Online Reinforcement Learningcited by 83
  2. 2021Conservative Offline Distributional Reinforcement Learningcited by 78
  3. 2024Training Large Language Models for Reasoning through Reverse Curriculum Reinforcement Learningcited by 63
  4. 2024Riemannian Flow Matching Policy for Robot Motion Learningcited by 61
  5. 2025BAPO: Stabilizing Off-Policy Reinforcement Learning for LLMs via Balanced Policy Optimization with Adaptive Clippingcited by 47
  6. 2025FlowRL: Matching Reward Distributions for LLM Reasoningcited by 38
  7. 2022Let Offline RL Flow: Training Conservative Agents in the Latent Space of Normalizing Flowscited by 21
  8. 2025floq: Training Critics via Flow-Matching for Scaling Compute in Value-Based RLcited by 20
  9. 2025KTAE: A Model-Free Algorithm to Key-Tokens Advantage Estimation in Mathematical Reasoningcited by 19
  10. 2025Qsharp: Provably Optimal Distributional RL for LLM Post-Trainingcited by 14
  11. 2024Mitigating Tail Narrowing in LLM Self-Improvement via Socratic-Guided Samplingcited by 13
  12. 2025Normalizing Flows are Capable Models for Continuous Controlcited by 13
  13. 2025CPGD: Toward Stable Rule-based Reinforcement Learning for Language Modelscited by 11
  14. 2025λ-GRPO: Unifying the GRPO Frameworks with Learnable Token Preferencescited by 10
  15. 2025VRPO: Rethinking Value Modeling for Robust RL under Noisy Supervision in LLM Post-Trainingcited by 9
  16. 2022Distributional Reinforcement Learning with Regularized Wasserstein Losscited by 8
  17. 2024Learning Robust Reward Machines from Noisy Labelscited by 7
  18. 2023Noise Distribution Decomposition based Multi-Agent Distributional Reinforcement Learningcited by 4
  19. 2024ARMCHAIR: integrated inverse reinforcement learning and model predictive control for human-robot collaborationcited by 2
  20. 2025FM-IRL: Flow-Matching for Reward Modeling and Policy Regularization in Reinforcement Learningcited by 2

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025DVPO: Distributional Value Modeling-based Policy Optimization for LLM Post-Training
  2. 2026Privy to the Foil: Recasting Value Estimation with a Self-Privileged Critic for RLVR
  3. 2026Robo-ValueRL: Reliable Value Estimation for Offline-to-Online Reinforcement Learning
  4. 2026Distributional value gradients for stochastic environments
  5. 2026Reinforcement Learning via Value Gradient Flow
  6. 2025FlowCritic: Bridging Value Estimation with Flow Matching in Reinforcement Learning
  7. 2026Bringing Value Models Back: Generative Critics for Value Modeling in LLM Reinforcement Learning
  8. 2026V0.5: Generalist Value Model as a Prior for Sparse RL Rollouts