Chapter17: Graph Neural Networks: AutoML

Kaixiong Zhou, Texas A&M University, zkxiong@tamu.edu
Zirui Liu, Texas A&M University, tradigrada@tamu.edu
Keyu Duan, Texas A&M University, k.duan@tamu.edu
Xia Hu, Texas A&M University, hu@cse.tamu.edu

Abstract

Graph neural networks (GNNs) are efficient deep learning tools to analyze networked data. Being widely applied in graph analysis tasks, the rapid evolution of GNNs has led to a growing number of novel architectures. In practice, both neural architecture construction and training hyperparameter tuning are crucial to the node representation learning and the final model performance. However, as the graph data characteristics vary significantly in the real-world systems, given a specific scenario, rich human expertise and tremendous laborious trials are required to identify a suitable GNN architecture and training hyperparameters. Recently, automated machinelearning (AutoML) has shown its potential in finding the optimal solutions automatically for machine learning applications. While releasing the burden of the manual tuning process, AutoML could guarantee access of the optimal solution without extensive expert experience. Motivated from the previous successes of AutoML, there have been some preliminary automated GNN (AutoGNN) frameworks developed to tackle the problems of GNN neural architecture search (GNN-NAS) and training hyperparameter tuning. This chapter presents a comprehensive and up-to-date review of AutoGNN in terms of two perspectives, namely search space and search algorithm. Specifically, we mainly focus on the GNN-NAS problem and present the state-of-the-art techniques in these two perspectives. We further discuss the open problems related to the existing methods for the future research.

Contents

  • Background
    • Notations of AutoGNN
    • Problem Definition of AutoGNN
    • Challenges in AutoGNN
  • Search Space
    • Architecture Search Space
    • Training Hyperparameter Search Space
    • Efficient Search Space
  • Search Algorithms
    • Random Search
    • Evolutionary Search
    • Reinforcement Learning Based Search
    • Differentiable Search
    • Efficient Performance Estimation
  • Future Directions

Citation

@incollection{GNNBook-ch17-zhou,
author = "Zhou, Kaixiong and Liu, Zirui and Duan, Keyu and Hu, Xia",
editor = "Wu, Lingfei and Cui, Peng and Pei, Jian and Zhao, Liang",
title = "Graph Neural Network: AutoML",
booktitle = "Graph Neural Networks: Foundations, Frontiers, and Applications",
year = "2022",
publisher = "Springer Singapore",
address = "Singapore",
pages = "371--389",
}

K. Zhou, Z. Liu, K. Duan, and X. Hu, “Graph neural network: Automl,” in Graph Neural Networks: Foundations, Frontiers, and Applications, L. Wu, P. Cui, J. Pei, and L. Zhao, Eds. Singapore: Springer Singapore, 2022, pp. 371–389.