利用PyG实现图池化minCUT Pool(图分类任务)
(内含minCUT Pool代码实现)《Spectral Clustering with Graph Neural Networks for Graph Pooling》作者在本文中提出了一种可以在样本外的图上快速评估的聚类函数。根据所提出的聚类方法,作者设计了一个图池算子minCUT,并在有监督和无监督任务上都取得了极佳的性能。通俗来讲,作者提出的基于GNN的实现是可微分的,不需要计算频谱分解,并且学习了一个可以在样本外图上快速评估的聚类函数。
原创
2022-10-22 12:35:42 ·
3764 阅读 ·
1 评论