Hybrid sampling-based contrastive learning for imbalanced node classification

Caixia Cui

doi:10.1007/s13042-022-01677-6

Hybrid sampling-based contrastive learning for imbalanced node classification

2022-10-24

7

PRE

AI

DOI：10.1007/s13042-022-01677-6

Original

Original request for help

Save

Abstract

En 中文

Imbalanced node classification is a vital task because it widely exists in many real-world applications, such as financial fraud detection, anti-money laundering, drug reaction prediction and so on. However, many recent methods are for balanced graph-structured datasets, and do not perform well on imbalanced data. Therefore, we propose a hybrid sampling-based contrastive learning method (HSCL) for imbalanced node classification to alleviate this problem. The core of our method is to adopt the hybrid sampling method in contrastive learning, that is, undersampling majority classes and oversampling minority classes, to achieve a balance of samples from different classes in contrastive learning and thus obtain a discriminative representation. HSCL has been evaluated extensively on five real-world data sets. Experimental results show that the proposed method obtains better performance than other state-of-the-art methods.

Keywords:

Imbalanced node classification

Contrastive learning

Hybrid sampling

Graph convolutional network

Hybrid sampling-based contrastive learning for imbalanced node classification

Abstract

Journal

Organization