mlsanityData to 7 Oct 2026

FLAC: Maximum Entropy RL via Kinetic Energy Regularized Bridge Matching

arXiv 2602.12829 ↗First posted 13 Feb 2026cs.LG cs.AIcited by 5

On the map: Diffusion models → Diffusion and flow policies for robotics

Abstract (arXiv)

Iterative generative policies, such as diffusion models and flow matching, offer superior expressivity for continuous control but complicate Maximum Entropy Reinforcement Learning because their action log-densities are not directly accessible. To address this, we propose Field Least-Energy Actor-Critic (FLAC), a likelihood-free framework that regulates policy stochasticity by penalizing the kinetic energy of the velocity field. Our key insight is to formulate policy optimization as a Generalized Schr"odinger Bridge (GSB) problem relative to a high-entropy reference process (e.g., uniform). Under this view, the maximum-entropy principle emerges naturally as staying close to a high-entropy reference while optimizing return, without requiring explicit action densities. In this framework, kinetic energy serves as a physically grounded proxy for divergence from the reference: minimizing path-space energy bounds the deviation of the induced terminal action distribution. Building on this view, we derive an energy-regularized policy iteration scheme and a practical off-policy algorithm that automatically tunes the kinetic energy via a Lagrangian dual mechanism. Empirically, FLAC achieves superior or comparable performance on high-dimensional benchmarks relative to strong baselines, while avoiding explicit density estimation.

Key facts. Cited by 5 papers in the map (data through 7 Oct 2026). Placed in Diffusion and flow policies for robotics, one of 14 directions in Diffusion models. Builds on 32 of its 39 references that are in the map. First posted on arXiv 13 Feb 2026.

Builds on

32 of 39 references.

Closest ancestors · 10

Same direction, newest first

  1. 2025Flow-GRPO: Training Flow Matching Models via Online RL8 May
  2. 2025Flow Q-Learning4 Feb
  3. 2025DIME:Diffusion-Based Maximum Entropy Reinforcement Learning4 Feb
  4. 2024Diffusion-based Reinforcement Learning via Q-weighted Variational Policy Optimization25 May
  5. 2024Diffusion Actor-Critic with Entropy Regulator24 May
  6. 2023Learning a Diffusion Model Policy from Rewards via Q-Score Matching18 Dec
  7. 2023Policy Representation via Diffusion Probability Model for Reinforcement Learning22 May
  8. 2023Contrastive Energy Prediction for Exact Energy-Guided Diffusion Sampling in Offline Reinforcement Learning25 Apr
  9. 2023Diffusion Policy: Visuomotor Policy Learning via Action Diffusion7 Mar
  10. 2022Diffusion Policies as an Expressive Policy Class for Offline Reinforcement Learning12 Aug

Other references · 22

  1. 2020Denoising Diffusion Probabilistic Modelscited by 20,851
  2. 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
  3. 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,185
  4. 2022Flow Matching for Generative Modelingcited by 5,426
  5. 2018Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actorcited by 4,635
  6. 2018Neural Ordinary Differential Equationscited by 3,733
  7. 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,550
  8. 2020Offline Reinforcement Learning: Tutorial, Review, and Perspectives on Open Problemscited by 1,873
  9. 2018DeepMind Control Suitecited by 935
  10. 2023TD-MPC2: Scalable, Robust World Models for Continuous Controlcited by 482
  11. 2013A survey of the Schr"odinger problem and some of its connections with optimal transportcited by 286
  12. 2023Diffusion Schr"odinger Bridge Matchingcited by 151
10 more
  1. 2021Solving Schr"odinger Bridges via Maximum Likelihoodcited by 140
  2. 2024HumanoidBench: Simulated Humanoid Benchmark for Whole-Body Locomotion and Manipulationcited by 101
  3. 2019Theoretical guarantees for sampling and inference in generative models with latent diffusionscited by 97
  4. 2023For SALE: State-Action Representation Learning for Deep Reinforcement Learningcited by 73
  5. 2025Flow-Based Policy for Online Reinforcement Learningcited by 33
  6. 2025SAC Flow: Sample-Efficient Reinforcement Learning of Flow-Based Policies via Velocity-Reparameterized Sequential Modelingcited by 27
  7. 2018The data-driven Schroedinger bridgecited by 26
  8. 2019Q-Learning for Continuous Actions with Cross-Entropy Guided Policiescited by 17
  9. 2024Sampling from Energy-based Policies using Diffusioncited by 15
  10. 2025Data-to-Energy Stochastic Dynamicscited by 5

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 5.

5 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026Boosting Maximum Entropy Reinforcement Learning via One-Step Flow Matching
  2. 2026Flow Matching Policy Optimization with Mirror Descent and Entropy Constraints
  3. 2025Energy-Weighted Flow Matching for Offline Reinforcement Learning
  4. 2026Truncated Rectified Flow Policy for Reinforcement Learning with One-Step Sampling
  5. 2025Max-Entropy Reinforcement Learning with Flow Matching and A Case Study on LQR
  6. 2025FlowQ: Energy-Guided Flow Policies for Offline Reinforcement Learning
  7. 2026MePoly: Max Entropy Polynomial Policy Optimization
  8. 2024Maximum Entropy Inverse Reinforcement Learning of Diffusion Models with Energy-Based Models