Entropic Regularized Optimal Transport
- Entropic OT is a method that adds a relative entropy penalty to classical optimal transport, ensuring strict convexity and unique solutions.
- The formulation leverages dual Schrödinger systems and elliptic regularization, yielding smooth potential functions and robust convergence as the regularization vanishes.
- Its computational advantages enable efficient algorithms like Sinkhorn for large-scale numerical and mean-field planning problems.
Entropic regularized optimal transport (Entropic OT) incorporates an additional relative entropy penalty into the classical Kantorovich formulation, yielding a strictly convex, smoothly solvable optimal transport problem with strong regularity and favorable computational properties. Entropic regularization leads to elliptic smoothing of the potential functions and the couplings, convergence guarantees as the regularization vanishes, and is a fundamental building block in large-scale numerical and statistical optimal transport, as well as mean-field games and planning problems.
1. Mathematical Formulation: Static and Dynamic Entropic OT
Entropic regularization modifies the Kantorovich OT problem between two probability densities on a bounded convex domain by penalizing the negative (Boltzmann–Shannon) entropy of couplings relative to : where for . With , the standard quadratic cost, this functional is strictly convex for and admits a unique minimizer (Porretta, 2022).
Dynamic formulations based on the Benamou–Brenier approach regularize the kinetic action by an entropy term along the path: subject to the continuity equation and boundary conditions .
2. Duality, Schrödinger Systems, and Elliptic Regularization
The entropic cost yields a Schrödinger bridge dual of the form
whose stationarity provides the optimal potentials. For the quadratic cost , differentiation yields a quasilinear elliptic PDE for the Kantorovich potential: where encodes reference measure weights. The entropic term provides a vanishing-viscosity elliptic regularization of the Hamilton–Jacobi equation encountered in classical OT.
The corresponding Schrödinger system for the coupling recovers the Gibbs-type structure: For the dynamic (path-space) entropic OT problem, duality induces an elliptic system in space-time for the value function and density , yielding the coupled system: with no-flux (Neumann) boundary conditions (Porretta, 2022).
3. Regularity and Limit
Assuming smooth, positive initial and terminal densities, and uniformly convex Hamiltonian, the entropic regularization yields a unique smooth classical solution and , with a full suite of a priori bounds:
- , gradient, Hölder, and higher-order Schauder estimates
- Uniform positivity and regularity of over
As , the solutions converge (in strong Hölder spaces) to the classical optimal transport geodesic and Kantorovich potential, recovering the singular first-order PDEs of standard OT. The elliptic regularization provides smooth approximants for use in variational problems, mean-field control, and planning under congestion (Porretta, 2022).
4. Displacement Convexity and Application to Mean-Field Problems
The entropic functional allows new displacement convexity estimates in the Eulerian approach. For any convex function with , displacement convexity implies: In the entropic case, convexity arguments provide additional local -bounds and control the supremum of density along the optimal flow. As these inequalities recover McCann's classical displacement convexity for OT (Porretta, 2022).
From a mean-field games/planning perspective, the entropic regularization principle originated in works by P.-L. Lions and enables smooth, strongly convex approximations to optimization problems with congestion penalization and mean-field interaction.
5. Key Algorithmic and Computational Features
The strict convexity and regularization provided by entropy enable efficient numerical solution by matrix scaling (Sinkhorn) algorithms and their modern variants. In the discrete case, the entropic OT problem yields a dual with smooth convex structure, solved efficiently with alternating row-column normalizations. Entropic regularization ensures well-conditioned iterations, uniform positivity of couplings, and facilitates convergence analysis:
- The entropy term prevents the degeneracy and support-sparsity of classic OT solutions, making all entries strictly positive.
- This ensures the applicability of high-performance first-order methods (Sinkhorn, Greenkhorn, APDAMD) and efficient Newton-type solvers exploiting the strict strong convexity/concavity of the regularized structure (Lin et al., 2019).
For small regularization, the entropic plan approximates the classical OT plan with vanishing error, while for large regularization, it interpolates toward the independent (product) coupling.
6. Summary Table: Key Properties of Entropic OT
| Aspect | Property/Result | Reference |
|---|---|---|
| Static formulation | Adds to cost | (Porretta, 2022) |
| Dual system | Schrödinger system for potentials; log-Gibbs kernel | (Porretta, 2022) |
| Regularity | Solutions smooth, elliptic regularity for potentials | (Porretta, 2022) |
| Limit | Classical OT geodesic and potential recovered | (Porretta, 2022) |
| Displacement convexity | Derived via regularized functionals, density bounds | (Porretta, 2022) |
| Numerical aspects | Efficient, well-posed, uniform positivity, parallelizable | (Porretta, 2022, Lin et al., 2019) |
Uniform elliptic regularization fundamentally improves theoretical and computational properties of OT, supporting further developments in analysis, numerical methods, and mean-field PDEs.
7. Further Directions and Applications
The entropic regularization of OT underpins much of the recent progress in large-scale computational optimal transport, statistical learning, and mean-field optimization. It provides a robust approximation framework for variational OT, JKO schemes for gradient flows, and dynamic mean-field game systems. The vanishing-viscosity interpretation is instrumental in regularizing degenerate PDEs in transport, enabling displacement convexity and strong a priori estimates in mean-field planning and control (Porretta, 2022).
The entropy regularized approach yields unique, smooth approximations that converge robustly to the sharp classical solution, allowing the exploitation of convex analytic, PDE-algorithmic, and computational advantages across a wide spectrum of OT-related fields.