Skip to content

Scripts and notebooks to reproduce the experiments and analyses of the paper Holger Trittenbach and Klemens Böhm. "One-Class Active Learning for Outlier Detection with Multiple Subspaces." CIKM 2019

License

Notifications You must be signed in to change notification settings

holtri/subsvdd-evaluation

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

SubSVDD Evaluation

This repository contains scripts and notebooks to reproduce the experiments and analyses of the paper

Holger Trittenbach and Klemens Böhm. 2019. One-Class Active Learning for Outlier Detection with Multiple Subspaces. In The 28th ACM International Conference on Information and Knowledge Management (CIKM ’19), November 3--7, 2019, Beijing, China. ACM, New York, NY, USA, 10 pages. https://doi.org/10.1145/3357384.3357873

For more information about this research project, see also the SubSVDD project website. For a general overview and a benchmark on one-class active learning see the OCAL project website.

Quick Start

The analysis and main results of the experiments can be found under notebooks:

  • illustration_outlier-asymmetry.ipynb: Figure 1 and Figure 2
  • illustration_page-data-example.ipynb: Figure 3b
  • R-experiment-evaluation
    • evaluation-part1.ipynb: Table 1 and Figure 5
    • evaluation-part2.ipynb: Figure 6
    • subspace_heatmap.ipynb: Figure 3a

To execute the notebooks, make sure you follow the setup section, and download the raw results into data/output/.

Prerequisites

The experiments are implemented in Julia, some of the evaluation notebooks are written in R. This repository contains code to setup the experiments, to execute them, and to analyze the results. The one-class classifiers (SubSVDD and the competitors SSAD and SVDDneg) and active learning methods are implemented in two separate Julia packages: SVDD.jl and OneClassActiveLearning.jl.

Setup

Just clone the repo.

$ git clone https://github.com/holtri/subsvdd-evaluation.git
  • Experiments require Julia 1.1, requirements are defined in Manifest.toml. To instantiate, start julia in the subsvdd-evaluation directory with julia --project=. and run julia> ]instantiate. See Julia documentation for general information on how to setup this project.
  • Notebooks require
    • Julia 1.1
    • R 3.6: tidyverse, assertthat, xtables

Repo Overview

  • data
    • input
      • raw: unprocessed data files
      • processed: output directory of preprocess_data.jl
    • output: output directory of experiments; generate_experiments.jl creates the folder structure and experiments; run_experiments.jl writes results and log files
  • notebooks: jupyter notebooks to analyze experimental results
    • illustration_outlier-asymmetry.ipynb: Figure 1 and Figure 2
    • illustration_page-data-example.ipynb: Figure 3b
    • R-experiment-evaluation
      • evaluation-part1.ipynb: Table 1 and Figure 5
      • evaluation-part2.ipynb: Figure 6
      • subspace_heatmap.ipynb: Figure 3a
  • scripts
    • config: configuration files for experiments
      • config.jl: high-level configuration
      • baslines.jl, largesubspaces.jl, smallsubspaces-wanglimit.jl: experiment configs
      • v-comparison.jl: experiment config for parameter comparison
    • check_pkg.jl: check package versions (for distributed execution)
    • preprocess_data.jl: preprocess data files into common format
    • generate_experiments.jl: generates experiments
    • reduce_results.jl: reduces result json files to single result csv
    • run_experiments: executes experiments

Overview

Each step of the experiments can be reproduced, from the raw data files to the final plots that are presented in the paper. The experiment is a pipeline of several dependent processing steps. Each of the steps can be executed standalone, and takes a well-defined input, and produces a specified output. The Section Experiment Pipeline describes each of the process steps.

Running the benchmark is compute intensive and takes many CPU hours. Therefore, we also provide the results to download (51 MB). This allows to analyze the results in the notebooks without having to run the whole pipeline.

The code is licensed under a MIT License and the result data under a Creative Commons Attribution 4.0 International License. If you use this code or data set in your scientific work, please reference the companion paper.

Experiment Pipeline

The benchmark pipeline uses config files to set paths and experiment parameters. There are two types of config files:

  • scripts/config.jl: this config defines high-level information on the experiment, such as where the data files are located, and log levels.
  • scripts/<baslines|largesubspaces|smallsubspaces-wanglimit|v-comparison>.jl: These config files define the experimental grid, including the data sets, classifiers, and active-learning strategies.
  1. Data Preprocessing: The preprocessing step transforms publicly available benchmark data sets into a common csv format, and subsamples large data sets to 1000 observations.

    • Input: Download semantic.tar.gz and literature.tar.gz containing the .arff files from the DAMI benchmark repository and extract into data/input/raw/<data set> (e.g. data/input/raw/Annthyroid/).
    • Execution:
       $ julia --project="." preprocess_data.jl <config.jl>
    • Output: .csv files in data/input/processed/noise

    We also provide our preprocessed data to download (3 MB).

  2. Generate Experiments: This step creates a set of experiments. Each experiment in this set is a specific combination of

    • data set path (e.g., "data/input/Annthyroid/Annthyroid_withoutdupl_norm_05_v01_r01.csv")
    • initial pool strategy (e.g., "Pu")
    • split strategy (e.g., "Sf")
    • model (e.g., SubSVDD)
    • init strategy (e.g., SimpleCombinedStrategy)
    • query strategy (e.g., DecisionBoundaryPQs)
    • parameters (e.g., number of active learning iterations)

    These specific combinations are created as a cross product of the vectors in the config file that is passed as an argument.

    • Input: Full path to config file <config_file.jl> (e.g., config/config_evaluation_part1.jl), preprocessed data files
    • Execution:
     $ julia --project="." generate_experiments.jl <config_file.jl>
    • Output:
      • Creates an experiment directory with the naming <exp_name>. The directories created contains several items:
        • log directory: skeleton for experiment logs (one file per experiment), and worker logs (one file per worker)
        • results directory: skeleton for result files
        • experiments.jser: this contains a serialized Julia Array with experiments. Each experiment is a Dict that contains the specific combination. Each experiment can be identified by a unique hash value.
        • experiment_hashes: file that contains the hash values of the experiments stored in experiments.jser
        • generate_experiments.jl: a copy of the file that generated the experiments
        • config.jl: a copy of the config file used to generate the experiments
  3. Run Experiments: This step executes the experiments created in Step 2. Each experiment is executed on a worker. In the default configuration, a worker is one process on the localhost. For distributed workers, see Section Infrastructure and Parallelization. A worker takes one specific configuration, runs the active learning experiment, and writes result and log files.

  • Input: Generated experiments from step 2.
  • Execution:
   $ julia --project="." run_experiments.jl /full/path/to/ocal-evaluation/scripts/config.jl
  • Output: The output files are named by the experiment hash
    • Experiment log (e.g., data/output/evaluation_part1/02-subsvdd-largesubspaces/results/10121309769577703138.log)
    • Result .json file (e.g., data/output/evaluation_part1/02-subsvdd-largesubspaces/results/Annthyroid/Annthyroid_withoutdupl_norm_05_v01_nnoise-mu=0.0_s=0.01__SubspaceQs{DecisionBoundaryPQs}_SubSVDD_10121309769577703138.json)
  1. Reduce Results: Merge of an experiment directory into one .csv by using summary statistics

    • Input: Full path to finished experiments.
    • Execution:
       $ julia --project="." reduce_results.jl </full/path/to/data/output>
    • Output: A result csv file, data/output/evaluation-part1.csv.
  2. Analyze Results: jupyter notebooks in the notebooksdirectory to analyze the reduced .csv, and individual .json files

Infrastructure and Parallelization

Step 3 Run Experiments can be parallelized over several workers. In general, one can use any ClusterManager. In this case, the node that executes run_experiments.jl is the driver node. The driver node loads the experiments.jser, and initiates a function call for each experiment on one of the workers via pmap.

Authors

We welcome contributions and bug reports.

This package is developed and maintained by Holger Trittenbach

About

Scripts and notebooks to reproduce the experiments and analyses of the paper Holger Trittenbach and Klemens Böhm. "One-Class Active Learning for Outlier Detection with Multiple Subspaces." CIKM 2019

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published