Hierarchical Optimal Transport for Document Representation
Not on the map. Why?
Abstract (arXiv)
The ability to measure similarity between documents enables intelligent summarization and analysis of large corpora. Past distances between documents suffer from either an inability to incorporate semantic similarities between words or from scalability issues. As an alternative, we introduce hierarchical optimal transport as a meta-distance between documents, where documents are modeled as distributions over topics, which themselves are modeled as distributions over words. We then solve an optimal transport problem on the smaller topic space to compute a similarity score. We give conditions on the topics under which this construction defines a distance, and we relate it to the word mover's distance. We evaluate our technique for k-NN classification and show better interpretability and scalability with comparable performance to current methods at a fraction of the cost.
Key facts. Cited by 53 papers in the map (data through 7 Oct 2026). First posted on arXiv 26 Jun 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2020: 1; 2021: 13; 2022: 10; 2023: 7; 2024: 5; 2025: 8; 2026: 9.
53 papers.
Sort:
By area:
- 2023Recent Advances in Optimal Transport for Machine Learningcited by 66
- 2022Representing Mixtures of Word Embeddings with Mixtures of Topic Embeddingscited by 36
- 2021A Survey on Optimal Transport for Machine Learning: Theory and Applicationscited by 35
- 2021Multi-Vector Models with Textual Guidance for Fine-Grained Scientific Document Similaritycited by 35
- 2023Scalable Optimal Transport Methods in Machine Learning: A Contemporary Surveycited by 27
- 2022Wasserstein Task Embedding for Measuring Task Similaritiescited by 24
- 2021On Transportation of Mini-batches: A Hierarchical Approachcited by 15
- 2022Adaptive Distribution Calibration for Few-Shot Learning with Hierarchical Optimal Transportcited by 14
- 2021Sentence Similarity Based on Contextscited by 12
- 2021Supervised Tree-Wasserstein Distancecited by 10
- 2021Re-evaluating Word Mover's Distancecited by 10
- 2021Hierarchical Optimal Transport for Unsupervised Domain Adaptationcited by 10
38 more
- 2025Lightspeed Geometric Dataset Distance via Sliced Optimal Transportcited by 10
- 2022Hilbert Curve Projection Distance for Distribution Comparisoncited by 9
- 2022OTExtSum: Extractive Text Summarisation with Optimal Transportcited by 8
- 2025Hierarchical Refinement: Optimal Transport to Infinity and Beyondcited by 8
- 2022Riemannian block SPD coupling manifold and its application to optimal transportcited by 7
- 2024A New Robust Partial p-Wasserstein-Based Metric for Comparing Distributionscited by 7
- 2022Hierarchical Optimal Transport for Comparing Histopathology Datasetscited by 6
- 2025Enhancing Unsupervised Graph Few-shot Learning via Set Functions and Optimal Transportcited by 6
- 2021Unsupervised Ground Metric Learning using Wasserstein Singular Vectorscited by 5
- 2024Tree-Wasserstein Distance for High Dimensional Data with a Latent Feature Hierarchycited by 5
- 2025Slicing Wasserstein Over Wasserstein Via Functional Optimal Transportcited by 5
- 2021Order Constraints in Optimal Transportcited by 4
- 2024Explanatory Model Monitoring to Understand the Effects of Feature Shifts on Performancecited by 4
- 2025Efficient Transferable Optimal Transport via Min-Sliced Transport Planscited by 4
- 2020Nested-Wasserstein Self-Imitation Learning for Sequence Generationcited by 3
- 2022Modeling Hierarchical Structural Distance for Unsupervised Domain Adaptationcited by 3
- 2023Few-Shot Object Detection via Synthetic Features with Optimal Transportcited by 3
- 2021Interpretable contrastive word mover's embeddingcited by 2
- 2024What is the Right Notion of Distance between Predict-then-Optimize Tasks?cited by 2
- 2026Robust Generalization with Adaptive Optimal Transport Priors for Decision-Focused Learningcited by 2
- 2021Limit Distribution Theory for the Smooth 1-Wasserstein Distance with Applicationscited by 1
- 2022Budget-Constrained Bounds for Mini-Batch Estimation of Optimal Transportcited by 1
- 2023Keypoint-Guided Optimal Transport: Models, Algorithms, and Applicationscited by 1
- 2023Towards Generalising Neural Topical Representationscited by 1
- 2023TaskDiff: A Similarity Metric for Task-Oriented Conversationscited by 1
- 2024SAVA: Scalable Learning-Agnostic Data Valuationcited by 1
- 2025Scalable Sobolev IPM for Probability Measures on a Graphcited by 1
- 2026OT on the Map: Quantifying Domain Shifts in Geographic Spacecited by 1
- 2021Hawkes Processes on Graphonscited by 0
- 2021Efficient Optimal Transport Algorithm by Accelerated Gradient descentcited by 0
- 2022Coherence-Based Distributed Document Representation Learning for Scientific Documentscited by 0
- 2023A Novel Dual of Shannon Information and Weighting Schemecited by 0
- 2025How Vital is the Jurisprudential Relevance: Law Article Intervened Legal Case Retrieval and Matchingcited by 0
- 2025Generalized Sobolev IPM for Graph-Based Measurescited by 0
- 2026HOT-POT: Optimal Transport for Sparse Stereo Matchingcited by 0
- 2026Grammatical Error Correction Evaluation by Optimally Transporting Edit Representationcited by 0
- 2026Calibrating Tabular Anomaly Detection via Optimal Transportcited by 0
- 2026Online Continual Learning with Dynamic Label Hierarchiescited by 0
50 most cited of 53 All 53 (JSON)
- 2026Coupled Usage-Sense Processes: Temporal and Attributable Lexical Semantic Change25 Sep 2026
- 2026Fidelity-Diversity Metrics for Text6 Jul 2026
- 2026RADAR: Relative Angular Divergence Across Representations21 May 2026
- 2026Online Continual Learning with Dynamic Label Hierarchies12 May 2026
- 2026OT on the Map: Quantifying Domain Shifts in Geographic Space17 Apr 2026
- 2026Calibrating Tabular Anomaly Detection via Optimal Transport6 Feb 2026
- 2026Grammatical Error Correction Evaluation by Optimally Transporting Edit Representation5 Feb 2026
- 2026Robust Generalization with Adaptive Optimal Transport Priors for Decision-Focused Learning1 Feb 2026
- 2026HOT-POT: Optimal Transport for Sparse Stereo Matching18 Jan 2026
- 2025Efficient Transferable Optimal Transport via Min-Sliced Transport Plans24 Nov 2025
- 2025Generalized Sobolev IPM for Graph-Based Measures29 Oct 2025
- 2025Slicing Wasserstein Over Wasserstein Via Functional Optimal Transport26 Sep 2025
38 more
- 2025Hierarchical Refinement: Optimal Transport to Infinity and Beyond4 Mar 2025
- 2025How Vital is the Jurisprudential Relevance: Law Article Intervened Legal Case Retrieval and Matching25 Feb 2025
- 2025Scalable Sobolev IPM for Probability Measures on a Graph2 Feb 2025
- 2025Lightspeed Geometric Dataset Distance via Sliced Optimal Transport31 Jan 2025
- 2025Enhancing Unsupervised Graph Few-shot Learning via Set Functions and Optimal Transport10 Jan 2025
- 2024Tree-Wasserstein Distance for High Dimensional Data with a Latent Feature Hierarchy28 Oct 2024
- 2024What is the Right Notion of Distance between Predict-then-Optimize Tasks?11 Sep 2024
- 2024Explanatory Model Monitoring to Understand the Effects of Feature Shifts on Performance24 Aug 2024
- 2024SAVA: Scalable Learning-Agnostic Data Valuation3 Jun 2024
- 2024A New Robust Partial p-Wasserstein-Based Metric for Comparing Distributions6 May 2024
- 2023TaskDiff: A Similarity Metric for Task-Oriented Conversations23 Oct 2023
- 2023Few-Shot Object Detection via Synthetic Features with Optimal Transport29 Aug 2023
- 2023Towards Generalising Neural Topical Representations24 Jul 2023
- 2023Recent Advances in Optimal Transport for Machine Learning28 Jun 2023
- 2023Scalable Optimal Transport Methods in Machine Learning: A Contemporary Survey8 May 2023
- 2023A Novel Dual of Shannon Information and Weighting Scheme25 Apr 2023
- 2023Keypoint-Guided Optimal Transport: Models, Algorithms, and Applications23 Mar 2023
- 2022Modeling Hierarchical Structural Distance for Unsupervised Domain Adaptation21 Nov 2022
- 2022Budget-Constrained Bounds for Mini-Batch Estimation of Optimal Transport24 Oct 2022
- 2022Adaptive Distribution Calibration for Few-Shot Learning with Hierarchical Optimal Transport9 Oct 2022
- 2022Wasserstein Task Embedding for Measuring Task Similarities24 Aug 2022
- 2022Hilbert Curve Projection Distance for Distribution Comparison30 May 2022
- 2022OTExtSum: Extractive Text Summarisation with Optimal Transport21 Apr 2022
- 2022Hierarchical Optimal Transport for Comparing Histopathology Datasets18 Apr 2022
- 2022Representing Mixtures of Word Embeddings with Mixtures of Topic Embeddings3 Mar 2022
- 2022Riemannian block SPD coupling manifold and its application to optimal transport30 Jan 2022
- 2022Coherence-Based Distributed Document Representation Learning for Scientific Documents8 Jan 2022
- 2021Hierarchical Optimal Transport for Unsupervised Domain Adaptation3 Dec 2021
- 2021Multi-Vector Models with Textual Guidance for Fine-Grained Scientific Document Similarity16 Nov 2021
- 2021Interpretable contrastive word mover's embedding1 Nov 2021
- 2021Order Constraints in Optimal Transport14 Oct 2021
- 2021Limit Distribution Theory for the Smooth 1-Wasserstein Distance with Applications28 Jul 2021
- 2021A Survey on Optimal Transport for Machine Learning: Theory and Applications3 Jun 2021
- 2021Re-evaluating Word Mover's Distance30 May 2021
- 2021Sentence Similarity Based on Contexts17 May 2021
- 2021Efficient Optimal Transport Algorithm by Accelerated Gradient descent12 Apr 2021
- 2021Unsupervised Ground Metric Learning using Wasserstein Singular Vectors11 Feb 2021
- 2021On Transportation of Mini-batches: A Hierarchical Approach11 Feb 2021
50 newest of 53 All 53 (JSON)
- 2023Recent Advances in Optimal Transport for Machine Learningcited by 66
- 2022Representing Mixtures of Word Embeddings with Mixtures of Topic Embeddingscited by 36
- 2021A Survey on Optimal Transport for Machine Learning: Theory and Applicationscited by 35
- 2021Multi-Vector Models with Textual Guidance for Fine-Grained Scientific Document Similaritycited by 35
- 2023Scalable Optimal Transport Methods in Machine Learning: A Contemporary Surveycited by 27
- 2022Wasserstein Task Embedding for Measuring Task Similaritiescited by 24
- 2021On Transportation of Mini-batches: A Hierarchical Approachcited by 15
- 2022Adaptive Distribution Calibration for Few-Shot Learning with Hierarchical Optimal Transportcited by 14
- 2021Sentence Similarity Based on Contextscited by 12
- 2021Supervised Tree-Wasserstein Distancecited by 10
- 2021Re-evaluating Word Mover's Distancecited by 10
- 2021Hierarchical Optimal Transport for Unsupervised Domain Adaptationcited by 10
12 most cited of 52 All 52 (JSON)
- 2026Coupled Usage-Sense Processes: Temporal and Attributable Lexical Semantic Change25 Sep 2026
- 2026Fidelity-Diversity Metrics for Text6 Jul 2026
- 2026RADAR: Relative Angular Divergence Across Representations21 May 2026
- 2026Online Continual Learning with Dynamic Label Hierarchies12 May 2026
- 2026OT on the Map: Quantifying Domain Shifts in Geographic Space17 Apr 2026
- 2026Calibrating Tabular Anomaly Detection via Optimal Transport6 Feb 2026
- 2026Grammatical Error Correction Evaluation by Optimally Transporting Edit Representation5 Feb 2026
- 2026Robust Generalization with Adaptive Optimal Transport Priors for Decision-Focused Learning1 Feb 2026
- 2026HOT-POT: Optimal Transport for Sparse Stereo Matching18 Jan 2026
- 2025Efficient Transferable Optimal Transport via Min-Sliced Transport Plans24 Nov 2025
- 2025Generalized Sobolev IPM for Graph-Based Measures29 Oct 2025
- 2025Slicing Wasserstein Over Wasserstein Via Functional Optimal Transport26 Sep 2025
12 newest of 52 All 52 (JSON)