Inverse Entropic Optimal Transport Solves Semi-supervised Learning via Data Likelihood Maximization
Mikhail Persiianov, Arip Asadulaev, Nikita Andreev, Nikita Starodubcev, Dmitry Baranchuk, Anastasis Kratsios, Evgeny Burnaev, Aleksandr Korotin
Abstract
Learning conditional distributions π * (•|x) is a central problem in machine learning, which is typically approached via supervised methods with paired data (x, y) ∼ π * . However, acquiring paired data samples is often challenging, especially in problems such as domain translation. This necessitates the development of semisupervised models that utilize both limited paired data and additional unpaired i.i.d. samples x ∼ π * x and y ∼ π * y from the marginal distributions. The usage of such combined data is complex and often relies on heuristic approaches. To tackle this issue, we propose a new learning paradigm called EBiEOT that integrates both paired and unpaired data seamlessly using data likelihood maximization techniques. We demonstrate that our approach also connects intriguingly with inverse entropic optimal transport (OT). This finding allows us to apply recent advances in computational OT to establish an end-to-end learning algorithm to get π * (•|x). In addition, we derive the universal approximation property, demonstrating that our approach can theoretically recover true conditional distributions with arbitrarily small error. Finally, we demonstrate through empirical tests that our method effectively learns conditional distributions using paired and unpaired data simultaneously. Source code: https: //github.com/MuXauJl11110/EBiEOT .
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 3dcf256d-e378-450d-a1e4-eee4014ac49eCited by top-tier papers1
Ask how each one uses itBuilds on43
- Structured Denoising Diffusion Models in Discrete State-SpacesJacob Austin, Daniel D. Johnson, Jonathan Ho, Daniel Tarlow et al.NeurIPS 2021 · 2,256 citations
- Diffusion Schrödinger Bridge with Applications to Score-Based Generative ModelingValentin De Bortoli, James Thornton, Jeremy Heng, Arnaud DoucetNeurIPS 2021 · 811 citations
- PyTorch 2: Faster Machine Learning Through Dynamic Python Bytecode Transformation and Graph CompilationJason Ansel, Edward Z. Yang, Horace He, Natalia Gimelshein et al.ASPLOS 2024 · 693 citations
- A Continuous Time Framework for Discrete Denoising ModelsAndrew Campbell, Joe Benton, Valentin De Bortoli, Thomas Rainforth et al.NeurIPS 2022 · 496 citations
- Discrete Flow MatchingItai Gat, Tal Remez, Neta Shaul, Felix Kreuk et al.NeurIPS 2024 · 363 citations
Related papers
- Optimal Transport-Guided Conditional Score-Based Diffusion ModelXiang Gu, Liwei Yang, Jian Sun, Zongben XuNeurIPS 2023 · 12 citations
- UOTIP: Unbalanced Optimal Transport Map for Unpaired Inverse ProblemsDonggyu Lee, Taekyung Lee, Jaewoong ChoiICML 2026
- Solving Discrete (Semi) Unbalanced Optimal Transport with Equivalent Transformation Mechanism and KKT-Multiplier RegularizationWeiming Liu, Xinting Liao, Jun Dan, Fan Wang et al.NeurIPS 2025 · 2 citations
- Variational Entropic Optimal TransportRoman Dyachenko, Nikita Gushchin, Kirill Sokolov, Petr Mokrov et al.ICML 2026 · 1 citation
- Generative Conditional Distributions by Neural (Entropic) Optimal TransportBao Nguyen, Binh Nguyen, Hieu Trung Nguyen, Viet Anh NguyenICML 2024 · 2 citations
