返回
A novel graph oversampling framework for node classification in class-imbalanced graphs
DOI:10.1007/s11432-023-3897-2.png)
摘要
En 中文
Graph neural network (GNN) is a promising method to analyze graphs. Most existing GNNs adopt the class-balanced assumption, which cannot deal with class-imbalanced graphs well. The oversampling technique is effective in alleviating class-imbalanced problems. However, most graph oversampling methods generate synthetic minority nodes and their edges after applying GNNs. They ignore the problem that the representations of the original and synthetic minority nodes are dominated by majority nodes caused by aggregating neighbor information through GNN before oversampling. In this paper, we propose a novel graph oversampling framework, termed distribution alignment-based oversampling for node classification in class-imbalanced graphs (named Graph-DAO). Our framework generates synthetic minority nodes before GNN to avoid the dominance of majority nodes caused by message passing in GNNs. Additionally, we introduce a distribution alignment method based on the sum-product network to learn more information about minority nodes. To our best knowledge, it is the first to use the sum-product network to solve the class-imbalanced problem in node classification. A large number of experiments on four real datasets show that our method achieves the optimal results on the node classification task for class-imbalanced graphs.
Keyword:
graph neural network
class-imbalanced graphs
sum-product network
oversampling
node classification
期刊
IF:
7.6
论文数:
4.9K
被引数:
8.9K
机构
引用论文
A systematic study of the class imbalance problem in convolutional neural networks卷积神经网络中类不平衡问题的系统研究
NEURAL NETWORKS
IF6.3
Study on strength and durability characteristics of nano-silica based blended concrete纳米sio_2基混合混凝土强度及耐久特性研究
A statistical study of the potential dependence of a transfer coefficient supports the Marcus theory

