Skip to content

Latest commit

 

History

History
171 lines (154 loc) · 9.39 KB

README.md

File metadata and controls

171 lines (154 loc) · 9.39 KB
Raw

Introduction

This github provides the implementation used for the results in "Lipschitz Continuous Autoencoders in Application to Anomaly Detection" accepted for AISTATS 2020 [paper][supplementary].

Directory tree

.
├── data_celeba
│   └── readme_celeba.md
├── Datapreprocessing_CelebA.ipynb
├── README.md
└── src
    ├── architecture.py
    ├── base.py
    ├── configuration.py
    ├── dataset
    │   ├── celeba.py
    │   ├── fmnist.py
    │   ├── kdd99.py
    │   ├── main.py
    │   └── mnist.py
    ├── LCAE_trainer.py
    ├── main.py
    ├── other_methods
    │   ├── alad
    │   │   ├── alad_trainer.py
    │   │   ├── architecture.py
    │   │   └── run_alad.py
    │   └── svdd
    │       ├── architecture.py
    │       ├── deepSVDD_trainer.py
    │       └── run_svdd.py
    ├── run_lcae.py
    └── utils.py

Commands to run experiments

We provide commands to run experiments in manuscript and supplementary material. Explanations about arguments are attached in 'Explanation about argments' in this document. Before running experiments on CelebA, please download and pre-process dataset using 'Datapreprocessing_CelebA.ipynb'.

(1) Proposed method

  1. (KDD99) To run experiments on Table 4 in the manuscript, please execute the following command at the src folder.
# KDD99 with uncontaminated training dataset (0%)
python3 main.py 'lcae' 'kdd99' -n 20 -e 200 -m 0.0 -l 10.0 -C 0.7 -c 0 -g 0
# KDD99 with contaminated training dataset (5%)
python3 main.py 'lcae' 'kdd99' -n 20 -e 200 -m 0.0 -l 10.0 -C 0.7 -c 5 -g 0

To run experiments on Tables 2 and 3 in the manuscript and Tables 5 and 6 in the supplementary material, please execute the above command after setting n to be 10 and varying m, l, and C.

  1. (MNIST and Fashion-MNIST) To run experiments on Tables 7 and 8 in the supplementary material, please execute the following command at the src folder.
# MNIST with uncontaminated training dataset (0%)
python3 main.py 'lcae' 'mnist'  -n 10 -e 50 -m 2.0 -l 2.0 -C 0.8 -c 0 -g 0 -N 0
# MNIST with contaminated training dataset (5%)
python3 main.py 'lcae' 'mnist' -n 10 -e 50 -m 2.0 -l 2.0 -C 0.8 -c 5 -g 0 -N 0
# Fashion-MNIST with uncontaminated training dataset (0%)
python3 main.py 'lcae' 'fmnist' -n 10 -e 50 -m 2.0 -l 2.0 -C 0.8 -c 0 -g 0 -N 0
# Fashion-MNIST with contaminated training dataset (5%)
python3 main.py 'lcae' 'fmnist' -n 10 -e 50 -m 2.0 -l 2.0 -C 0.8 -c 5 -g 0 -N 0
  1. (CelebA) To run experiments on Table 4 in the manuscript, please execute the following command at the src folder.
# CelebA (glasses) with uncontaminated training dataset (0%)
python3 main.py 'lcae' 'celeba' -n 5 -e 100 -m 2.0 -l 2.0 -C 0.8 -c 0 -g 0 -N -1 -a 15
# CelebA (glasses) with contaminated training dataset (5%)
python3 main.py 'lcae' 'celeba' -n 5 -e 100 -m 2.0 -l 2.0 -C 0.8 -c 5 -g 0 -N -1 -a 15

(2) Other methods

  1. (KDD99) To run experiments on Table 4 in the manuscript, please execute the following command at the src folder.
# Deep SVDD: KDD99 with uncontaminated training dataset (0%)
python3 main.py 'svdd' 'kdd99' -n 20 -e 200 -c 0 -g 0
# Deep SVDD: KDD99 with contaminated training dataset (5%)
python3 main.py 'svdd' 'kdd99' -n 20 -e 200 -c 5 -g 0
# ALAD: KDD99 with uncontaminated training dataset (0%)
python3 main.py 'alad' 'kdd99' -n 20 -e 200 -c 0 -g 0
# ALAD: KDD99 with contaminated training dataset (5%)
python3 main.py 'alad' 'kdd99' -n 20 -e 200 -c 5 -g 0
  1. (MNIST and Fashion-MNIST) To run experiments on Tables 7 and 8 in the supplementary material, please execute the following command at the src folder.
# Deep SVDD: MNIST with uncontaminated training dataset (0%)
python3 main.py 'svdd' 'mnist' -n 10 -e 50 -c 0 -g 0 -N 0
# Deep SVDD: MNIST with contaminated training dataset (5%)
python3 main.py 'svdd' 'mnist' -n 10 -e 50 -c 5 -g 0 -N 0
# ALAD: MNIST with uncontaminated training dataset (0%)
python3 main.py 'alad' 'mnist' -n 10 -e 50 -c 0 -g 0 -N 0
# ALAD: MNIST with contaminated training dataset (5%)
python3 main.py 'alad' 'mnist' -n 10 -e 50 -c 5 -g 0 -N 0
# Deep SVDD: Fashion-MNIST with uncontaminated training dataset (0%) 
python3 main.py 'svdd' 'fmnist' -n 10 -e 50 -c 0 -g 0 -N 0
# Deep SVDD: Fashion-MNIST with contaminated training dataset (5%)
python3 main.py 'svdd' 'fmnist' -n 10 -e 50 -c 5 -g 0 -N 0
# ALAD: Fashion-MNIST with uncontaminated training dataset (0%)
python3 main.py 'alad' 'fmnist' -n 10 -e 50 -c 0 -g 0 -N 0
# ALAD: Fashion-MNIST with contaminated training dataset (5%)
python3 main.py 'alad' 'fmnist' -n 10 -e 50 -c 5 -g 0 -N 0
  1. (CelebA) To run experiments on Table 4 in the manuscript, please execute the following command at the src folder.
# Deep SVDD: CelebA (glasses) with uncontaminated training dataset (0%)
python3 main.py 'svdd' 'celeba' -n 5 -e 100 -c 0 -g 0 -N -1 --a 15
# Deep SVDD: CelebA (glasses) with contaminated training dataset (5%)
python3 main.py 'svdd' 'celeba' -n 5 -e 100 -c 5 -g 0 -N -1 --a 15
# ALAD: CelebA (glasses) with uncontaminated training dataset (0%)
python3 main.py 'alad' 'celeba' -n 5 -e 100 -c 0 -g 0 -N -1 --a 15
# ALAD: CelebA (glasses) with contaminated training dataset (5%)
python3 main.py 'alad' 'celeba' -n 5 -e 100 -c 5 -g 0 -N -1 --a 15

Explanations about argments

a (or attribute): attribute class in experiments with CelebA. Default value is 0.
C (or lipschitzconstant): Lipschitz constant ('K' in the equation (2) in the manuscript) in experiments with LCAE. Default value is 0.95.
c (or contamratio): contaminated ratio in training set. Default value is 0. Possible choices are 0 and 5.
d (or decide): number or list of experiment. Possible choices are 'int' and 'list'.
dataset: the name of the dataset you want to run the experiment on. Possible choices are 'kdd99', 'mnist', 'fmnist', and 'celeba'.
degree: degree of L-p norm in experiments with ALAD. Default value is 1.
e (or epoch): number of epochs. Default value is 50.
enable_dzz: enable dzz discriminator in experiments with ALAD. Default value is True.
enable_early_stop: enable early_stopping in experiments with ALAD. Default value is True.
enable_sm: enable TF summaries in experiments with ALAD. Default value is True.
g (or gpu): which gpu to use. Default value is 0.
m: mode/method for discriminator loss in experiments with ALAD. Possible choices are 'cross-e' and 'fm'. Default value is 'fm'.
m (or mmdweight): mmd loss weight ('lambda' in the equation (2) in the manuscript) in experiments with LCAE. Default value is 0.0.
model: the model name of the example you want to run. Possible choices are 'lcae', 'svdd', and 'alad'.
N (or normalclass): normal class in experiments on MNIST, Fashion-MNIST, and CelebA. Possible choices are 0, 1, ..., 9 in MNIST and Fashion-MNIST, and-1 and 1 in CelebA. Default value is 0.
n (or number): number or list of experiment. Default value is 10.
sn: enable spectral_norm in experiments with ALAD. Default value is True.
t (or tsne): tsne for original/reconstructed data in experiments with LCAE. Default value is False.
l (or lipschitzweight): Lipschitz loss weight ('phi' in the equation (2) in the manuscript) in experiments with LCAE. Default value is 0.0.
w: weight in experiments with ALAD. Default value is 0.1.

Reproducibility

The results of the proposed method and Deep SVDD are reproducible, while those of ALAD are not. For Deep SVDD and ALAD, we adopted and modified the codes published by [1] and [2], respectively. Initially, both codes were not reproducible when a GPU was used, so we revised the code to ensure reproducibility. The modified Deep SVDD implementation yields consistent results in the system environment detailed in the 'Dependency' section. In contrast, the code from the ALAD work, which is built on an older version of TensorFlow, does not achieve reproducibility. We attempted to minimize oscillations as much as possible. For details about the reproducibility issue in TensorFlow, please refer https://github.com/NVIDIA/tensorflow-determinism.

Dependency

argparse 1.1
imageio 2.3.0
ipykernel 4.8.2
json 2.0.9
logging 0.5.1.2
matplotlib 3.1.2
numpy 1.16.1
pandas 0.25.3
pickleshare 0.7.4
PIL 5.1.0
python 3.6.5
skimage 0.16.2
sklearn 0.20.2
tensorflow 1.9.0
torch 1.0.1.post2
torchvision 0.2.2
urllib3 1.22

References for the implementation

[1] Ruff, Lukas, et al. "Deep one-class classification." International conference on machine learning. 2018. URL: https://github.com/lukasruff/Deep-SVDD-PyTorch
[2] Zenati, Houssam, et al. "Adversarially learned anomaly detection." 2018 IEEE International Conference on Data Mining (ICDM). IEEE, 2018. URL: https://github.com/houssamzenati/Adversarially-Learned-Anomaly-Detection