Represent the quadratic OT potential with a strongly convex input-convex neural network and train it by matching the distribution of its gradient pushforward to the target distribution in a weak dual metric. Median-center the potential on every minibatch so that optimization does not waste capacity or suffer instability from the additive constant ambiguity. The paper's stability inequality predicts that this can produce a more stable potential estimate than directly optimizing a transport-map…
Select the coordinates of a sparse adapter or sparse fine-tuning mask using both Fisher width and inverse-Fisher width. The mask should avoid parameter subsets that are cheap in the Fisher geometry but extremely large in the inverse-Fisher geometry, or vice versa, thereby controlling both prediction sensitivity and estimator-like uncertainty.
Add a distribution-level regularizer that compares augmented second-moment matrices of neural activations using the affine-invariant Riemannian metric on SPD matrices. This aligns means, variances, and selected nonlinear moments while remaining invariant to invertible linear reparameterizations of feature coordinates.