LAUSR.org creates dashboard-style pages of related content for over 1.5 million academic articles. Sign Up to like articles & get recommendations!

Hierarchical Model Selection for Graph Neural Networks

Photo by thinkmagically from unsplash

Node classification on graph data is a major problem in machine learning, and various graph neural networks (GNNs) have been proposed. Variants of GNNs such as H2GCN and CPF outperform… Click to show full abstract

Node classification on graph data is a major problem in machine learning, and various graph neural networks (GNNs) have been proposed. Variants of GNNs such as H2GCN and CPF outperform graph convolutional networks (GCNs) by improving on the weaknesses of the traditional GNN. However, there are some graph data in which these GNN variants fail to perform well than other GNNs in the node classification task. This is because H2GCN has similar feature values on graph data with a high average degree, and CPF gives rise to a problem with label-propagation suitability. Accordingly, we propose a hierarchical model selection framework (HMSF) that selects an appropriate GNN model to predict the class of nodes for each graph data. HMSF uses average degree and edge homophily ratio as indicators to decide the useful model based on our analyses. In the experiment, we show that the model selected by our HMSF achieves high performance on node classification for various types of graph data.

Keywords: hierarchical model; neural networks; model selection; graph neural; model; graph data

Journal Title: IEEE Access
Year Published: 2022

Link to full text (if available)


Share on Social Media:                               Sign Up to like & get
recommendations!

Related content

More Information              News              Social Media              Video              Recommended



                Click one of the above tabs to view related content.