Optimal Transport enhances machine learning with new methods.
problem Comparing and manipulating probability distributions in machine learning.
method Probabilistic framework rooted in rich history and theory.
result New solutions in generative modeling and transfer learning.
A novel Federated Learning scheme using Optimal Transport for personalized model training.
problem Training models with data from clients having non-identically distributed data.
method Personalized Federated Learning scheme based on Optimal Transport (FedOT).
result FedOT scheme effectively transfers data from multiple distributions to a common domain and optimizes the prediction model.
We explore the use of deep learning and deep reinforcement learning for optimization problems in transportation. Many transportation system analysis tasks are formulated as an optimization problem - such as optimal control problems in intelligent transportation systems and long term urban planning. Often transportation…
In this paper, we present a novel and principled approach to learn the optimal transport between two distributions, from samples. Guided by the optimal transport theory, we learn the optimal Kantorovich potential which induces the optimal transport map. This involves learning two convex functions, by solving a novel mi…
Bayesian approach to optimal transport with stochastic costs.
problem Inferring optimal transport plans with uncertain costs.
method Bayesian framework and Hamiltonian Monte Carlo (HMC) sampling.
result Inference of optimal transport plans under stochastic cost functions.
Survey of deep RL in intelligent transportation systems.
problem Optimizing traffic signals and autonomous driving using deep RL.
method Comprehensive review of deep RL applications in traffic control and autonomous driving.
result Summarizes existing works in deep RL-based transportation applications.
New framework formalizes estimating valid transport maps, revealing their statistical limits.
problem Estimating valid transport maps in generative modeling.
method Formalized a minimax framework for estimating valid transport maps.
result Estimating any valid transport map is as hard as estimating the optimal transport map under standard stability assumptions.
Deep learning improves trip prediction accuracy in transportation planning.
problem Traditional models fail to accurately predict person and vehicle trips due to complexity and dynamics.
method Developed and trained a deep learning model using NHTS data.
result Deep learning model achieved 98% accuracy for person trip prediction and 96% for vehicle trip estimation.
New method uses nested optimal transport for financial time series evaluation.
problem Lack of consensus metric for evaluating generative models in finance.
method Nested optimal transport distance for time-causal tasks, with a parallelizable algorithm.
result Substantial speedups and robustness to financial tasks.
Optimal transport learns Riemannian metrics for evolving probability measures.
problem Learning metrics for evolving probability measures on Riemannian manifolds.
method Neural parametrization of a metric tensor via optimal transport, alternating optimization scheme.
result Improved trajectory inference on scRNA and bird migration data.
New findings on optimal transport gradient for generative models, addressing numerical instabilities.
problem Numerical instabilities in training Wasserstein Generative Adversarial Networks (WGAN).
method Valid differentiation theorem for entropic regularized transport, semi-discrete gradient formulation, and optimization algorithm.
result Existence of optimal transport gradient for generative models under specified conditions.
Introduces a new divergence measure for optimal transport.
problem Optimal transport distances and information divergences.
method Infimal convolution formulation of proximal optimal transport divergence.
result Establishes connections to dynamic formulations and partial differential equations.
Study develops a new method for creating fair models.
problem Ensuring equal outcomes for different protected groups.
method Introduces a new group-fair constraint based on transport maps.
result Develops a novel algorithm FTM for training group-fair models.
We propose an optimal transport (OT) framework for generalized zero-shot learning (GZSL), seeking to distinguish samples for both seen and unseen classes, with the assist of auxiliary attributes. The discrepancy between features and attributes is minimized by solving an optimal transport problem. {Specifically, we buil…
New algorithm improves on existing methods for solving transport problems.
problem Finding a map to transport one distribution to another.
method Iterative Markovian Fitting (IMF) and Diffusion Schrödinger Bridge Matching (DSBM).
result DSBM significantly improves over previous SB numerics and recovers various transport methods.
Study proves convergence of subgradients for optimal transport-based objectives.
problem Ensuring statistical consistency and optimization stability in transport-based models.
method Proves graphical convergence of subdifferentials to the subdifferential of the population objective.
result Standard subgradient methods consistently approach stationary points of the population-level problem.
Neural framework for conditional OT maps learns from categorical and continuous variables.
problem Learning conditional optimal transport maps between complex distributions.
method Hypernetwork generates adaptive transport layer parameters based on conditioning variables.
result Our method outperforms simpler conditioning methods in comprehensive ablation studies.
NOT learns optimal transport plans, kernel costs improve performance.
problem NOT algorithm learns non-optimal plans with weak quadratic costs.
method Introduced kernel weak quadratic costs to improve NOT's performance.
result Kernel costs provide improved theoretical and practical guarantees.
This work clarifies different transport map constructions and their causal interpretations.
problem Identifying distinct transport map constructions and their equivalence.
method Comparative analysis of three transport map constructions: cyclically monotone, quantile-preserving, and triangular monotone.
result Conditions for equivalence of different transport map constructions.
The paper explores rectified flows and their relation to optimal transport.
problem Understanding the connection between rectified flows and optimal transport.
method Investigates invariance properties, explicit constructions, and analysis of rectified flows in various settings.
result Rectified flows, when gradient constrained, do not generally solve optimal transport problems.
This paper proposes a new method for conditional sampling using optimal transport.
problem Sampling conditional distributions in Bayesian inference and density estimation.
method Iterative block-triangular transport maps solving an optimal transport problem with a weighted L2 cost function.
result The proposed method extends the data-driven approach for conditional sampling.
The paper tackles efficient computation of optimal transport by approximating conjugates with amortized optimization.
problem Efficient computation of convex conjugates in optimal transport is challenging and limits the quality of transport maps.
method The approach combines amortized approximations of conjugates with a fine-tuning solver to improve transport map quality.
result The method significantly improves the quality of transport maps for the Wasserstein-2 benchmark and models many 2D couplings and flows.
OTAD uses optimal transport to create robust models against adversarial attacks.
problem Vulnerability of deep neural networks to adversarial perturbations.
method OTAD combines optimal transport and Lipschitz networks to create a robust model.
result OTAD outperforms other robust models on diverse datasets.
Optimal transport aligns source and target distributions for linear regression in 2D.
problem Domain adaptation for linear regression in 2D with limited target data.
method Combining K-means and optimal transport for estimating geometric transformations.
result Optimal transport recovers geometric transformations like rotations, translations, and homotheties.
Optimal transport simplifies machine learning by comparing probability measures.
problem Comparing and manipulating probability distributions in machine learning.
method Uses optimal transport to compare and manipulate probability distributions, combining statistical and geometric perspectives.
result Optimal transport provides a unified framework for various machine learning tasks.
A new CO-Optimal Transport method optimizes transport maps between samples and features.
problem Optimal transport's limitations when samples are on different spaces.
method COOT: a novel OT problem that optimizes transport maps between samples and features simultaneously.
result COOT leads to performance improvements over state-of-the-art methods in domain adaptation and co-clustering.
Optimal transport between Gaussian Mixture Models improves domain adaptation efficiency.
problem Adapting machine learning models to new data distributions with minimal access.
method Optimal transport between Gaussian Mixture Models (GMMs) for domain adaptation.
result Our methods are more efficient and scalable with sample size and dimensions.
Improved spatial distribution learning with Bayesian transport maps and parametric shrinkage.
problem Learning non-Gaussian spatial distributions with limited training data.
method Proposed ShrinkTM approach using Bayesian transport maps with parametric shrinkage.
result ShrinkTM outperforms existing BTM, especially with few training samples.
COT-GAN generates sequential data with a causal optimal transport approach.
problem Generating sequential data with temporal causality constraints.
method Adversarial training with Causal Optimal Transport (COT) and entropic penalization.
result COT-GAN effectively learns time-dependent data distributions and generates stable time series data.
A new dataset distance using optimal transport, agnostic of model and label sets.
problem Quantifying task similarity across datasets without model dependence.
method Optimal transport for model-agnostic dataset comparison.
result The new distance correlates with transfer learning difficulty across various datasets.
Optimal transport and neural networks improve trade modeling accuracy.
problem Capturing subtler drivers of trade beyond supply and demand.
method Employing optimal transport and deep neural networks to learn a time-dependent cost function from data.
result Consistently outperforms traditional gravity models in accuracy.
FedAVOT improves federated learning by aligning user distributions.
problem Partial client participation leads to biased and unstable updates in federated learning.
method Formulates aggregation as masked optimal transport to align availability and importance distributions.
result Achieves a standard O(1/√T) rate, independent of the number of participating users per round.
Develops a novel global pooling framework using optimal transport.
problem Sub-optimal performance in global pooling operations.
method Regularized Optimal Transport (ROT) for generalized global pooling.
result ROTP layers can improve performance in various machine learning scenarios.
SPOT uses optimal transport to select important prototypes.
problem Summarizing datasets for better understanding and decision making.
method Modeling prototype selection as a submodular optimization problem and using a greedy algorithm.
result Our approach efficiently selects prototypes with optimal transport that best represent the target dataset.
A simple baseline outperforms deep learning methods in transportation forecasting.
problem The importance of stationarity and recurrent patterns in transportation data.
method A naive baseline based on average weekly patterns and linear regression.
result The baseline method achieves comparable or better results than state-of-the-art deep learning approaches.
DPOT uses deep learning to compute optimal transport efficiently.
problem Computing optimal transport between continuous distributions from unpaired samples.
method DeepParticle methods for min-min optimization without network structure restrictions.
result Established weak convergence and error bounds between learned and optimal maps.
Adaptive optimal transport priors improve few-shot learning robustness.
problem Limited supervision and distribution shifts in few-shot learning.
method Prototype-Guided Distributionally Robust Optimization (PG-DRO) framework.
result PG-DRO achieves stronger robust generalization in few-shot scenarios.
Optimal Transport has recently gained interest in machine learning for applications ranging from domain adaptation, sentence similarities to deep learning. Yet, its ability to capture frequently occurring structure beyond the "ground metric" is limited. In this work, we develop a nonlinear generalization of (discrete) …
New control methods improve dynamic measure transport paths.
problem Improving paths for dynamic measure transport.
method Connecting mean-field games to optimization problems for learning paths, advocating for smoothness of velocities.
result Our method recovers more efficient and smooth transport models compared to untilted paths.
Proposes a method to correct for covariate shift in meta-analysis of randomized trials.
problem Invalidation of standard IPD meta-analysis due to covariate shift across studies.
method Placebo-anchored transport framework that treats source-trial outcomes as proxy signals and target-trial placebo outcomes as gold labels.
result Yields target-identified effect estimates in connected targets and a principled screen--then--transport procedure in disconnected targets.
Framework tests group fairness in machine learning models.
problem Detecting biases in machine learning classifiers.
method Optimal transport projections to audit group fairness.
result Statistical test for various fairness notions efficiently computed.
Unified deep learning framework solves various optimal transport problems.
problem Solving variational problems in optimal transport with computational challenges.
method Unified deep learning framework leveraging dual formulation of Lagrangians.
result Outperforms previous approaches in single-cell trajectory inference.
A new method computes high-dimensional optimal transport using flow neural networks.
problem Computing optimal transport for high-dimensional data.
method Optimizing a flow model to minimize transport cost between two arbitrary distributions.
result Trained optimal transport flow enables downstream tasks like DRE and domain adaptation.
A new framework improves ASR alignment accuracy via optimal transport.
problem Peaky behavior and alignment inaccuracies in E2E ASR models.
method Differentiable alignment framework based on 1D optimal transport.
result Significant improvement in ASR alignment performance compared to CTC.
This paper explores how entropic regularization improves Wasserstein estimators' performance.
problem Improving the approximation and estimation properties of Wasserstein estimators.
method Entropic regularization of optimal transport costs to smooth Wasserstein estimators.
result Entropic regularization can achieve comparable statistical performance to un-regularized estimators at lower computational cost.
pyLOT library simplifies machine learning on 3D point clouds via linearized optimal transport.
problem Performing machine learning tasks on 3D point clouds.
method Linearized optimal transport (LOT) to embed distributions into Hilbert space, enabling linear machine learning.
result Downstream tasks on embedded representations are simplified to linear operations.
Optimal Transport (OT) naturally arises in many machine learning applications, yet the heavy computational burden limits its wide-spread uses. To address the scalability issue, we propose an implicit generative learning-based framework called SPOT (Scalable Push-forward of Optimal Transport). Specifically, we approxima…
New model generates data without mode collapse or mixture.
problem Mode collapse and mode mixture in generative models.
method Inspired by AE-OT, improved sample generation algorithm.
result Free from mode collapse and mixture issues.