论文标题
用图神经网络简化聚类
Simplifying Clustering with Graph Neural Networks
论文作者
论文摘要
光谱群集中使用的目标函数通常由两个术语组成：i）一个术语最小化群集分配的局部二次变化，并且； ii）一个可以平衡聚类分区并有助于避免退化解决方案的术语。本文表明，配备合适消息传递层的图形神经网络可以通过仅优化平衡项来生成良好的集群分配。归因图数据集的结果显示了拟议方法在聚类性能和计算时间方面的有效性。
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.
