mlsanityData to 7 Oct 2026

Optimal Transport Meets Reinforcement Learning: A Survey

arXiv 2610.01413 ↗First posted 1 Oct 2026stat.ML cs.AI cs.LGcited by 0

On the map: Offline RL and imitation learning from data → Diffusion and flow policies for offline RL also fits: Offline RL algorithms

Abstract (arXiv)

Reinforcement learning (RL) algorithms frequently compare probability distributions, such as state visitation distributions induced by policies and experts, action distributions from learned policies and offline datasets, or transition distributions from learned models and environments. However, commonly used divergences may become ineffective when these distributions overlap weakly, which is frequently encountered in imitation learning, offline RL, and deployment under distribution shift. Optimal transport (OT) offers an alternative by measuring the cost of moving probability mass from one distribution to another under a ground cost that encodes task geometry. This survey covers how OT is used inside RL objectives and algorithms. For each method, we identify: the role OT plays, the distributions compared, the OT formulation used, and the treatment of temporal structure. Beyond categorising existing methods, we discuss the motivations behind different OT choices, practical considerations such as cost design and computational challenges, and highlight open problems including scalable trajectory-level transport, principled handling of mass mismatch, and theoretical analysis for OT-regularised RL.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Diffusion and flow policies for offline RL, one of 13 directions in Offline RL and imitation learning from data. First posted on arXiv 1 Oct 2026.

Builds on

0 of 0 references.

No references here yet.

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2023Optimal Transport for Offline Imitation Learning
  2. 2024Rethinking Optimal Transport in Offline Reinforcement Learning
  3. 2025Score-Based Diffusion Policy Compatible with Reinforcement Learning via Optimal Transport
  4. 2023Leveraging Optimal Transport for Enhanced Offline Reinforcement Learning in Surgical Robotic Environments
  5. 2024Align Your Intents: Offline Imitation Learning via Optimal Transport
  6. 2026Designing Reinforcement Learning for Diffusion Models: A Unified Path-Space View
  7. 2024Towards Sample-Efficiency and Generalization of Transfer and Inverse Reinforcement Learning: A Comprehensive Literature Review
  8. 2025Diffusion Models for Reinforcement Learning: Foundations, Taxonomy, and Development