Multilevel Clustering via Wasserstein Means
Not on the map. Why?
Abstract (arXiv)
We propose a novel approach to the problem of multilevel clustering, which aims to simultaneously partition data in each group and discover grouping patterns among groups in a potentially large hierarchically structured corpus of data. Our method involves a joint optimization formulation over several spaces of discrete probability measures, which are endowed with Wasserstein distance metrics. We propose a number of variants of this problem, which admit fast optimization algorithms, by exploiting the connection to the problem of finding Wasserstein barycenters. Consistency properties are established for the estimates of both local and global clusters. Finally, experiment results with both synthetic and real data are presented to demonstrate the flexibility and scalability of the proposed approach.
Key facts. Cited by 49 papers in the map (data through 14 Oct 2026). First posted on arXiv 13 Jun 2017.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 18; 2022: 8; 2023: 11; 2024: 4; 2025: 4; 2026: 4.
49 papers.
Sort:
By area:
- 2021Point-set Distances for Learning Representations of 3D Point Cloudscited by 48
- 2021Improving Mini-batch Optimal Transport via Partial Transportationcited by 41
- 2021Automatic Text Evaluation through the Lens of Wasserstein Barycenterscited by 38
- 2021Averaging on the Bures-Wasserstein manifold: dimension-free convergence of gradient descentcited by 32
- 2021Unbalanced Optimal Transport through Non-negative Penalized Linear Regressioncited by 31
- 2021Wasserstein barycenters are NP-hard to computecited by 28
- 2021On Robust Optimal Transport: Computational Complexity and Barycenter Computationcited by 27
- 2022Revisiting Sliced Wasserstein on Images: From Vectorization to Convolutioncited by 25
- 2022Hierarchical Sliced Wasserstein Distancecited by 23
- 2023Quasi-Monte Carlo for 3D Sliced Wassersteincited by 22
- 2025On the Relation between Rectified Flows and Optimal Transportcited by 18
- 2021On Transportation of Mini-batches: A Hierarchical Approachcited by 16
37 more
- 2022On Unbalanced Optimal Transport: Gradient Methods, Sparsity and Approximation Errorcited by 16
- 2023Sliced Wasserstein Estimation with Control Variatescited by 16
- 2022Wasserstein Barycenter-based Model Fusion and Linear Mode Connectivity of Neural Networkscited by 15
- 2021Semi-Discrete Optimal Transport: Hardness, Regularization and Numerical Solutioncited by 13
- 2021Dimensionality Reduction for Wasserstein Barycentercited by 13
- 2021Entropic Gromov-Wasserstein between Gaussian Distributionscited by 12
- 2022Small Transformers Compute Universal Metric Embeddingscited by 12
- 2023Lower Complexity Adaptation for Empirical Entropic Optimal Transportcited by 12
- 2021Hierarchical Optimal Transport for Unsupervised Domain Adaptationcited by 10
- 2024Interaction-Force Transport Gradient Flowscited by 9
- 2021Projection Robust Wasserstein Barycenterscited by 8
- 2021On Multimarginal Partial Optimal Transport: Equivalent Forms and Computational Complexitycited by 8
- 2021Functional optimal transport: map estimation and domain adaptation for functional datacited by 7
- 2023Self-Attention Amortized Distributional Projection Optimization for Sliced Wasserstein Point-Cloud Reconstructioncited by 7
- 2024Tree-Sliced Wasserstein Distance: A Geometric Perspectivecited by 7
- 2024Prototypical Partial Optimal Transport for Universal Domain Adaptationcited by 7
- 2023LCOT: Linear circular optimal transportcited by 4
- 2023Fast Dual Subgradient Optimization of the Integrated Transportation Distance Between Stochastic Kernelscited by 4
- 2022Improving Generative Flow Networks with Path Regularizationcited by 3
- 2024Approximate Algorithms For k-Sparse Wasserstein Barycenter With Outlierscited by 3
- 2025Sobolev Gradient Ascent for Optimal Transport: Barycenter Optimization and Convergence Analysiscited by 2
- 2026An Efficient and Effective Evaluator for Text2SQL Models on Unseen and Unlabeled Datacited by 2
- 2026UniPROT: Uniform Prototype Selection via Partial Optimal Transport with Submodular Guaranteescited by 2
- 2021Limit Distribution Theory for the Smooth 1-Wasserstein Distance with Applicationscited by 1
- 2022Stability of Entropic Wasserstein Barycenters and application to random geometric graphscited by 1
- 2023Keypoint-Guided Optimal Transport: Models, Algorithms, and Applicationscited by 1
- 2023The geometry of financial institutions -- Wasserstein clustering of financial datacited by 1
- 2023Parameter Estimation in DAGs from Incomplete Data via Optimal Transportcited by 1
- 2023A Specialized Semismooth Newton Method for Kernel-Based Optimal Transportcited by 1
- 2025Federated Calculation of the Free-Support Transportation Barycenter by Single-Loop Dual Decompositioncited by 1
- 2021Selective information exchange in collaborative clustering using regularized Optimal Transportcited by 0
- 2021Estimation of Stationary Optimal Transport Planscited by 0
- 2022Efficient Forecasting of Large Scale Hierarchical Time Series via Multilevel Clusteringcited by 0
- 2023Learning to Quantize Vulnerability Patterns and Match to Locate Statement-Level Vulnerabilitiescited by 0
- 2025Partial Distribution Alignment via Adaptive Optimal Transportcited by 0
- 2026Multi-Domain Clustering via Measure Quantizationcited by 0
- 2026Robust blind unmixing: A geometric approach to overcoming basis variationcited by 0
- 2026Robust blind unmixing: A geometric approach to overcoming basis variation2 Oct 2026
- 2026Multi-Domain Clustering via Measure Quantization18 Sep 2026
- 2026UniPROT: Uniform Prototype Selection via Partial Optimal Transport with Submodular Guarantees13 Apr 2026
- 2026An Efficient and Effective Evaluator for Text2SQL Models on Unseen and Unlabeled Data8 Mar 2026
- 2025Federated Calculation of the Free-Support Transportation Barycenter by Single-Loop Dual Decomposition25 Jul 2025
- 2025On the Relation between Rectified Flows and Optimal Transport26 May 2025
- 2025Sobolev Gradient Ascent for Optimal Transport: Barycenter Optimization and Convergence Analysis19 May 2025
- 2025Partial Distribution Alignment via Adaptive Optimal Transport7 Mar 2025
- 2024Prototypical Partial Optimal Transport for Universal Domain Adaptation2 Aug 2024
- 2024Tree-Sliced Wasserstein Distance: A Geometric Perspective19 Jun 2024
- 2024Interaction-Force Transport Gradient Flows27 May 2024
- 2024Approximate Algorithms For k-Sparse Wasserstein Barycenter With Outliers20 Apr 2024
37 more
- 2023Fast Dual Subgradient Optimization of the Integrated Transportation Distance Between Stochastic Kernels3 Dec 2023
- 2023A Specialized Semismooth Newton Method for Kernel-Based Optimal Transport21 Oct 2023
- 2023LCOT: Linear circular optimal transport9 Oct 2023
- 2023Quasi-Monte Carlo for 3D Sliced Wasserstein21 Sep 2023
- 2023Lower Complexity Adaptation for Empirical Entropic Optimal Transport23 Jun 2023
- 2023Learning to Quantize Vulnerability Patterns and Match to Locate Statement-Level Vulnerabilities26 May 2023
- 2023Parameter Estimation in DAGs from Incomplete Data via Optimal Transport25 May 2023
- 2023The geometry of financial institutions -- Wasserstein clustering of financial data5 May 2023
- 2023Sliced Wasserstein Estimation with Control Variates30 Apr 2023
- 2023Keypoint-Guided Optimal Transport: Models, Algorithms, and Applications23 Mar 2023
- 2023Self-Attention Amortized Distributional Projection Optimization for Sliced Wasserstein Point-Cloud Reconstruction12 Jan 2023
- 2022Stability of Entropic Wasserstein Barycenters and application to random geometric graphs19 Oct 2022
- 2022Wasserstein Barycenter-based Model Fusion and Linear Mode Connectivity of Neural Networks13 Oct 2022
- 2022Improving Generative Flow Networks with Path Regularization29 Sep 2022
- 2022Hierarchical Sliced Wasserstein Distance27 Sep 2022
- 2022Small Transformers Compute Universal Metric Embeddings14 Sep 2022
- 2022Efficient Forecasting of Large Scale Hierarchical Time Series via Multilevel Clustering27 May 2022
- 2022Revisiting Sliced Wasserstein on Images: From Vectorization to Convolution4 Apr 2022
- 2022On Unbalanced Optimal Transport: Gradient Methods, Sparsity and Approximation Error8 Feb 2022
- 2021Hierarchical Optimal Transport for Unsupervised Domain Adaptation3 Dec 2021
- 2021Dimensionality Reduction for Wasserstein Barycenter18 Oct 2021
- 2021Automatic Text Evaluation through the Lens of Wasserstein Barycenters27 Aug 2021
- 2021Entropic Gromov-Wasserstein between Gaussian Distributions24 Aug 2021
- 2021Improving Mini-batch Optimal Transport via Partial Transportation22 Aug 2021
- 2021On Multimarginal Partial Optimal Transport: Equivalent Forms and Computational Complexity18 Aug 2021
- 2021Limit Distribution Theory for the Smooth 1-Wasserstein Distance with Applications28 Jul 2021
- 2021Estimation of Stationary Optimal Transport Plans25 Jul 2021
- 2021Averaging on the Bures-Wasserstein manifold: dimension-free convergence of gradient descent16 Jun 2021
- 2021Unbalanced Optimal Transport through Non-negative Penalized Linear Regression8 Jun 2021
- 2021Selective information exchange in collaborative clustering using regularized Optimal Transport22 Mar 2021
- 2021Semi-Discrete Optimal Transport: Hardness, Regularization and Numerical Solution10 Mar 2021
- 2021On Robust Optimal Transport: Computational Complexity and Barycenter Computation13 Feb 2021
- 2021On Transportation of Mini-batches: A Hierarchical Approach11 Feb 2021
- 2021Point-set Distances for Learning Representations of 3D Point Clouds8 Feb 2021
- 2021Functional optimal transport: map estimation and domain adaptation for functional data7 Feb 2021
- 2021Projection Robust Wasserstein Barycenters5 Feb 2021
- 2021Wasserstein barycenters are NP-hard to compute4 Jan 2021
- 2021Point-set Distances for Learning Representations of 3D Point Cloudscited by 48
- 2021Improving Mini-batch Optimal Transport via Partial Transportationcited by 41
- 2021Automatic Text Evaluation through the Lens of Wasserstein Barycenterscited by 38
- 2021Averaging on the Bures-Wasserstein manifold: dimension-free convergence of gradient descentcited by 32
- 2021Unbalanced Optimal Transport through Non-negative Penalized Linear Regressioncited by 31
- 2021Wasserstein barycenters are NP-hard to computecited by 28
- 2021On Robust Optimal Transport: Computational Complexity and Barycenter Computationcited by 27
- 2022Revisiting Sliced Wasserstein on Images: From Vectorization to Convolutioncited by 25
- 2022Hierarchical Sliced Wasserstein Distancecited by 23
- 2023Quasi-Monte Carlo for 3D Sliced Wassersteincited by 22
- 2021On Transportation of Mini-batches: A Hierarchical Approachcited by 16
- 2022On Unbalanced Optimal Transport: Gradient Methods, Sparsity and Approximation Errorcited by 16
12 most cited of 48 All 48 (JSON)
- 2026Robust blind unmixing: A geometric approach to overcoming basis variation2 Oct 2026
- 2026Multi-Domain Clustering via Measure Quantization18 Sep 2026
- 2026UniPROT: Uniform Prototype Selection via Partial Optimal Transport with Submodular Guarantees13 Apr 2026
- 2026An Efficient and Effective Evaluator for Text2SQL Models on Unseen and Unlabeled Data8 Mar 2026
- 2025Federated Calculation of the Free-Support Transportation Barycenter by Single-Loop Dual Decomposition25 Jul 2025
- 2025Sobolev Gradient Ascent for Optimal Transport: Barycenter Optimization and Convergence Analysis19 May 2025
- 2025Partial Distribution Alignment via Adaptive Optimal Transport7 Mar 2025
- 2024Prototypical Partial Optimal Transport for Universal Domain Adaptation2 Aug 2024
- 2024Tree-Sliced Wasserstein Distance: A Geometric Perspective19 Jun 2024
- 2024Interaction-Force Transport Gradient Flows27 May 2024
- 2024Approximate Algorithms For k-Sparse Wasserstein Barycenter With Outliers20 Apr 2024
- 2023Fast Dual Subgradient Optimization of the Integrated Transportation Distance Between Stochastic Kernels3 Dec 2023
12 newest of 48 All 48 (JSON)