Graph Cuts with Arbitrary Size Constraints Through Optimal Transport - Archive ouverte HAL
Article Dans Une Revue Transactions on Machine Learning Research Journal Année : 2024

Graph Cuts with Arbitrary Size Constraints Through Optimal Transport

Chakib Fettal
Lazhar Labiod
Mohamed Nadif

Résumé

A common way of partitioning graphs is through minimum cuts. One drawback of classical minimum cut methods is that they tend to produce small groups, which is why more balanced variants such as normalized and ratio cuts have seen more success. However, we believe that with these variants, the balance constraints can be too restrictive for some applications like for clustering of imbalanced datasets, while not being restrictive enough for when searching for perfectly balanced partitions. Here, we propose a new graph cut algorithm for partitioning graphs under arbitrary size constraints. We formulate the graph cut problem as a Gromov-Wasserstein with a concave regularizer problem. We then propose to solve it using an accelerated proximal GD algorithm which guarantees global convergence to a critical point, results in sparse solutions and only incurs an additional ratio of $\mathcal{O}(\log(n))$ compared to the classical spectral clustering algorithm but was seen to be more efficient.
Fichier principal
Vignette du fichier
2585_Graph_Cuts_with_Arbitrary.pdf (627.68 Ko) Télécharger le fichier
Origine Fichiers éditeurs autorisés sur une archive ouverte

Dates et versions

hal-03917041 , version 1 (31-12-2022)
hal-03917041 , version 2 (11-05-2023)
hal-03917041 , version 3 (30-08-2023)
hal-03917041 , version 4 (07-02-2024)
hal-03917041 , version 5 (04-10-2024)

Identifiants

  • HAL Id : hal-03917041 , version 5

Citer

Chakib Fettal, Lazhar Labiod, Mohamed Nadif. Graph Cuts with Arbitrary Size Constraints Through Optimal Transport. Transactions on Machine Learning Research Journal, 2024. ⟨hal-03917041v5⟩
539 Consultations
325 Téléchargements

Partager

More