Relative Entropy-Regularized Optimal Transport on a Graph: a new algorithm and an experimental comparison

Courtain, Sylvain;Guex, Guillaume;Ilkka Kivimäki;Saerens, Marco
(2022) International Journal of Machine Learning and Cybernetics — Vol. 13, n° 12, p. 1-26 (2022)

Files

Article.pdf
  • Open Access
  • Adobe PDF
  • 1.29 MB

Details

Authors
  • Courtain, Sylvainorcid-logoUCLouvain
    Author
  • Guex, GuillaumeUniversité de Lausanne
    Author
  • Ilkka KivimäkiAalto University
    Author
  • Author
Abstract
The present work investigates a new relative entropy-regularized algorithm for solving the optimal transport on a graph problem within the randomized shortest paths formalism. More precisely, a unit flow is injected into a set of input nodes and collected from a set of output nodes with specified marginals, while minimizing the expected transportation cost, together with a paths-based relative entropy regularization term, providing a randomized routing policy. The main advantage of this new formulation is the fact that it can easily accommodate edge flow capacity constraints which commonly occur in real-world problems. The resulting optimal routing policy, i.e., the probability distribution of following an edge in each node, is Markovian and is computed after constraining the input and output flows to the prescribed marginal probabilities. In addition, experimental comparisons with other recently developed techniques show that the distance measure between nodes derived from the introduced model provides competitive results on semi-supervised classification tasks.
Affiliations

Citations

Courtain, S., Guex, G., Ilkka Kivimäki, & Saerens, M. (2022). Relative Entropy-Regularized Optimal Transport on a Graph: a new algorithm and an experimental comparison. International Journal of Machine Learning and Cybernetics, 13(12), 1-26. https://doi.org/10.1007/s13042-022-01704-6 (Original work published 2022)