A pseudo-labeling approach based on knowledge distillation for graph few-shot learning

被引:0
作者
Wu, Zongqian [1 ]
Zhou, Peng [3 ]
Wen, Guoqiu [1 ,4 ]
Zhu, Xiaofeng [1 ,2 ]
机构
[1] Univ Elect Sci & Technol China, Sch Comp Sci & Technol, Chengdu, 611731, Peoples R China
[2] Hainan Univ, Sch Comp Sci & Technol, Haikou, Peoples R China
[3] Hunan Univ, Coll Comp Sci & Elect Engn, Changsha, Peoples R China
[4] Guangxi Normal Univ, Guangxi Key Lab Multisource Informat Min & Secur, Guilin, Peoples R China
关键词
Few-shot learning; Knowledge distillation; Graph convolutional networks; Few-shot node classification;
D O I
10.1016/j.ipm.2025.104268
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Graph-based few-shot node classification (FSNC) has emerged as a promising solution to the challenge of limited labeled nodes in complex network analysis. Although existing pseudo-labeling FSNC methods have shown encouraging results, they often struggle with wrong or over-confident pseudo-labels, which can negatively impact model generalization. To overcome these limitations, we propose PLD-FSNC, a novel pseudo-labeling FSNC framework leveraging knowledge distillation. Our PLD-FSNC framework is composed of two modules, i.e., embedding transfer and pseudo-label improvement. The embedding transfer module transfers knowledge from a pre-trained source model to a target model, enhancing pseudo-label selection quality. The pseudo-label improvement module mitigates the impact of wrong and over-confident pseudo-labels by using soft labels from the source model to supervise the target model's predictions. We also provide theoretical justification for our pseudo-label improvement module and demonstrate its effectiveness through extensive experiments on six real-world datasets.
引用
收藏
页数:16
相关论文
共 41 条
[1]   Hypergraph convolution and hypergraph attention [J].
Bai, Song ;
Zhang, Feihu ;
Torr, Philip H. S. .
PATTERN RECOGNITION, 2021, 110
[2]   Knowledge Distillation with the Reused Teacher Classifier [J].
Chen, Defang ;
Mei, Jian-Ping ;
Zhang, Hailin ;
Wang, Can ;
Feng, Yan ;
Chen, Chun .
2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2022, :11923-11932
[3]   Few-shot learning with unsupervised part discovery and part-aligned similarity [J].
Chen, Wentao ;
Zhang, Zhang ;
Wang, Wei ;
Wang, Liang ;
Wang, Zilei ;
Tan, Tieniu .
PATTERN RECOGNITION, 2023, 133
[4]  
Chung F. R. K., 1997, CBMS Regional Conference Series in Mathematics, DOI DOI 10.1090/CBMS/092
[5]  
Deng X, 2021, PROCEEDINGS OF THE THIRTIETH INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE, IJCAI 2021, P2321
[6]   Graph Prototypical Networks for Few-shot Learning on Attributed Networks [J].
Ding, Kaize ;
Wang, Jianling ;
Li, Jundong ;
Shu, Kai ;
Liu, Chenghao ;
Liu, Huan .
CIKM '20: PROCEEDINGS OF THE 29TH ACM INTERNATIONAL CONFERENCE ON INFORMATION & KNOWLEDGE MANAGEMENT, 2020, :295-304
[7]  
Finn C, 2017, PR MACH LEARN RES, V70
[8]  
Furlanello T, 2018, PR MACH LEARN RES, V80
[9]   node2vec: Scalable Feature Learning for Networks [J].
Grover, Aditya ;
Leskovec, Jure .
KDD'16: PROCEEDINGS OF THE 22ND ACM SIGKDD INTERNATIONAL CONFERENCE ON KNOWLEDGE DISCOVERY AND DATA MINING, 2016, :855-864
[10]  
Hamilton WL, 2017, ADV NEUR IN, V30