Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Revisiting Semi-Supervised Learning with Graph Embeddings

About

We present a semi-supervised learning framework based on graph embeddings. Given a graph between instances, we train an embedding for each instance to jointly predict the class label and the neighborhood context in the graph. We develop both transductive and inductive variants of our method. In the transductive variant of our method, the class labels are determined by both the learned embeddings and input feature vectors, while in the inductive variant, the embeddings are defined as a parametric function of the feature vectors, so predictions can be made on instances not seen during training. On a large and diverse set of benchmark tasks, including text classification, distantly supervised entity extraction, and entity classification, we show improved performance over many of the existing models.

Zhilin Yang, William W. Cohen, Ruslan Salakhutdinov• 2016

Related benchmarks

TaskDatasetResultRank
Node ClassificationCora
Accuracy75.7
1225
Node ClassificationCiteseer
Accuracy64.7
1037
Node ClassificationCiteseer (test)
Accuracy0.647
1013
Node ClassificationCora (test)
Mean Accuracy75.7
951
Node ClassificationChameleon
Accuracy45.27
936
Node ClassificationPubmed
Accuracy87.36
902
Node ClassificationCornell
Accuracy43.24
900
Node ClassificationTexas
Accuracy0.6216
859
Node ClassificationPubMed (test)
Accuracy77.2
628
Node ClassificationRoman-Empire
Accuracy41.09
398
Showing 10 of 79 rows
...

Other info

Follow for update