Paper
16 October 2024 Discriminative relation network for imbalanced graph node classification
Yu Wang, Hengyuan Sun
Author Affiliations +
Proceedings Volume 13291, Ninth International Symposium on Advances in Electrical, Electronics, and Computer Engineering (ISAEECE 2024); 132915Q (2024) https://doi.org/10.1117/12.3033996
Event: Ninth International Symposium on Advances in Electrical, Electronics, and Computer Engineering (ISAEECE 2024), 2024, Changchun, China
Abstract
Graph-structured data is widely used in biology, chemistry, finance, and other fields. Graph neural networks can learn useful representations from graph-structured data and then various tasks, such as node classification and link prediction. In many graph classification scenarios, the number of samples in each class is usually highly unbalanced, making models to prioritize learning features from the majority class, neglecting minority classes. Traditional solutions to address graph imbalance often involve oversampling, potentially resulting in issues like overfitting and noise introduction. Meanwhile, they fail to leverage the valuable features and structural information in unlabeled data. To solve this problem, this paper designs a framework different from traditional sampling, which first uses an iterative classification algorithm to obtain high-confidence pseudo-labels for unlabeled nodes, and then utilizes a relational network to learn the feature differences between samples, making full use of the information of unlabeled data to effectively generate decision boundaries. Training and learning from three graph imbalance datasets, our experiments demonstrate that the proposed approach outperforms all baseline methods.
(2024) Published by SPIE. Downloading of the abstract is permitted for personal use only.
Yu Wang and Hengyuan Sun "Discriminative relation network for imbalanced graph node classification", Proc. SPIE 13291, Ninth International Symposium on Advances in Electrical, Electronics, and Computer Engineering (ISAEECE 2024), 132915Q (16 October 2024); https://doi.org/10.1117/12.3033996
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Machine learning

Data modeling

Education and training

Statistical modeling

Independent component analysis

Matrices

Overfitting

Back to Top