This repository provides a PyTorch implementation of MADGA, which transforms the unsupervised anomaly detection to graph alignment problem.
We test our method for five publicly processed datasets, e.g., SWaT
, WADI
, PSM
, MSL
, and SMD
.
SWaT
WADI
PSM
is released inRANSynCoders
.MSL
is released inGDN
.SMD
is released inOmniAnomaly
.
mkdir Dataset
cd Dataset
mkdir input
Download the dataset in Data/input
.
- train for MADGA For example, training for WADI
sh script/run_WADI.sh
- train for
DeepSVDD
,DeepSAD
,DROCC
, andALOCC
.
python3 baseline_train.py --name SWaT --model DeepSVDD
- train for
USAD
andDAGMM
We report the results by the implementations in the following links:USAD
andDAGMM
We provide the pretained model of MADGA.
For example, testing for WADI
sh script/test_WADI.sh
If you find this paper or repository helpful, please cite our paper. Thanks a lot~