模式识别与人工智能
Friday, Apr. 4, 2025 Home      About Journal      Editorial Board      Instructions      Ethics Statement      Contact Us                   中文
Pattern Recognition and Artificial Intelligence  2022, Vol. 35 Issue (11): 955-964    DOI: 10.16451/j.cnki.issn1003-6059.202211001
Papers and Reports Current Issue| Next Issue| Archive| Adv Search |
Imbalanced Node Classification Algorithm Based on Self-Supervised Learning
CUI Caixia1,2, WANG Jie3, PANG Tianjie2, LIANG Jiye1
1. School of Computer and Information Technology, Shanxi University, Taiyuan 030006;
2. College of Computer Science and Technology, Taiyuan Normal University, Jinzhong 030619;
3. College of Computer Science and Technology, Taiyuan University of Science and Technology, Taiyuan 030024

Download: PDF (2813 KB)   HTML (1 KB) 
Export: BibTeX | EndNote (RIS)      
Abstract  In real-world node classification scenarios, only a few nodes are labeled and their class labels are imbalanced. In most of the existing methods, the lack of the supervision information and the imbalance of node classes are not taken into account at the same time, and the improvement of node classification performance cannot be guaranteed. Therefore, an imbalanced node classification algorithm based on self-supervised learning is proposed. Firstly, different views of the original graph are generated through graph data augmentation. Then, node representations are learned by maximizing the consistency of node representations across views using self-supervised learning. The supervised information is expanded and the expressive ability of nodes is enhanced by self-supervised learning. In addition, a semantic constraint loss is designed to ensure semantic consistency in graph data augmentation along with cross-entropy loss and self-supervised contrastive loss. Experimental results on three real graph datasets show that the proposed algorithm achieves better performance on solving the imbalanced node classification problem.
Key wordsSelf-Supervised Learning      Imbalanced Node Classification      Graph Neural Network      Data Augmentation      Semantic Constraint Loss     
Received: 23 August 2022     
ZTFLH: TP 391  
Fund:National Natural Science Foundation of China(No.61976184,62272285)
Corresponding Authors: LIANG Jiye, Ph.D., professor. His research interests include data mining, machine learning, big data analysis and artificial intelligence.   
About author:: CUI Caixia, Ph.D. candidate. Her research interests include data mining and machine learning. WANG Jie, Ph.D., lecturer. His research interests include data mining and machine learning.PANG Tianjie, master, associate profe-ssor. His research interests include data mi-ning and machine learning.
Service
E-mail this article
Add to my bookshelf
Add to citation manager
E-mail Alert
RSS
Articles by authors
CUI Caixia
WANG Jie
PANG Tianjie
LIANG Jiye
Cite this article:   
CUI Caixia,WANG Jie,PANG Tianjie等. Imbalanced Node Classification Algorithm Based on Self-Supervised Learning[J]. Pattern Recognition and Artificial Intelligence, 2022, 35(11): 955-964.
URL:  
http://manu46.magtech.com.cn/Jweb_prai/EN/10.16451/j.cnki.issn1003-6059.202211001      OR     http://manu46.magtech.com.cn/Jweb_prai/EN/Y2022/V35/I11/955
Copyright © 2010 Editorial Office of Pattern Recognition and Artificial Intelligence
Address: No.350 Shushanhu Road, Hefei, Anhui Province, P.R. China Tel: 0551-65591176 Fax:0551-65591176 Email: bjb@iim.ac.cn
Supported by Beijing Magtech  Email:support@magtech.com.cn