Enregistré dans:
Détails bibliographiques
Auteurs principaux: Neocosmos, Kibidi, Baptista, Diego, Ludwig, Nicole
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:https://arxiv.org/abs/2512.05623
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866910148096688128
author Neocosmos, Kibidi
Baptista, Diego
Ludwig, Nicole
author_facet Neocosmos, Kibidi
Baptista, Diego
Ludwig, Nicole
contents In community detection, many methods require the user to specify the number of clusters in advance since an exhaustive search over all possible values is computationally infeasible. While some classical algorithms can infer this number directly from the data, this is typically not the case for graph neural networks (GNNs): even when a desired number of clusters is specified, standard GNN-based methods often fail to return the exact number due to the way they are designed. In this work, we address this limitation by introducing a flexible and principled way to control the number of communities discovered by GNNs. Rather than assuming the true number of clusters is known, we propose a framework that allows the user to specify a plausible range and enforce these bounds during training. However, if the user wants an exact number of clusters, it may also be specified and reliably returned.
format Preprint
id arxiv_https___arxiv_org_abs_2512_05623
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Bounded Graph Clustering with Graph Neural Networks
Neocosmos, Kibidi
Baptista, Diego
Ludwig, Nicole
Machine Learning
In community detection, many methods require the user to specify the number of clusters in advance since an exhaustive search over all possible values is computationally infeasible. While some classical algorithms can infer this number directly from the data, this is typically not the case for graph neural networks (GNNs): even when a desired number of clusters is specified, standard GNN-based methods often fail to return the exact number due to the way they are designed. In this work, we address this limitation by introducing a flexible and principled way to control the number of communities discovered by GNNs. Rather than assuming the true number of clusters is known, we propose a framework that allows the user to specify a plausible range and enforce these bounds during training. However, if the user wants an exact number of clusters, it may also be specified and reliably returned.
title Bounded Graph Clustering with Graph Neural Networks
topic Machine Learning
url https://arxiv.org/abs/2512.05623