CDCGAN: Class Distribution-aware Conditional GAN-based minority augmentation for imbalanced node classification

Bojia Liu

doi:10.1016/j.neunet.2024.106933

CDCGAN: Class Distribution-aware Conditional GAN-based minority augmentation for imbalanced node classification

2025-03-01

0

PRE

AI

DOI：10.1016/j.neunet.2024.106933

原文链接

原文求助

摘要

En 中文

Node classification is a fundamental task of Graph Neural Networks (GNNs). However, GNN models tend to suffer from the class imbalance problem which deteriorates the representation ability of minority classes, thus leading to unappealing classification performance. The most straightforward and effective solution is to augment the minority samples for balancing the representations of majority and minority classes. Previous methods leverage a limited number of labeled nodes to generate new samples, without considering the overall class characteristics and failing to reflect the underlying class distributions. Besides, they often yield less distinguishable nodes that cannot represent their original classes well, because they may incorporate useless information from other classes to form node representations. To address this issue, we propose a C lass D istribution-aware C onditional G enerative A dversarial N etwork ( CDCGAN ) to generate diverse and distinguishable minority nodes based on their class distribution characteristics. Specifically, we extract the node embeddings and class distributions while preserving the topology and attribute information, thus capturing the overall class characteristics. Then, the obtained class distributions are used to design a conditional generator, which incorporates nonlinear transformations to generate diverse minority nodes and leverages adversarial learning to maintain intrinsic class distribution characteristics. At last, to ensure the distinguishability of node representations, a unique discriminator is implemented to jointly discriminate and classify nodes of the augmented graph. Extensive experiments conducted on six datasets demonstrate that the proposed CDCGAN outperforms the state-of-the-art methods on widely used evaluation metrics. The source code is available at https://github.com/Crystal-LiuBojia/CDCGAN.

Keyword:

Graph neural networks

Generative adversarial networks

Imbalanced node classification

CDCGAN: Class Distribution-aware Conditional GAN-based minority augmentation for imbalanced node classification

摘要

期刊

机构

引用论文