English

Learn Layer-wise Connections in Graph Neural Networks

Machine Learning 2021-12-28 v1 Artificial Intelligence

Abstract

In recent years, Graph Neural Networks (GNNs) have shown superior performance on diverse applications on real-world datasets. To improve the model capacity and alleviate the over-smoothing problem, several methods proposed to incorporate the intermediate layers by layer-wise connections. However, due to the highly diverse graph types, the performance of existing methods vary on diverse graphs, leading to a need for data-specific layer-wise connection methods. To address this problem, we propose a novel framework LLC (Learn Layer-wise Connections) based on neural architecture search (NAS) to learn adaptive connections among intermediate layers in GNNs. LLC contains one novel search space which consists of 3 types of blocks and learnable connections, and one differentiable search algorithm to enable the efficient search process. Extensive experiments on five real-world datasets are conducted, and the results show that the searched layer-wise connections can not only improve the performance but also alleviate the over-smoothing problem.

Keywords

Cite

@article{arxiv.2112.13585,
  title  = {Learn Layer-wise Connections in Graph Neural Networks},
  author = {Lanning Wei and Huan Zhao and Zhiqiang He},
  journal= {arXiv preprint arXiv:2112.13585},
  year   = {2021}
}
R2 v1 2026-06-24T08:32:20.770Z