.png){kind=icon}
返回
UniTrain: A universal iterative semi-supervised training framework for graph representation learning
.png){kind=icon}
PRE
AI
DOI：10.1016/j.neunet.2026.108623.png)
.png){kind=icon}
原文求助.png){kind=icon}
分享.png){kind=icon}
收藏
摘要
En 中文
Graph neural networks (GNNs) and graph transformers (GTs) perform well in graph-related tasks, but their potential is often limited in semi-supervised settings due to label scarcity. Although robust encoders and pre-training tasks enhance performance, GNNs or GTs remain prone to over-fitting and task gaps. To address these issues, we propose the Universal iterative semi-supervised Training framework (UniTrain), which generates high-quality pseudo-labels for unlabeled nodes in a multi-stage manner. Within UniTrain, a semantic graph is constructed using hidden vector representations from pre-training stage. Label knowledge propagation with uncertainty filtering is then applied to infer or refine labels for unlabeled nodes. By incorporating high-confidence pseudo-labels, the framework mitigates noise and compensates for the limited guidance of original labels. UniTrain enhances fine-tuning in existing self-supervised methods and is compatible with any GNN or GT encoder. Extensive experiments conduct on seven graph-based benchmarks (Cora, Citeseer, Pubmed, Actor, Cornell, Texas, and Wisconsin) demonstrate significant improvements in node classification performance when applying UniTrain to both GNNs and GTs. These results validate the effectiveness and generalization capability of our method.
期刊
IF:
6.3
论文数:
8.2K
被引数:
3.0W
机构
引用论文
Semi-supervised node classification via fine-grained graph auxiliary augmentation learning.png)
PATTERN RECOGNITION
PATTERN RECOGNITIONIF7.6
PREAI
分享收藏
Pseudo-label guided dual classifier domain adversarial network for unsupervised cross-domain fault diagnosis with small samples伪标签引导的双分类器域对抗网络用于小样本无监督跨域故障诊断
PREAI
分享收藏
查看更多内容