Propagation Model Search for Graph Neural Networks

This paper presents a novel neural architecture search (NAS) framework for graph neural networks (GNNs). We design an expressive search space that focuses on a common and critical component of GNNs -- propagation model. Specifically, we search for propagation matrices and the connections between propagation steps. Our search space covers various graph types, e.g., homogeneous graphs, heterogeneous graphs, and can be naturally extended to higher-dimensional recommender systems and spatial-temporal data. We propose a sampling-based one-shot NAS algorithm to search for appropriate propagation patterns efficiently. Extensive experiments in three different scenarios are used to evaluate the proposed framework. We show that the performance of the models obtained by our framework is better than state-of-the-art GNN methods. Furthermore, our framework can discover explainable meta-graphs in heterogeneous graphs.