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

NodeNet: A Graph Regularised Neural Network for Node Classification

About

Real-world events exhibit a high degree of interdependence and connections, and hence data points generated also inherit the linkages. However, the majority of AI/ML techniques leave out the linkages among data points. The recent surge of interest in graph-based AI/ML techniques is aimed to leverage the linkages. Graph-based learning algorithms utilize the data and related information effectively to build superior models. Neural Graph Learning (NGL) is one such technique that utilizes a traditional machine learning algorithm with a modified loss function to leverage the edges in the graph structure. In this paper, we propose a model using NGL - NodeNet, to solve node classification task for citation graphs. We discuss our modifications and their relevance to the task. We further compare our results with the current state of the art and investigate reasons for the superior performance of NodeNet.

Shrey Dabhi, Manojkumar Parmar• 2020

Related benchmarks

TaskDatasetResultRank
Node ClassificationCora
Accuracy86.8
885
Node ClassificationCiteseer
Accuracy80.09
804
Node ClassificationPubmed
Accuracy90.21
742
Node ClassificationACTIVSg200
Accuracy80.15
18
Node ClassificationACTIVSg500
Accuracy95
18
Node ClassificationCora-ML+ (80% train 20% test)
Average Accuracy84.03
2
Node ClassificationCora with TF-IDF vectors
Accuracy85.17
1
Node ClassificationCiteseer with TF-IDF vectors
Accuracy78.02
1
Node ClassificationCora additional (train)--
1
Showing 9 of 9 rows

Other info

Follow for update