A PyTorch implementation of "ANEMONE: Graph Anomaly Detection with Multi-Scale Contrastive Learning", CIKM-21

Last update: Dec 14, 2022

Overview

ANEMONE

A PyTorch implementation of "ANEMONE: Graph Anomaly Detection with Multi-Scale Contrastive Learning", CIKM-21

Dependencies

python==3.6.1
dgl==0.4.1
matplotlib==3.3.4
networkx==2.5
numpy==1.19.2
pyparsing==2.4.7
scikit-learn==0.24.1
scipy==1.5.2
sklearn==0.24.1
torch==1.8.1

To install all dependencies:

pip install -r requirements.txt

Usage

Anomalies have been injected into three datasets under the ./dataset directory. Please refer to GRAND-Lab/CoLA for graph anomaly injection.

To train and evaluate on Cora:

python run.py --expid 1 --device cuda:0 --runs 1 --alpha 0.8

To train and evaluate on Citeseer:

python run.py --dataset citeseer --expid 2 --device cuda:0 --runs 1 --alpha 0.6

To train and evaluate on Pubmed:

python run.py --dataset pubmed --expid 3 --device cuda:0 --runs 1 --alpha 0.8 --negsamp_ratio_patch 6 --negsamp_ratio_context 1

Citation

If you use our code in your research, please cite the following article:

@inproceedings{jin2021anomaly,
  title={ANEMONE: Graph Anomaly Detection with Multi-Scale Contrastive Learning},
  author={Ming Jin and Yixin Liu and Yu Zheng and Lianhua Chi and Yuan-Fang Li and Shirui Pan},
  booktitle={Proceedings of the 30th ACM International Conference on Information \& Knowledge Management},
  year={2021}
}

A PyTorch implementation of "ANEMONE: Graph Anomaly Detection with Multi-Scale Contrastive Learning", CIKM-21

Related tags

Overview

ANEMONE

Dependencies

Usage

Citation

Owner

Graph Analysis & Deep Learning Laboratory, GRAND

Pytorch implementation of Hinton's Dynamic Routing Between Capsules

Deep Learning to Improve Breast Cancer Detection on Screening Mammography

A fast Protein Chain / Ligand Extractor and organizer.

Real-time Object Detection for Streaming Perception, CVPR 2022

Code release for NeurIPS 2020 paper "Co-Tuning for Transfer Learning"

A learning-based data collection tool for human segmentation

Code for the ICCV 2021 Workshop paper: A Unified Efficient Pyramid Transformer for Semantic Segmentation.

Code for all the Advent of Code'21 challenges mostly written in python

Multi-Template Mouse Brain MRI Atlas (MBMA): both in-vivo and ex-vivo

Deep Learning agent of Starcraft2, similar to AlphaStar of DeepMind except size of network.

Codebase for testing whether hidden states of neural networks encode discrete structures.

Curated list of awesome GAN applications and demo

Syllabic Quantity Patterns as Rhythmic Features for Latin Authorship Attribution

Personalized Federated Learning using Pytorch (pFedMe)

Deep learning PyTorch library for time series forecasting, classification, and anomaly detection

Package to compute Mauve, a similarity score between neural text and human text. Install with `pip install mauve-text`.

Count the MACs / FLOPs of your PyTorch model.

Model that predicts the probability of a Twitter user being anti-vaccination.

Cascaded Deep Video Deblurring Using Temporal Sharpness Prior and Non-local Spatial-Temporal Similarity

Dynamic Token Normalization Improves Vision Transformers