arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2011.11288cs.LGcs.AIcs.SI
收藏

AutoGraph:自动图神经网络

AutoGraph: Automated Graph Neural Network

The Chinese University of Hong Kong(香港中文大学);Lenovo Machine Intelligence Center(联想机器智能中心)

Yaoman Li, Irwin King

摘要:

图在许多应用中起着重要的作用。最近,图神经网络(GNNs)在图分析任务中取得了可喜的成果。已经提出了一些最先进的GNN模型,例如,图卷积网络(GCN)、图注意力网络(GAT)等。尽管取得了这些成功,但大多数GNN只有浅层结构。这导致了GNN的低表达能力。为了充分利用深度神经网络的能力,最近已经提出了一些深度GNN。然而,深度GNN的设计需要大量的架构工程。在这项工作中,我们提出了一种自动化深度GNNs设计的方法。在我们提出的方法中,我们向GNNs搜索空间添加了一种新型的跳过连接,以鼓励特征重用并缓解消失梯度问题。我们还允许我们的进化算法在进化过程中增加GNN的层数,以生成更深的网络。我们在图节点分类任务中评估我们的方法。实验结果表明,该方法生成的GNN在Cora、Citeseer、Pubmed和PPI数据集上都能获得最佳结果。

英文摘要:

Graphs play an important role in many applications. Recently, Graph Neural Networks (GNNs) have achieved promising results in graph analysis tasks. Some state-of-the-art GNN models have been proposed, e.g., Graph Convolutional Networks (GCNs), Graph Attention Networks (GATs), etc. Despite these successes, most of the GNNs only have shallow structure. This causes the low expressive power of the GNNs. To fully utilize the power of the deep neural network, some deep GNNs have been proposed recently. However, the design of deep GNNs requires significant architecture engineering. In this work, we propose a method to automate the deep GNNs design. In our proposed method, we add a new type of skip connection to the GNNs search space to encourage feature reuse and alleviate the vanishing gradient problem. We also allow our evolutionary algorithm to increase the layers of GNNs during the evolution to generate deeper networks. We evaluate our method in the graph node classification task. The experiments show that the GNNs generated by our method can obtain state-of-the-art results in Cora, Citeseer, Pubmed and PPI datasets.

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

↑