Graph Transductive Sharpening: Leveraging Unlabeled Predictions in Node Classification
Summary
Transductive Sharpening (TS) is a novel loss-level modification for semi-supervised node classification, designed to exploit predictions on unlabeled nodes in a transductive setting. It minimizes prediction entropy on unlabeled nodes while counterbalancing this effect on labeled nodes, using Tsallis entropy (Gini impurity) for stable optimization. Evaluated across 13 node-classification benchmarks, TS consistently improves performance for Graph Neural Networks (GCN, GAT, GraphSAGE) and Multi-Layer Perceptrons (MLPs), adding only a single scalar hyperparameter, lambda. A universal lambda of 0.25 proved effective, demonstrating robustness without architecture changes or significant computational overhead.
Key takeaway
For Machine Learning Engineers developing node classification models in transductive settings, Transductive Sharpening offers a simple, architecture-agnostic method to boost performance. By exploiting unlabeled node predictions through a loss-level modification, you can achieve consistent gains across various GNNs and MLPs. Consider integrating TS with a conservative lambda, such as 0.25, to enhance model confidence and accuracy without increasing architectural complexity or computational overhead.
Key insights
Discarded unlabeled node predictions provide a valuable signal for improving transductive graph learning.
Principles
- Unlabeled predictions offer a direct training signal.
- Symmetric uncertainty shaping prevents overconfidence.
- Moderate sharpening improves decision boundaries.
Method
TS augments supervised loss by adding a term that minimizes Tsallis entropy on unlabeled nodes and maximizes it on labeled nodes, controlled by a scalar hyperparameter lambda.
In practice
- Apply TS to existing GNN or MLP architectures.
- Set the sharpening coefficient lambda between 0 and 0.5.
- Use Tsallis entropy (Gini impurity) for stable training.
Topics
- Transductive Node Classification
- Graph Neural Networks
- Loss Functions
- Entropy Minimization
- Semi-supervised Learning
- Tsallis Entropy
Code references
Best for: Research Scientist, AI Scientist, Machine Learning Engineer
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by cs.LG updates on arXiv.org.