3 papers
cs.LG2026
LOTFormer: Doubly-Stochastic Linear Attention via Low-Rank Optimal Transport
Ashkan Shahbazi, Chayne Thrash, Yikun Bai +3
Transformers have proven highly effective across modalities, but standard softmax attention scales quadratically with sequence length, limiting long context modeling. Linear attent…
stat.ML2025
Recovering Wasserstein Distance Matrices from Few Measurements
Muhammad Rana, Abiy Tasissa, HanQin Cai +2
This paper proposes two algorithms for estimating square Wasserstein distance matrices from a small number of entries. These matrices are used to compute manifold learning embeddin…
stat.ML2025
Neighbor Embeddings Using Unbalanced Optimal Transport Metrics
Muhammad Rana, Keaton Hamm
This paper proposes the use of the Hellinger--Kantorovich metric from unbalanced optimal transport (UOT) in a dimensionality reduction and learning (supervised and unsupervised) pi…