English

Rethinking Graph Regularization for Graph Neural Networks

Machine Learning 2020-12-22 v2 Machine Learning

Abstract

The graph Laplacian regularization term is usually used in semi-supervised representation learning to provide graph structure information for a model f(X)f(X). However, with the recent popularity of graph neural networks (GNNs), directly encoding graph structure AA into a model, i.e., f(A,X)f(A, X), has become the more common approach. While we show that graph Laplacian regularization brings little-to-no benefit to existing GNNs, and propose a simple but non-trivial variant of graph Laplacian regularization, called Propagation-regularization (P-reg), to boost the performance of existing GNN models. We provide formal analyses to show that P-reg not only infuses extra information (that is not captured by the traditional graph Laplacian regularization) into GNNs, but also has the capacity equivalent to an infinite-depth graph convolutional network. We demonstrate that P-reg can effectively boost the performance of existing GNN models on both node-level and graph-level tasks across many different datasets.

Keywords

Cite

@article{arxiv.2009.02027,
  title  = {Rethinking Graph Regularization for Graph Neural Networks},
  author = {Han Yang and Kaili Ma and James Cheng},
  journal= {arXiv preprint arXiv:2009.02027},
  year   = {2020}
}

Comments

AAAI2021