Math: Optimal transport

Machine-learning ideas tagged Optimal transport in the Math taxonomy of the Math2NN corpus.

✗ Failed on benchmark 2026

Wasserstein-Controlled Gaussian-Mixture Rollouts

Replace single-Gaussian uncertainty propagation in a neural state-space or world model with a finite mixture of Gaussian latent states. Each component is propagated through the learned nonlinear dynamics, and components are merged or pruned only when their Wasserstein discrepancy is below a prescribed tolerance, preserving multimodal futures while keeping computation bounded.

Useful8/10
Difficulty6/10
Novelty7/10
Paper: Stochastic Nonlinear Model Predictive Control with Gaussian Mixture Uncertainty Propagation arXiv:2608.29272
△ Mechanism confirmed, baseline not beaten 2026

Intrinsic Schrödinger Bridge Diffusion

Replace an unconstrained Euclidean diffusion sampler plus projection with a controlled diffusion whose state always lies on an embedded manifold \(\mathcal M\). The neural controller predicts a tangent vector, while the stochastic forcing is also tangent; this preserves constraints during every intermediate denoising step and avoids the bias caused by repeatedly projecting off-manifold states.

Useful8/10
Difficulty6/10
Novelty7/10
Paper: Hard-Constrained Sampling on Embedded Riemannian Manifolds via Adjoint Schrödinger Bridges arXiv:2608.25838
✗ Mechanism failed 2026

Reference-Preserving Martingale Layer

Replace an unconstrained stochastic transition between categorical or discretized latent distributions by a transition matrix that preserves a prescribed reference distribution while mapping relative populations through a martingale. This prevents the layer from inventing arbitrarily sharp deviations from the reference and imposes a convex-order monotonicity condition on uncertainty across layers or diffusion time steps.

Useful8/10
Difficulty6/10
Novelty7/10
Paper: State convertibility and fluctuation theorems from a dynamical reference: majorization meets martingales arXiv:2608.19391
✗ Failed on benchmark 2026

Entropy-Calibrated Robust Bellman Backup

Replace a fixed robust-RL ambiguity radius with a radius computed from the agent’s current belief over environment models. High posterior entropy enlarges the Wasserstein uncertainty set and suppresses catastrophic actions; posterior concentration automatically reduces conservatism and approaches ordinary expected-reward planning.

Useful8/10
Difficulty6/10
Novelty6/10
Paper: Quantifying Risk Under Evolving Uncertainty: Belief-Dependent Robustness for Safe Sequential Decision Making arXiv:2608.17574
△ Mechanism confirmed, baseline not beaten 2026

Finite-Batch OT Reflow for Straighter Flow Matching

Replace fixed random source-target pairings in flow matching by an outer loop that repeatedly solves exact OT assignments inside minibatches, trains the velocity field on the resulting pairings, and regenerates pairings from the learned flow. The mathematical guarantee is not global OT optimality: for batch size N, any limiting coupling is N-cyclically monotone and the squared endpoint cost cannot increase through the alternating updates. This should produce straighter trajectories and permit…

Useful8/10
Difficulty5/10
Novelty5/10
Paper: Limit Points of Reflow with Minibatch Optimal Transport arXiv:2608.07042
✗ Mechanism failed 2026

Multi-source conditional OT adversarial training

Replace ordinary empirical-risk minimization on pooled heterogeneous data with worst-case conditional risk over joint distributions that remain close to every source under an optimal-transport budget. The adversary transports source context-label pairs toward high-loss, target-event-like examples, while source-specific radii prevent arbitrary shifts. This should improve performance on rare target contexts and unseen domains without requiring abundant target labels.

Useful8/10
Difficulty6/10
Novelty5/10
Paper: Harnessing Heterogeneous Data for Conditional Optimization via Optimal Transport arXiv:2607.19761
△ Mechanism confirmed, baseline not beaten 2026

Monotone transport-map latent space

Represent every nonnegative equal-mass one-dimensional state by its CDT quantile map relative to a fixed reference density, then train the neural dynamics model in this transformed space rather than on Eulerian grid values. The latent manifold for translations and transport-dominated evolution is substantially flatter: linear transport lies in the span of the initial transformed state and the constant function, while nonlinear conservative dynamics have algebraic approximation error bounds.

Useful8/10
Difficulty5/10
Novelty6/10
Paper: Reduced Order Modeling of One-Dimensional Conservative PDEs via the Cumulative Distribution Transform arXiv:2607.17066
✗ Failed on benchmark 2026

Distribution-Aware Contraction Scheduler

Estimate the local contraction rate along minibatch couplings of neural ODE or flow-matching trajectories instead of using one global Lipschitz lower bound. Use the resulting displacement-weighted rate to trigger adaptive solver tolerances, training-time regularization, or early stopping when the transported distributions have entered a strongly contracting region.

Useful8/10
Difficulty5/10
Novelty7/10
Paper: Wasserstein Stability of Contracting Flows: Effective Rates, Euler Self-Correction, and Noise Tightening arXiv:2607.14291
△ Mechanism confirmed, baseline not beaten 2026

Entropic Wasserstein adversarial augmentation

Replace the nonsmooth Wasserstein inner supremum in robust training by the paper's entropic log-expectation, evaluated with Gaussian perturbation samples. The resulting loss continuously interpolates between ordinary averaging and soft worst-case selection, producing differentiable adversarial augmentation without an inner PGD loop.

Useful8/10
Difficulty4/10
Novelty6/10
Paper: First-Order Methods for Distributionally Robust Constrained Optimization arXiv:2607.11460
✓ Mechanism works 2026

Transport-Signature Density Decoder

Replace a pixelwise density decoder with a decoder that predicts coefficients of a transport potential relative to a fixed reference density. The reconstructed density is the pushforward of the reference measure through a differentiable transport map, so positivity and total mass are structural properties rather than learned penalties.

Useful8/10
Difficulty6/10
Novelty6/10
Paper: Structure-Preserving Reduced-Order Modeling via Low-Rank Transport Signatures arXiv:2607.01696
✗ Mechanism failed 2026

Barycentric One-to-Many Alignment

Replace pointwise cross-modal or prototype-to-instance matching with a loss that compares source pairwise inner products to inner products between target conditional means. A source prototype can align to a cloud of target instances while preserving its semantic barycenter, instead of being forced to match every target instance individually.

Useful8/10
Difficulty5/10
Novelty6/10
Paper: Barycentric Weak Inner-Product Gromov-Wasserstein arXiv:2608.25145
✗ Mechanism failed 2026

Projected Bures Covariance Pooling

Replace Euclidean or unprojected covariance averaging with a projected Bures-Wasserstein barycenter layer. Each unit-step barycenter update is followed by eigenvalue clipping into \([\alpha,\beta]\), preserving positive definiteness and preventing ill-conditioning without an additional eigendecomposition.

Useful7/10
Difficulty5/10
Novelty6/10
Paper: Projected Riemannian Gradient Descent for the Bures-Wasserstein Barycenter: Dimension-Independent Linear Convergence at Unit Step Size arXiv:2609.03762
✗ Failed on benchmark 2026

Gauge-Free Inverse OT Attention

Replace ordinary row-softmax attention with a doubly stochastic Sinkhorn attention plan W, and periodically recover a gauge-fixed pairwise cost from W using the exact double-centering identity. Use this recovered cost to initialize or regularize a structured attention score, making the attention geometry identifiable despite arbitrary query and key row and column offsets.

Useful7/10
Difficulty5/10
Novelty7/10
Paper: An exact and fast solution of the inverse Regularized Optimal Transport problem arXiv:2609.01278
✗ Failed on benchmark 2026

Wasserstein Speed-Limit Controller

Wrap stochastic optimization or iterative neural inference in a controller that measures how far the state distribution moves during each interval and compares this motion with the available noise-dependent entropy-production budget. The controller increases the learning rate or reduces inference steps only while the trajectory remains inside the predicted speed-limit region, preventing fast jumps that cause accuracy collapse.

Useful7/10
Difficulty5/10
Novelty7/10
Paper: The thermodynamic freedom of a thermodynamic computer arXiv:2608.27938
△ Mechanism confirmed, baseline not beaten 2026

Dimension-Free Brenier Transport Layer

Build a neural transport layer by parameterizing a convex potential whose gradient maps a semi-log-concave latent distribution into a compact convex data domain. Use the paper's dimension-free Lipschitz certificate to set the layer's Jacobian scale, initialize the potential, and reject or regularize parameter updates that create excessive curvature. The goal is a bounded-output transport module that is less sensitive to latent dimension than diameter-based spectral heuristics.

Useful7/10
Difficulty6/10
Novelty7/10
Paper: Dimension-Free Lipschitz Bounds for Brenier Maps to Compactly Supported Log-Concave Targets arXiv:2608.15906
△ Mechanism confirmed, baseline not beaten 2026

OT Primitive Universal Flow

Parameterize a generative or density-evolving model as a composition of diffeomorphic optimal-mass-transport maps rather than unconstrained residual layers. Each layer transports one smooth positive density to another through a learned squared-distance OT map, while compositions provide a principled universal family for transformations connected to the identity.

Useful7/10
Difficulty7/10
Novelty6/10
Paper: The Holonomy of Optimal Mass Transport: The Smooth Case arXiv:2608.15585
✓✓ Beats tuned baseline 2026

Latent-Component Schrödinger Bridge

Represent both endpoint distributions as Gaussian mixtures and explicitly transport their component labels along with continuous states. Use an entropic coupling between source and target components, then run a separate Gaussian bridge for every selected component pair, with covariance inflation preventing unstable Riccati or Cholesky computations.

Useful7/10
Difficulty6/10
Novelty6/10
Paper: On Bridging Mixture Distributions arXiv:2608.13383
✗ Failed on benchmark 2026

Gauge-Free Spectral OT Layer

Parameterize an entropic OT cost only in directions that can change the transport plan, removing row-plus-column potential directions that are invisible because of OT gauge invariance. Whiten the remaining feature coordinates using their empirical covariance, producing an OT layer whose identifiable parameters have substantially more uniform sensitivity.

Useful7/10
Difficulty5/10
Novelty7/10
Paper: Sinkhorn Linearization and the Spectral Proxy: Unifying the Statistical and Algorithmic Theory of Feature-Parameterized Inverse Optimal Transport via a Single Spectral Sandwich arXiv:2608.13201
✓✓ Beats tuned baseline 2026

Displacement-Huber distribution pooling

Replace ordinary Wasserstein or arithmetic pooling of distribution-valued features with a barycenter whose individual quantile displacements are Huberized. Small changes between input distributions remain averaged quadratically, while a corrupted token, expert, graph neighborhood, or augmentation cannot move the pooled distribution arbitrarily far. The module is especially cheap for one-dimensional distributions represented by fixed quantile vectors.

Useful7/10
Difficulty4/10
Novelty7/10
Paper: Huber-Wasserstein barycenters for robust distribution-valued data arXiv:2608.13131
✗ Mechanism failed 2026

Coarse-to-fine active-support transport attention

Replace dense cross-attention weights with a balanced transport plan whose nonzero query-key edges are maintained by a multiscale active-set procedure. Solve the coarse token-group problem first, lift its support to the fine token grid, add only edges indicated by local cost or marginal residuals, and warm-start the fine problem from the lifted plan. This should provide a principled sparse attention pattern rather than fixing a global top-k pattern before seeing the transport solution.

Useful7/10
Difficulty7/10
Novelty6/10
Paper: A Multiscale Primal-Dual Interior-Point Relaxation Method for Large-Scale Optimal Transport Problems arXiv:2608.12060
△ Mechanism confirmed, baseline not beaten 2026

Submetry-Lifted Relational Alignment

Represent a graph, set, or attributed network as a measurable Z-valued kernel and train on lifted representatives while explicitly minimizing over node couplings. The quotient objective is invariant to relabeling by construction, while the lifted loss gives a dense correspondence signal that can stabilize graph attention and relational encoders.

Useful7/10
Difficulty5/10
Novelty5/10
Paper: Metric Geometry of Lebesgue, Wasserstein, and Gromov-Wasserstein Spaces: Submetries, Curvature, and Geodesics arXiv:2608.11680
✗ Failed on benchmark 2026

Differentiable Persistence Landscape Layer

Convert each persistence diagram produced from an input, intermediate feature map, or graph filtration into a discretized persistence landscape and feed it to an MLP or concatenate it with ordinary neural features. Unlike a variable-size list of birth-death pairs, the landscape has a fixed tensor shape and is provably nonexpansive with respect to the diagram Wasserstein distance.

Useful7/10
Difficulty4/10
Novelty5/10
Paper: A Hilbert space embedding of persistence diagrams and barcodes arXiv:2608.08858
△ Mechanism confirmed, baseline not beaten 2026

Exact doubly stochastic low-rank attention

Replace an n-by-n attention or token-mixing matrix with two nonnegative rank-r factors having row-simplex constraints and a shared latent column marginal. The induced matrix is exactly doubly stochastic at every accepted update, while applying it to values uses two thin matrix multiplications and never constructs the dense attention matrix.

Useful7/10
Difficulty6/10
Novelty7/10
Paper: Exact Rank-Space KL Projection for Shared-Marginal Low-Rank Factors: Application to Doubly Stochastic Clustering arXiv:2608.08642
✓✓ Beats tuned baseline 2026

Entropy-Regularized Wasserstein Actor

Replace the usual parameter-space actor update with an action-space transport update. For every visited state, move sampled actions along a critic-improving velocity field while adding the entropy velocity, then fit the transported action cloud back to the actor's Gaussian mean and covariance. This preserves the paper's key idea that policy improvement is a Wasserstein flow over conditional action laws while remaining implementable for neural actors.

Useful7/10
Difficulty5/10
Novelty7/10
Paper: Wasserstein Policy Gradient for Entropy-Regularized Linear-Quadratic Control arXiv:2608.07433