Wasserstein Wormhole: Scalable Optimal Transport Distance with Transformers

Doron Haviv, Russell Zhang Kunes, Thomas Dougherty +4

Optimal transport (OT) and the related Wasserstein metric (W) are powerful and ubiquitous tools for comparing distributions. However, computing pairwise Wasserstein distances rapid…