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

Real-time LIDAR-based Urban Road and Sidewalk detection for Autonomous Vehicles 🚗

Code for binary and multiclass model change active learning, with spectral truncation implementation.

AquaTimer - Programmable Timer for Aquariums based on ATtiny414/814/1614

The 3rd place solution for competition

Official code repository for the work: "The Implicit Values of A Good Hand Shake: Handheld Multi-Frame Neural Depth Refinement"

bio_inspired_min_nets_improve_the_performance_and_robustness_of_deep_networks

[CVPR22] Official codebase of Semantic Segmentation by Early Region Proxy.

Run object detection model on the Raspberry Pi

Fine-grained Control of Image Caption Generation with Abstract Scene Graphs

Code for our ALiBi method for transformer language models.

A torch.Tensor-like DataFrame library supporting multiple execution runtimes and Arrow as a common memory format

Lecture materials for Cornell CS5785 Applied Machine Learning (Fall 2021)

Graph Robustness Benchmark: A scalable, unified, modular, and reproducible benchmark for evaluating the adversarial robustness of Graph Machine Learning.

Towards End-to-end Video-based Eye Tracking

Methods to get the probability of a changepoint in a time series.

Official Pytorch Implementation of Relational Self-Attention: What's Missing in Attention for Video Understanding

Meta-TTS: Meta-Learning for Few-shot SpeakerAdaptive Text-to-Speech

PyTorch IPFS Dataset

Arxiv harvester - Poor man's simple harvester for arXiv resources

DeepMetaHandles: Learning Deformation Meta-Handles of 3D Meshes with Biharmonic Coordinates