Scalable Nearest Neighbor Search for Optimal Transport
Not on the map. Why?
Abstract (arXiv)
The Optimal Transport (a.k.a. Wasserstein) distance is an increasingly popular similarity measure for rich data domains, such as images or text documents. This raises the necessity for fast nearest neighbor search algorithms according to this distance, which poses a substantial computational bottleneck on massive datasets. In this work we introduce Flowtree, a fast and accurate approximation algorithm for the Wasserstein-1 distance. We formally analyze its approximation factor and running time. We perform extensive experimental evaluation of nearest neighbor search algorithms in the W1 distance on real-world dataset. Our results show that compared to previous state of the art, Flowtree achieves up to 7.4 times faster running time.
Key facts. Cited by 25 papers in the map (data through 14 Oct 2026). First posted on arXiv 9 Oct 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 11; 2022: 3; 2023: 4; 2024: 6; 2025: 0; 2026: 1.
25 papers.
Sort:
- 2021Multi-Vector Models with Textual Guidance for Fine-Grained Scientific Document Similaritycited by 35
- 2021Breaking the Linear Iteration Cost Barrier for Some Well-known Conditional Gradient Methods Using MaxIP Data-structurescited by 19
- 2021Diffusion Earth Mover's Distance and Distribution Embeddingscited by 18
- 2023Editable User Profiles for Controllable Text Recommendationcited by 16
- 2021Sentence Similarity Based on Contextscited by 12
- 2021Scalable Optimal Transport in High Dimensions for Graph Distances, Embedding Alignment, and Morecited by 11
- 2024Graph Edit Distance with General Costs Using Neural Set Divergencecited by 11
- 2021Supervised Tree-Wasserstein Distancecited by 10
- 2021Re-evaluating Word Mover's Distancecited by 10
- 2021Fixed Support Tree-Sliced Wasserstein Barycentercited by 10
- 2022Approximating 1-Wasserstein Distance with Treescited by 9
- 2024A New Robust Partial p-Wasserstein-Based Metric for Comparing Distributionscited by 7
13 more
- 2023Fast Optimal Transport through Sliced Wasserstein Generalized Geodesicscited by 5
- 2024Tree-Wasserstein Distance for High Dimensional Data with a Latent Feature Hierarchycited by 5
- 2021Landmarks Augmentation with Manifold-Barycentric Oversamplingcited by 4
- 2021Embedding Signals on Knowledge Graphs with Unbalanced Diffusion Earth Mover's Distancecited by 4
- 2023Neural approximation of Wasserstein distance via a universal architecture for symmetric and factorwise group invariant functionscited by 4
- 2022Learning Ultrametric Trees for Optimal Transport Regressioncited by 3
- 2024Efficient approximation of Earth Mover's Distance Based on Nearest Neighbor Searchcited by 3
- 2024Learning Structured Representations by Embedding Class Hierarchy with Fast Optimal Transportcited by 3
- 2022Budget-Constrained Bounds for Mini-Batch Estimation of Optimal Transportcited by 1
- 2021MURAL: An Unsupervised Random Forest-Based Embedding for Electronic Health Record Datacited by 0
- 2023An Empirical Study of Self-supervised Learning with Wasserstein Distancecited by 0
- 2024A Class of Topological Pseudodistances for Fast Comparison of Persistence Diagramscited by 0
- 2026Sierpi'nski--Knopp Wasserstein Distance for Persistence Diagrams and Applications to 2-Wasserstein Approximationcited by 0
- 2026Sierpi'nski--Knopp Wasserstein Distance for Persistence Diagrams and Applications to 2-Wasserstein Approximation1 Sep 2026
- 2024Tree-Wasserstein Distance for High Dimensional Data with a Latent Feature Hierarchy28 Oct 2024
- 2024Learning Structured Representations by Embedding Class Hierarchy with Fast Optimal Transport4 Oct 2024
- 2024Graph Edit Distance with General Costs Using Neural Set Divergence26 Sep 2024
- 2024A New Robust Partial p-Wasserstein-Based Metric for Comparing Distributions6 May 2024
- 2024A Class of Topological Pseudodistances for Fast Comparison of Persistence Diagrams22 Feb 2024
- 2024Efficient approximation of Earth Mover's Distance Based on Nearest Neighbor Search14 Jan 2024
- 2023An Empirical Study of Self-supervised Learning with Wasserstein Distance16 Oct 2023
- 2023Neural approximation of Wasserstein distance via a universal architecture for symmetric and factorwise group invariant functions1 Aug 2023
- 2023Fast Optimal Transport through Sliced Wasserstein Generalized Geodesics4 Jul 2023
- 2023Editable User Profiles for Controllable Text Recommendation9 Apr 2023
- 2022Budget-Constrained Bounds for Mini-Batch Estimation of Optimal Transport24 Oct 2022
13 more
- 2022Learning Ultrametric Trees for Optimal Transport Regression21 Oct 2022
- 2022Approximating 1-Wasserstein Distance with Trees24 Jun 2022
- 2021Breaking the Linear Iteration Cost Barrier for Some Well-known Conditional Gradient Methods Using MaxIP Data-structures30 Nov 2021
- 2021MURAL: An Unsupervised Random Forest-Based Embedding for Electronic Health Record Data19 Nov 2021
- 2021Multi-Vector Models with Textual Guidance for Fine-Grained Scientific Document Similarity16 Nov 2021
- 2021Fixed Support Tree-Sliced Wasserstein Barycenter8 Sep 2021
- 2021Embedding Signals on Knowledge Graphs with Unbalanced Diffusion Earth Mover's Distance26 Jul 2021
- 2021Scalable Optimal Transport in High Dimensions for Graph Distances, Embedding Alignment, and More14 Jul 2021
- 2021Re-evaluating Word Mover's Distance30 May 2021
- 2021Sentence Similarity Based on Contexts17 May 2021
- 2021Landmarks Augmentation with Manifold-Barycentric Oversampling2 Apr 2021
- 2021Diffusion Earth Mover's Distance and Distribution Embeddings25 Feb 2021
- 2021Supervised Tree-Wasserstein Distance27 Jan 2021
- 2021Multi-Vector Models with Textual Guidance for Fine-Grained Scientific Document Similaritycited by 35
- 2021Breaking the Linear Iteration Cost Barrier for Some Well-known Conditional Gradient Methods Using MaxIP Data-structurescited by 19
- 2021Diffusion Earth Mover's Distance and Distribution Embeddingscited by 18
- 2023Editable User Profiles for Controllable Text Recommendationcited by 16
- 2021Sentence Similarity Based on Contextscited by 12
- 2021Scalable Optimal Transport in High Dimensions for Graph Distances, Embedding Alignment, and Morecited by 11
- 2024Graph Edit Distance with General Costs Using Neural Set Divergencecited by 11
- 2021Supervised Tree-Wasserstein Distancecited by 10
- 2021Re-evaluating Word Mover's Distancecited by 10
- 2021Fixed Support Tree-Sliced Wasserstein Barycentercited by 10
- 2022Approximating 1-Wasserstein Distance with Treescited by 9
- 2024A New Robust Partial p-Wasserstein-Based Metric for Comparing Distributionscited by 7
12 most cited of 25 All 25 (JSON)
- 2026Sierpi'nski--Knopp Wasserstein Distance for Persistence Diagrams and Applications to 2-Wasserstein Approximation1 Sep 2026
- 2024Tree-Wasserstein Distance for High Dimensional Data with a Latent Feature Hierarchy28 Oct 2024
- 2024Learning Structured Representations by Embedding Class Hierarchy with Fast Optimal Transport4 Oct 2024
- 2024Graph Edit Distance with General Costs Using Neural Set Divergence26 Sep 2024
- 2024A New Robust Partial p-Wasserstein-Based Metric for Comparing Distributions6 May 2024
- 2024A Class of Topological Pseudodistances for Fast Comparison of Persistence Diagrams22 Feb 2024
- 2024Efficient approximation of Earth Mover's Distance Based on Nearest Neighbor Search14 Jan 2024
- 2023An Empirical Study of Self-supervised Learning with Wasserstein Distance16 Oct 2023
- 2023Neural approximation of Wasserstein distance via a universal architecture for symmetric and factorwise group invariant functions1 Aug 2023
- 2023Fast Optimal Transport through Sliced Wasserstein Generalized Geodesics4 Jul 2023
- 2023Editable User Profiles for Controllable Text Recommendation9 Apr 2023
- 2022Budget-Constrained Bounds for Mini-Batch Estimation of Optimal Transport24 Oct 2022
12 newest of 25 All 25 (JSON)