UM  > Faculty of Science and Technology
Residential Collegefalse
Status已發表Published
Neighbor Distribution Learning for Minority Class Augmentation
Zhou, Mengting; Gong, Zhiguo
2024-12
Source PublicationIEEE Transactions on Knowledge and Data Engineering
ISSN1041-4347
Volume36Issue:12Pages:8901-8913
Abstract

Graph Neural Networks (GNNs) have achieved remarkable success in graph-based tasks. However, learning unbiased node representations under class-imbalanced training data remains challenging. Existing solutions may face overfitting due to extensive reuse of those limited labeled data in minority classes. Furthermore, many works address the class-imbalanced issue based on the embeddings generated from the biased GNNs, which make models intrinsically biased towards majority classes. In this paper, we propose a novel data augmentation strategy GraphGLS for semi-supervised class-imbalanced node classification, which aims to select informative unlabeled nodes to augment minority classes with consideration of both global and local information. Specifically, we first design a Global Selection module to learn global information (pseudo-labels) for unlabeled nodes and then select potential ones from them for minority classes. The Local Selection module further conducts filtering over those potential nodes by comparing their neighbor distributions with minority classes. To achieve this, we further design a neighbor distribution auto-encoder to learn a robust node-level neighbor distribution for each node. Then, we define class-level neighbor distribution to capture the overall neighbor characteristics of nodes within the same class. We conduct extensive experiments on multiple datasets, and the results demonstrate the superiority of GraphGLS over state-of-the-art baselines.

KeywordTraining Topology Graph Neural Networks Data Models Accuracy Task Analysis Image Color Analysis Class-imbalanced Learning Data Mining Node Classification
DOI10.1109/TKDE.2024.3447014
URLView the original
Indexed BySCIE
Language英語English
WOS Research AreaComputer Science ; Engineering
WOS SubjectComputer Science, Artificial Intelligence ; Computer Science, Information Systems ; Engineering, Electrical & Electronic
WOS IDWOS:001354743800109
PublisherIEEE COMPUTER SOC, 10662 LOS VAQUEROS CIRCLE, PO BOX 3014, LOS ALAMITOS, CA 90720-1314
Scopus ID2-s2.0-85201778128
Fulltext Access
Citation statistics
Document TypeJournal article
CollectionFaculty of Science and Technology
THE STATE KEY LABORATORY OF INTERNET OF THINGS FOR SMART CITY (UNIVERSITY OF MACAU)
DEPARTMENT OF COMPUTER AND INFORMATION SCIENCE
Corresponding AuthorGong, Zhiguo
AffiliationState Key Laboratory of Internet of Things for Smart City and Department of Computer and Information Science, University of Macau, Macao
First Author AffilicationUniversity of Macau
Corresponding Author AffilicationUniversity of Macau
Recommended Citation
GB/T 7714
Zhou, Mengting,Gong, Zhiguo. Neighbor Distribution Learning for Minority Class Augmentation[J]. IEEE Transactions on Knowledge and Data Engineering, 2024, 36(12), 8901-8913.
APA Zhou, Mengting., & Gong, Zhiguo (2024). Neighbor Distribution Learning for Minority Class Augmentation. IEEE Transactions on Knowledge and Data Engineering, 36(12), 8901-8913.
MLA Zhou, Mengting,et al."Neighbor Distribution Learning for Minority Class Augmentation".IEEE Transactions on Knowledge and Data Engineering 36.12(2024):8901-8913.
Files in This Item:
There are no files associated with this item.
Related Services
Recommend this item
Bookmark
Usage statistics
Export to Endnote
Google Scholar
Similar articles in Google Scholar
[Zhou, Mengting]'s Articles
[Gong, Zhiguo]'s Articles
Baidu academic
Similar articles in Baidu academic
[Zhou, Mengting]'s Articles
[Gong, Zhiguo]'s Articles
Bing Scholar
Similar articles in Bing Scholar
[Zhou, Mengting]'s Articles
[Gong, Zhiguo]'s Articles
Terms of Use
No data!
Social Bookmark/Share
All comments (0)
No comment.
 

Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.