arXiv · 2207.08779
Simplifying Clustering with Graph Neural Networks
Abstract
The objective functions used in spectral clustering are usually composed of two terms: i) a term that minimizes the local quadratic variation of the cluster assignments on the graph and; ii) a term that balances the clustering partition and helps avoiding degenerate solutions. This paper shows that a graph neural network, equipped with suitable message passing layers, can generate good cluster assignments by optimizing only a balancing term. Results on attributed graph datasets show the effectiveness of the proposed approach in terms of clustering performance and computation time.
Explore related subjects
Keep this discovery
Filippo Maria Bianchi. 2022-07-18. Simplifying Clustering with Graph Neural Networks. https://doi.org/10.7557/18.6790
Cite the original work for its findings. Save a collection to share your selection of sources.