AI & ML
Deep Barycentric Regression for Optimal Transport Map Estimation and Its Statistical Optimality
arXiv stat.ML0 views
arXiv:2609.06598v2 Announce Type: replace-cross Abstract: The optimal transport (OT) map provides a geometric transformation for aligning probability distributions and has become a useful tool in machine learning. However, existing estimators of the OT map still exhibit a gap between sharp statistical guarantees and practical parametric estimation based on stable training objectives. Theoretical estimators achieve minimax optimal convergence rates, but they are typically nonparametric and can incur demanding implementation design or inference costs. Practical estimators are parametric and scalable, but their statistical guarantees remain underexplored, and their min-max, adversarial-like training objectives can be sensitive to optimization algorithms. We propose BROT (Barycentric Regression for Optimal Transport), a simple two-step method that first computes the unregularized OT plan and then fits a deep neural network (DNN) to the induced barycentric targets by least-squares regression. Under standard regularity conditions, we prove that the DNN estimator of BROT attains the minimax convergence rate, when the ground-truth OT map is Lipschitz. Numerical studies on synthetic datasets and an image dataset show that BROT provides accurate map estimates, strong target distribution matching, and competitive transport costs, compared to existing estimation methods. Experiments on two downstream tasks, single-cell perturbation prediction and unsupervised domain adaptation, further suggest that the accurate estimation of BROT can translate into stronger task performance.
Source:https://arxiv.org/abs/2609.06598PreviousGeometry-Constrained Kolmogorov-Arnold Networks: Learning Edge Geometry via Banach DualityNextSpeed limits for information acquisition in stochastic learning dynamics
Related posts
AI agents are your new customers. I built a checker to see which stores they can actually read
0dev.to AI
VibeVoice: Turn an Hour of Audio Into a Full Transcript in One Go
0dev.to AI
I built an app with ChatGPT or Claude. How do I put it online, with a database and logins?
0dev.to WebDev
MetaGPT Builds, Sapper Attacks, Trifecta Fortress Defends - The Complete AI Swarm Triangle
0dev.to WebDev
Atelier Fusion Family Salon
0dev.to WebDev
How to add an llms.txt file to WordPress (and what it will and won't do)
0dev.to WebDev