Revisiting Semi-Supervised Learning with Graph Embeddings
Zhilin YangWilliam W. CohenRuslan Salakhutdinov
Proposes a graph embedding framework for semi-supervised learning that jointly predicts class labels and neighborhood context across both transductive and inductive settings, outperforming standard methods on text and entity classification benchmarks.
The article addresses challenges in semi-supervised learning, where limited labeled data and graph structures between instances must be leveraged to improve classification accuracy in tasks such as text categorization and entity recognition. Traditional graph-based methods rely on regularization to enforce label consistency among connected nodes but often fail to fully exploit distributional patterns in the graph or produce reusable features. This matters now because many real-world applications, including knowledge-base completion and web-scale information extraction, face scarce labels yet abundant relational data.
The article set out to evaluate a framework called Planetoid that trains instance embeddings to jointly predict class labels and neighborhood context in the graph. It develops both a transductive version, which uses free embeddings for observed nodes, and an inductive version, which defines embeddings as a function of input features so predictions can be made on unseen instances.
The authors tested the approach through controlled experiments on five benchmark datasets spanning text classification, distantly supervised entity extraction, and entity classification in a knowledge base. They compared the new methods against established baselines including label propagation, manifold regularization, transductive support vector machines, semi-supervised embeddings, and unsupervised graph embeddings, using accuracy and recall metrics under varying label rates and consistent data splits.
The inductive variant outperformed the best prior inductive method by up to 18.7 percentage points and 7.8 points on average; the strongest of either variant beat the best competing method by up to 8.5 points and 4.1 on average. Joint training with label information improved embedding quality over purely unsupervised graph embedding, and predicting graph context proved more effective than Laplacian regularization alone. Performance gains were largest when label rates were low and when graph structure carried more signal than raw features.
These results indicate that embedding-based context prediction can deliver higher accuracy and better generalization in label-scarce settings, potentially reducing annotation costs and improving reliability for downstream systems that rely on node classification. The inductive formulation additionally enables deployment on streaming or large-scale data where new instances arrive after training.
The article recommends extending the framework to recurrent networks and testing cases where graphs are derived from feature distances rather than external relations. Further work is needed to assess sensitivity to hyper-parameters and to confirm gains on additional domains before broad adoption.
The findings rest on five datasets and fixed hyper-parameter choices; results may vary if feature vectors are uninformative or if graphs contain substantial noise. Confidence is moderate to high for the reported benchmarks but readers should verify behavior on their own data distributions before making deployment decisions.
- Paper: Manifold Regularization: A Geometric Framework for Learning from Labeled and Unlabeled Examples, Mikhail Belkin et al. (2006). It establishes the foundational manifold regularization framework using graph Laplacians for semi-supervised learning that the source explicitly sets out to improve upon.
- Paper: Learning with Local and Global Consistency, Dengyong Zhou et al. (2003). It introduces classic label propagation and consistency on graphs, which serves as a core baseline and foundational principle revisited by the source.
- Paper: LINE: Large-scale Information Network Embedding, Jian Tang et al. (2015). It introduces scalable network embedding preserving first- and second-order proximities, providing foundational principles for neighborhood context modeling used in the source.
- Paper: Translating Embeddings for Modeling Multi-relational Data, Antoine Bordes et al. (2013). It provides foundational knowledge graph embedding methods (TransE) that the source builds upon and benchmarks against for entity classification tasks.
- Paper: Embedding Entities and Relations for Learning and Inference in Knowledge Bases, Bishan Yang et al. (2014). It details relational embedding frameworks for knowledge bases that establish the multi-relational baselines compared in the source.
- Paper: Combining active learning and semi-supervised learning using Gaussian fields and harmonic functions, Xiaojin Zhu et al. (2003). It details harmonic function-based label propagation on graphs, offering essential background for transductive graph semi-supervised learning.
- Paper: Distant supervision for relation extraction without labeled data, Mike D. Mintz et al. (2009). It presents distant supervision for relation extraction, formulating the setting and benchmark task evaluated in the source's empirical study.
- Paper: Spectral Networks and Locally Connected Networks on Graphs, Joan Bruna et al. (2014). It provides foundational formulations of spectral and spatial neural networks on graphs that contextualize neural feature propagation.
- Paper: Semi-Supervised Classification with Graph Convolutional Networks, Thomas N. Kipf et al. (2017). It builds directly on the semi-supervised graph learning paradigm by introducing Graph Convolutional Networks (GCNs) and uses the source's Planetoid data splits and baselines as primary benchmarks.
- Paper: Inductive Representation Learning on Large Graphs, William L. Hamilton et al. (2017). It extends inductive graph representation learning to large-scale networks using neighborhood sampling and aggregation functions.
- Paper: Graph Attention Networks, Petar Veličković et al. (2018). It advances semi-supervised classification on citation graphs by introducing attention mechanisms over neighbor features.
- Paper: Modeling Relational Data with Graph Convolutional Networks, Michael Schlichtkrull et al. (2018). It adapts graph convolutional architectures to multi-relational knowledge graphs, extending relational modeling beyond the embedding strategies in the source.
- Paper: Representation Learning on Graphs with Jumping Knowledge Networks, Keyulu Xu et al. (2018). It improves upon semi-supervised graph neural networks by introducing adaptive aggregation depths across multi-hop neighborhoods.
- Paper: Deeper Insights into Graph Convolutional Networks for Semi-Supervised Learning, Qimai Li et al. (2018). It provides theoretical and empirical insights into semi-supervised graph convolutions under very low label rates like those evaluated in the source.
- Paper: Predict then Propagate: Graph Neural Networks meet Personalized PageRank, Johannes Gasteiger et al. (2019). It unifies neural predictions with personalized PageRank propagation to overcome oversmoothing in semi-supervised node classification.
- Paper: Deep Graph Infomax, Petar Veličković et al. (2019). It explores unsupervised and self-supervised node representation learning by maximizing mutual information between local patches and global graph context.
- Paper: metapath2vec: Scalable Representation Learning for Heterogeneous Networks, Yuxiao Dong et al. (2017). It generalizes node representation learning to heterogeneous information networks using meta-path-guided random walks.
- Paper: A survey on semi-supervised learning, Jesper E. van Engelen et al. (2019). It provides an extensive survey synthesizing two decades of transductive and inductive semi-supervised learning methods.
