GraphMixup: Improving Class-Imbalanced Node Classification on Graphs by Self-supervised Context Prediction

06/21/2021
by   Lirong Wu, et al.
0

Recent years have witnessed great success in handling node classification tasks with Graph Neural Networks (GNNs). However, most existing GNNs are based on the assumption that node samples for different classes are balanced, while for many real-world graphs, there exists the problem of class imbalance, i.e., some classes may have much fewer samples than others. In this case, directly training a GNN classifier with raw data would under-represent samples from those minority classes and result in sub-optimal performance. This paper presents GraphMixup, a novel mixup-based framework for improving class-imbalanced node classification on graphs. However, directly performing mixup in the input space or embedding space may produce out-of-domain samples due to the extreme sparsity of minority classes; hence we construct semantic relation spaces that allows the Feature Mixup to be performed at the semantic level. Moreover, we apply two context-based self-supervised techniques to capture both local and global information in the graph structure and then propose Edge Mixup specifically for graph data. Finally, we develop a Reinforcement Mixup mechanism to adaptively determine how many samples are to be generated by mixup for those minority classes. Extensive experiments on three real-world datasets show that GraphMixup yields truly encouraging results for class-imbalanced node classification tasks.

READ FULL TEXT

page 1

page 2

page 3

page 4

research
03/16/2021

GraphSMOTE: Imbalanced Node Classification on Graphs with Graph Neural Networks

Node classification is an important research topic in graph learning. Gr...
research
03/29/2021

Graph Classification by Mixture of Diverse Experts

Graph classification is a challenging research problem in many applicati...
research
06/16/2023

GraphSHA: Synthesizing Harder Samples for Class-Imbalanced Node Classification

Class imbalance is the phenomenon that some classes have much fewer inst...
research
06/10/2022

Synthetic Over-sampling for Imbalanced Node Classification with Graph Neural Networks

In recent years, graph neural networks (GNNs) have achieved state-of-the...
research
03/18/2023

UNREAL:Unlabeled Nodes Retrieval and Labeling for Heavily-imbalanced Node Classification

Extremely skewed label distributions are common in real-world node class...
research
10/01/2022

Diving into Unified Data-Model Sparsity for Class-Imbalanced Graph Representation Learning

Even pruned by the state-of-the-art network compression methods, Graph N...
research
06/15/2022

CLNode: Curriculum Learning for Node Classification

Node classification is a fundamental graph-based task that aims to predi...

Please sign up or login with your details

Forgot password? Click here to reset