Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Random Features Strengthen Graph Neural Networks

About

Graph neural networks (GNNs) are powerful machine learning models for various graph learning tasks. Recently, the limitations of the expressive power of various GNN models have been revealed. For example, GNNs cannot distinguish some non-isomorphic graphs and they cannot learn efficient graph algorithms. In this paper, we demonstrate that GNNs become powerful just by adding a random feature to each node. We prove that the random features enable GNNs to learn almost optimal polynomial-time approximation algorithms for the minimum dominating set problem and maximum matching problem in terms of approximation ratios. The main advantage of our method is that it can be combined with off-the-shelf GNN models with slight modifications. Through experiments, we show that the addition of random features enables GNNs to solve various problems that normal GNNs, including the graph convolutional networks (GCNs) and graph isomorphism networks (GINs), cannot solve.

Ryoma Sato, Makoto Yamada, Hisashi Kashima• 2020

Related benchmarks

TaskDatasetResultRank
Graph RegressionZINC 12K (test)
MAE0.1621
164
Graph ClassificationMolHIV
ROC AUC75.98
82
Graph ClassificationMOLTOX21
ROC-AUC0.7514
38
Molecular property predictionMOLESOL
RMSE1.052
37
Graph ClassificationMOLBACE
ROC AUC0.7428
31
Small molecule classificationOGBG-MOLHIV (test)
ROC-AUC75.52
19
Showing 6 of 6 rows

Other info

Follow for update