HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Pick and Choose: A GNN-based Imbalanced Learning Approach for Fraud Detection

{Qing He Hao Yang Jinghua Feng Jianfeng Chi Zidi Qin Xiang Ao Yang Liu1}

Abstract

Graph-based fraud detection approaches have escalated lots of attention recently due to the abundant relational information ofgraph-structured data, which may be beneficial for the detection of fraudsters. However, the GNN-based algorithms could fare poorly when the label distribution of nodes is heavily skewed, and it is common in sensitive areas such as financial fraud, etc. To remedy the class imbalance problem of graph-based fraud detection, we propose a Pick and Choose Graph Neural Network (PC-GNN for short) for imbalanced supervised learning on graphs. First, nodes and edges are picked with a devised label-balanced sampler to construct sub-graphs for mini-batch training. Next, for each node in the sub-graph, the neighbor candidates are chosen by a proposed neighborhood sampler. Finally, information from the selected neighbors and different relations are aggregated to obtain the final representation of a target node. Experiments on both benchmark and real-world graph-based fraud detection tasks demonstrate that PCGNN apparently outperforms state-of-the-art baselines.

Benchmarks

BenchmarkMethodologyMetrics
fraud-detection-on-amazon-fraudPC-GNN
AUC-ROC: 95.86
Averaged Precision: 85.49
fraud-detection-on-yelp-fraudPC-GNN
AUC-ROC: 79.87
Averaged Precision: 48.10
node-classification-on-amazon-fraudPC-GNN
AUC-ROC: 95.86
node-classification-on-yelpchiPC-GNN
AUC-ROC: 79.87

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp