Multitask Active Learning for Graph Anomaly Detection

February 12, 2025 · View on GitHub

Requirements

This code requires the following:

  • Python==3.8
  • PyTorch==2.0.1
  • Numpy==1.24.4
  • DGL==1.1.1+cu102

Usage

Take Cora dataset as an example:

python main.py  --dataset cora --strategy_ad medoids_spec_nent_diff --device 0 --alpha 1.25 --beta 0.5 --gamma 1 --cluster_num 24 --tau 0.95  --phi 1.25 

The hyperparameters for other datasets are reported as follows.

CoraCiteseerBlogCatalogFlickrAmazonYelpChi
τ\tau0.950.900.980.980.9990.80
α\alpha1.250.501.251.251.000.40
β\beta0.502.001.000.501.001.25
ϕ\phi1.252.001.000.501010
mm242418271020

The Amazon and YelpChi datasets can be found from GADBench.