Improving Graph Neural Networks with Simple Architecture Design
Sunil Kumar Maurya Tokyo Institute of TechnologyTokyoJapan , Xin Liu AIRC, AISTTokyoJapan and Tsuyoshi Murata Tokyo Institute of TechnologyTokyoJapan
Abstract
Graph Neural Networks have emerged as a useful tool to learn on the data by applying additional constraints based on the graph structure. These graphs are often created with assumed intrinsic relations between the entities. In recent years, there have been tremendous improvements in the architecture design, pushing the performance up in various prediction tasks. In general, these neural architectures combine layer depth and node feature aggregation steps. This makes it challenging to analyze the importance of features at various hops and the expressiveness of the neural network layers. As different graph datasets show varying levels of homophily and heterophily in features and class label distribution, it becomes essential to understand which features are important for the prediction tasks without any prior information. In this work, we decouple the node feature aggregation step and depth of graph neural network and introduce several key design strategies for graph neural networks. More specifically, we propose to use softmax as a regularizer and ”Soft-Selector” of features aggregated from neighbors at different hop distances; and ”Hop-Normalization” over GNN layers. Combining thes
原文 arXiv:2105.07634;中英对照 + 大白话阅读 https://aha.fim.ai/paper/2105.07634v1