arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2201.09698cs.LGcs.AI
收藏

面向半监督学习的图神经扩散网络

Graph Neural Diffusion Networks for Semi-supervised Learning

Tongji University(同济大学);University of California, Santa Barbara(加州大学圣巴巴拉分校)

Wei Ye, Zexi Huang, Yunqi Hong, Ambuj Singh

AI总结:

针对GCN在稀疏标记图上的欠平滑与过平滑问题,提出融合神经网络与传统图扩散的GND-Nets,单层利用顶点局部全局邻域信息,实验验证其优于现有方法。

摘要:

图卷积网络(GCN)是基于图的半监督学习领域的开创性模型。但GCN在稀疏标记图上表现不佳:其两层版本无法有效将标签信息传播到整个图结构(即欠平滑问题),而深层版本则会出现过平滑且难以训练(即过平滑问题)。为解决这两个问题,我们提出一种名为GND-Nets(图神经扩散网络)的新型图神经网络,它在单层中利用顶点的局部和全局邻域信息。使用浅层网络可缓解过平滑问题,而利用局部和全局邻域信息则可缓解欠平滑问题。顶点局部与全局邻域信息的利用通过一种名为神经扩散的新型图扩散方法实现,该方法将神经网络融入传统的线性和非线性图扩散中。神经网络的采用使神经扩散能够适配不同数据集。在多种稀疏标记图上开展的大量实验证明,与当前最优方法相比,GND-Nets兼具有效性和高效性。

英文摘要:

Graph Convolutional Networks (GCN) is a pioneering model for graph-based semi-supervised learning. However, GCN does not perform well on sparsely-labeled graphs. Its two-layer version cannot effectively propagate the label information to the whole graph structure (i.e., the under-smoothing problem) while its deep version over-smoothens and is hard to train (i.e., the over-smoothing problem). To solve these two issues, we propose a new graph neural network called GND-Nets (for Graph Neural Diffusion Networks) that exploits the local and global neighborhood information of a vertex in a single layer. Exploiting the shallow network mitigates the over-smoothing problem while exploiting the local and global neighborhood information mitigates the under-smoothing problem. The utilization of the local and global neighborhood information of a vertex is achieved by a new graph diffusion method called neural diffusions, which integrate neural networks into the conventional linear and nonlinear graph diffusions. The adoption of neural networks makes neural diffusions adaptable to different datasets. Extensive experiments on various sparsely-labeled graphs verify the effectiveness and efficiency of GND-Nets compared to state-of-the-art approaches.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑