Rethinking Fair Graph Neural Networks from Re-balancing

July 15, 2024 ยท View on GitHub

This is the official implementation of the following paper:

Rethinking Fair Graph Neural Networks from Re-balancing (SIGKDD'2024)

Table of Contents

Dependencies

  • python>=3.7
  • torch==2.0.1
  • torch-geometric==2.3.1
  • torch-scatter==2.1.1
  • numpy==1.24.4
  • scikit-learn==1.3.0

Datasets

We conduct experiments on three widely used real-world datasets, namely German Credit, Bail, and Credit Defaulter. The detailed information of the datasets is as follows. Please unzip datasets in ./dataset before running the model.

DatasetGermanBailCredit
# Nodes1,00018,87630,000
# Edges22,242321,308152,377
# Attributes271813
Sens.GenderRaceAge
LabelCredit statusBail decisionFuture default

Running

The run.sh includes details to reproduce experimental results in the paper:

bash run.sh

Acknowledgement

Our code is based on the FairVGNN (Improving fairness in graph neural networks via mitigating sensitive attribute leakage) and GraphENS (Graphens: Neighbor-aware ego network synthesis for class-imbalanced node classification).