Skip to content
/ CoLA Public
forked from TrustAGI-Lab/CoLA

Anomaly Detection on Attributed Networks via Contrastive Self-Supervised Learning (CoLA), TNNLS-21

License

Notifications You must be signed in to change notification settings

YunyunWan/CoLA

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Anomaly Detection on Attributed Networks via Contrastive Self-Supervised Learning

This is the source code of TNNLS paper Anomaly Detection on Attributed Networks via Contrastive Self-Supervised Learning (CoLA).

The proposed framework

Requirments

This code requires the following:

  • Python>=3.7
  • PyTorch>=1.8.1
  • Numpy>=1.19.2
  • Scipy>=1.6.2
  • Scikit-learn>=0.24.1
  • Networkx>=2.5.1
  • Tqdm>=4.59.0
  • DGL==0.4.0 (Do not use the version which is newer than that!)

Running the experiments

Step 1: Anomaly Injection

This is a pre-processing step which injects anomalies into the original clean datasets. Take Cora dataset as an example:

python inject_anomaly.py --dataset cora

After anomaly injection, the disturbed datasets are saved into "dataset" folder.

Note that the disturbed ACM dataset is directly borrowed from the code of DOMINANT and do not need injection.

Step 2: Anomaly Detection

This step is to run the CoLA framework to detect anomalies in the network datasets. Take Cora dataset as an example:

python run.py --dataset cora

The hyperparameters are set to be the values reported in our paper. You can change the default values of other parameters to simulate different conditions.

Cite

If you compare with, build on, or use aspects of CoLA framework, please cite the following:

@article{liu2021anomaly,
  title={Anomaly Detection on Attributed Networks via Contrastive Self-Supervised Learning},
  author={Liu, Yixin and Li, Zhao and Pan, Shirui and Gong, Chen and Zhou, Chuan and Karypis, George},
  journal={IEEE Transactions on Neural Networks and Learning Systems},
  year={2021},
  publisher={IEEE}
}

About

Anomaly Detection on Attributed Networks via Contrastive Self-Supervised Learning (CoLA), TNNLS-21

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%