Graph augmentation for node-level few-shot learning

被引:1
作者
Wu, Zongqian [1 ]
Zhou, Peng [3 ]
Ma, Junbo [4 ]
Zhang, Jilian [5 ]
Yuan, Guoqin [6 ]
Zhu, Xiaofeng [1 ,2 ]
机构
[1] Univ Elect Sci & Technol China, Sch Comp Sci & Engn, Chengdu 611731, Peoples R China
[2] Univ Elect Sci & Technol China, Shenzhen Inst Adv Study, Shenzhen 518000, Peoples R China
[3] Hunan Univ, Coll Comp Sci & Elect Engn, Changsha 410082, Peoples R China
[4] Guangxi Normal Univ, Dept Comp Sci & Engn, Guilin 541004, Peoples R China
[5] Jinan Univ, Coll Cyber Secur, Guangzhou 510632, Peoples R China
[6] Chinese Acad Sci, Changchun 130033, Peoples R China
基金
中国国家自然科学基金;
关键词
Few-shot learning; Node classification; Graph augmentation;
D O I
10.1016/j.knosys.2024.111872
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
In graph few-shot learning, few-shot node classification (FSNC) at the node-level is a popular downstream task. Previous FSNC methods primarily rely on meta-learning or metric learning techniques, aiming to mine prior knowledge from the base classes. However, these methods still have some limitations that need to be addressed, namely: (1) conducting multiple tasks for parameter initialization leads to expensive time costs. (2) ignoring the rich information present in novel classes leads to model over-fitting. To address these issues, this paper proposes a novel graph augmentation method for FSNC on graph data, which includes both parameter initialization and parameter fine-tuning. Specifically, the parameter initialization conducts only one multi-classification task on the base classes, improving generalization ability and reducing time costs. The parameter fine-tuning is designed to include two data augmentation modules ( i.e. , support augmentation and shot augmentation) on the novel classes to mine the rich information, thus alleviating model over-fitting. As a result, this paper introduces the first graph augmentation method for FSNC. Experimental results showed that our method achieves supreme performance, compared with state-of-the-art FSNC methods.
引用
收藏
页数:12
相关论文
共 46 条
[1]  
Chen JF, 2022, AAAI CONF ARTIF INTE, P10492
[2]  
Chen MY, 2019, Arxiv, DOI arXiv:1909.01515
[3]   Meta-Baseline: Exploring Simple Meta-Learning for Few-Shot Learning [J].
Chen, Yinbo ;
Liu, Zhuang ;
Xu, Huijuan ;
Darrell, Trevor ;
Wang, Xiaolong .
2021 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2021), 2021, :9042-9051
[4]   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
[5]   Meta-Learning of Neural Architectures for Few-Shot Learning [J].
Elsken, Thomas ;
Staffler, Benedikt ;
Metzen, Jan Hendrik ;
Hutter, Frank .
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2020), 2020, :12362-12372
[6]  
Fayjie A.R., 2021, Semi-supervised adversarial few-shot learning for medical image segmentation
[7]  
Finn C, 2017, PR MACH LEARN RES, V70
[8]  
Fu Xingcheng, 2023, WWW '23: Proceedings of the ACM Web Conference 2023, P460, DOI 10.1145/3543507.3583403
[9]   Multigraph Fusion for Dynamic Graph Convolutional Network [J].
Gan, Jiangzhang ;
Hu, Rongyao ;
Mo, Yujie ;
Kang, Zhao ;
Peng, Liang ;
Zhu, Yonghua ;
Zhu, Xiaofeng .
IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2024, 35 (01) :196-207
[10]  
Huang Q., 2020, Advances in Neural Information Processing Systems, V33