Skip to content

README.md

Latest commit

 

History

History
103 lines (82 loc) · 4.76 KB

README.md

File metadata and controls

103 lines (82 loc) · 4.76 KB

ROAST (Risk-Aware Outlier Exposure Selective Training)

This is the code repository for the ROAST framework, which improves Anomaly Detector (AD) performance against evasion attacks in safety-critical domains like healthcare. Link to paper: [To be added once arXiv link is avaialable]

Repository Overview

This repository unifies and orchestrates multiple model pipelines across three distinct time-series datasets used for the ROAST experiments:

  • OhioT1DM: Continuous Glucose Monitoring (CGM) forecasting.
  • MIMIC: Clinical time-series analysis (MIMIC-IV).
  • PhysioNetCinC: Sepsis Early Prediction on ICU Data.

The project features automated data preprocessing, isolated virtual environments, unified adversarial attack integration (e.g., URET and FGSM), risk profiling, and selective training for anomaly detectors.

Dataset Acquisition

1. OhioT1DM

The OhioT1DM dataset is protected under an NDA and therefore we cannot disclose it publicly. However, it can be obtained from Ohio University by following the instructions listed here.

  • Action Required: Place the dataset at the following path: ./OhioT1DM/data/raw/

2. MIMIC-IV

The MIMIC dataset access is restricted and subject to meeting the following requirements:

3. PhysioNetCinC

The PhysioNetCinC sepsis prediction dataset is publicly available and can be accessed from the PhysioNet website, as long as you conform to the terms of the Creative Commons Attribution 4.0 International Public License.

  • Download the dataset using the following command: 
    wget --user [YOUR_PHYSIONET_USERNAME] --ask-password -r -N -c -np -nH --cut-dirs=3 --reject "index.html*" https://physionet.org/files/challenge-2019/1.0.0/training/
  • Action Required: Place the dataset at the following path: ./PhysioNetCinC/input/

Project Setup

Run the setup.sh script to build isolated Python virtual environments for each respective project and install their dependencies:

bash setup.sh

Configuration

Control the entire execution pipeline through the central pipeline_config.yml configuration point. Here, you can toggle pipeline steps independently:

# OhioT1DM Pipeline
ohiot1dm_preprocess: false
ohiot1dm_model: false
ohiot1dm_dataset: "all"  # Options: 2018, 2020, all
ohiot1dm_attack_type: "URET"  # Options: URET, FGSM
ohiot1dm_risk_profile: false
ohiot1dm_cluster: false
ohiot1dm_cluster_method: "hierarchical"  # Options: hierarchical, kmeans
ohiot1dm_generate_defense_datasets: false
ohiot1dm_evaluate_defense: false
ohiot1dm_defense_type: "all"  # Options: knn, oneclasssvm, madgan, all
ohiot1dm_plot_defense_results: false

# MIMIC Pipeline
mimic_preprocess: false
mimic_model_train: false
mimic_model_test: false
mimic_attack_type: "URET"  # Options: URET, FGSM
mimic_risk_profile: false
mimic_cluster: false
mimic_cluster_method: "hierarchical"  # Options: hierarchical, kmeans
mimic_generate_defense_datasets: false
mimic_evaluate_defense: false
mimic_defense_type: "all"  # Options: knn, oneclasssvm, madgan, all
mimic_plot_defense_results: false

# PhysioNetCinC Pipeline
physionetcinc_model: false
physionetcinc_dataset: "all"  # Options: A, B, all
physionetcinc_attack_type: "URET"  # Options: URET, FGSM
physionetcinc_risk_profile: false
physionetcinc_cluster: false
physionetcinc_cluster_method: "hierarchical"  # Options: hierarchical, kmeans
physionetcinc_generate_defense_datasets: false
physionetcinc_evaluate_defense: false
physionetcinc_defense_type: "all"  # Options: knn, oneclasssvm, madgan, all
physionetcinc_plot_defense_results: false

Running the Pipeline

Once your configuration is configured, orchestrate execution using run_pipeline.sh. The script applies overrides directly from the command-line arguments:

# Run using the configuration provided in pipeline_config.yml
./run_pipeline.sh

# Or directly override specific flags on the command line
./run_pipeline.sh --ohiot1dm_dataset=2018 --ohiot1dm_model=true --mimic_model=false