AutoGraph: Automated Graph Neural Network

By Yaoman Li, Irwin King. 2020

57 citations recorded by the source corpus.

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.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition, Information Systems

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Huan Zhao
  2. Lan Wei
  3. Quanming Yao
  4. Zhiqiang He

In Computer Vision and Pattern Recognition

  1. Huan Zhao
  2. Lan Wei
  3. Quanming Yao
  4. Zhiqiang He

In Information Systems

  1. Lan Wei
  2. Huan Zhao
  3. Quanming Yao
  4. Zhiqiang He

Citations in the corpus