ITSPACE: Monotone Gaussian Optimal Transport Updates
Not on the map. Why?
Abstract (arXiv)
Covariance matrices serve as compact descriptors of feature distributions in many machine-learning pipelines, including domain adaptation and Gaussian embeddings. Under a centered Gaussian approximation, the unregularized Wasserstein-2 optimal-transport (OT) discrepancy admits a closed form on covariances given by the Bures-Wasserstein (BW) objective on the symmetric positive definite (SPD) cone. We propose ITSPACE (Iterative Transport for Stable Proximal Alignment of Covariance Embeddings), a proximal majorization-minimization method that directly optimizes this exact BW objective through closed-form updates in a square-root factorization. In exact arithmetic, each iteration satisfies a sufficient-decrease inequality for the BW objective; under inexact polar computations, we provide an explicit certificate-gap bound controlling deviations from exact descent. The resulting iterations preserve PSD structure by construction and naturally support rank-restricted factors, making ITSPACE well-suited as a lightweight inner-loop primitive in settings where adaptation must be performed from unlabeled target batches under strict step and compute budgets. Across real-world covariance-alignment benchmarks, ITSPACE reaches low-BW-gap solutions substantially faster than BW-gradient descent, methods based on other covariance geometries, and entropically regularized sample-OT baselines.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 11 of its 25 references that are in the map. First posted on arXiv 29 Jun 2026.
Builds on
11 of 25 references.
References · 11
- 2013Sinkhorn Distances: Lightspeed Computation of Optimal Transportation Distancescited by 2,316
- 2020WILDS: A Benchmark of in-the-Wild Distribution Shiftscited by 1,375
- 2016Deep CORAL: Correlation Alignment for Deep Domain Adaptationcited by 1,200
- 2018Recognition in Terra Incognitacited by 658
- 2017VisDA: The Visual Domain Adaptation Challengecited by 435
- 2016Stochastic Optimization for Large-scale Optimal Transportcited by 189
- 2022Variational inference via Wasserstein gradient flowscited by 85
- 2023Forward-backward Gaussian variational inference via JKO in the Bures-Wasserstein Spacecited by 36
- 2021On Riemannian Optimization over Positive Definite Matrices with the Bures-Wasserstein Geometrycited by 32
- 2024Wasserstein Flow Matching: Generative modeling over families of distributionscited by 28
- 2025Learning to Normalize on the SPD Manifold under Bures-Wasserstein Geometrycited by 7
Built on by
Not cited yet.