From 35f761263b0f68d3aa3b7a7cc7b1cc62c5d85d8d Mon Sep 17 00:00:00 2001 From: Neea Date: Sat, 6 Jan 2024 00:17:30 -0500 Subject: [PATCH] refactor --- LICENSE | 2 +- README.md | 37 ++++++++++++++++++++++--------------- cge/types.py | 2 +- exp/attack.py | 6 +++--- exp/hopskip.py | 2 +- exp/pgd.py | 2 +- exp/zoo.py | 2 +- 7 files changed, 30 insertions(+), 23 deletions(-) diff --git a/LICENSE b/LICENSE index fe5b507..6d0dce9 100644 --- a/LICENSE +++ b/LICENSE @@ -1,6 +1,6 @@ MIT License -Copyright (c) 2023 Augusta University, Cyber Attack Detection +Copyright (c) 2023 AU Cyber Attack Detection Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal diff --git a/README.md b/README.md index b3d8033..599d071 100644 --- a/README.md +++ b/README.md @@ -5,7 +5,12 @@ This implementation demonstrates an approach to introduce constraints to unconstrained adversarial machine learning evasion attacks. We develop a constraint validation algorithm, _Contraint Guaranteed Evasion_ (CGE), that guarantees generated evasive adversarial examples satisfy domain constraints. -The experimental setup allows running various adversarial evasion attacks, enhanced with CGE, on different data sets and victim classifiers. +Complete implementation of the CGE algorithm is in [`/cge`](https://github.com/aucad/cge/tree/main/cge) directory. +Examples of how to define constraints can be found, e.g., [here](https://github.com/aucad/cge/blob/main/config/iot23.yaml). +The constraints are converted to executable form using this [preprocessor](https://github.com/aucad/cge/blob/main/exp/preproc.py#L14-L27). +Examples showing how to integrate CGE into existing adversarial evasion attacks are [here](https://github.com/aucad/cge/blob/main/exp/hopskip.py#L26-L28) and [here](https://github.com/aucad/cge/blob/main/exp/pgd.py#L44) and [here](https://github.com/aucad/cge/blob/main/exp/zoo.py#L44). + +This repository also includes an experimental setup for running various adversarial evasion attacks, enhanced with CGE, on different data sets and victim classifiers. The following options are included. - **Attacks**: Projected Gradient Descent (PGD), Zeroth-Order Optimization (ZOO), HopSkipJump attack. @@ -15,6 +20,7 @@ The following options are included. **Comparison.** We also include a comparison attack, _Constrained Projected Gradient Descent_ (C-PGD). It uses a different constraint evaluation approach introduced by [Simonetto et al](https://arxiv.org/abs/2112.01156). +The C-PGD implementation is from [here](https://github.com/serval-uni-lu/constrained-attacks) and has its own, separate software license. **Data sets** @@ -36,17 +42,17 @@ It uses a different constraint evaluation approach introduced by [Simonetto et a ### Repository organization -| Directory | Description | -|:-------------|:--------------------------------------------------| -| `.github` | Automated workflows, development instructions | -| `cge` | CGE algorithm implementation | -| `comparison` | C-PGD attack implementation and its license | -| `config` | Experiment configuration files | -| `data` | Preprocessed input data sets | -| `exp` | Source code for running experiments | -| `plot` | Utilities for plotting experiment results | -| `ref_result` | Referential result for comparison | -| `test` | Unit tests | +| Directory | Description | +|:-------------|:----------------------------------------------| +| `.github` | Automated workflows, development instructions | +| `cge` | CGE algorithm implementation | +| `comparison` | C-PGD attack implementation and its license | +| `config` | Experiment configuration files | +| `data` | Preprocessed input data sets | +| `exp` | Source code for running experiments | +| `plot` | Plotting of experiment results | +| `ref_result` | Referential result for comparison | +| `test` | Unit tests | - The Makefile contains pre-configured commands to ease running experiments. - The `data/feature_*.csv` files are exclusively for use with C-PGD attack. @@ -54,7 +60,8 @@ It uses a different constraint evaluation approach introduced by [Simonetto et a ## Experiment workflow -A single experiment consists of (1) preprocessing and setup (2) training a classification model on a choice data set (3) applying an adversarial attack to that model (4) scoring and (5) recording the result. A constraint-validation approach can be enabled or disabled during the attack to impact the validity of the generated adversarial examples. +A single experiment consists of (1) preprocessing and setup (2) training a classification model on a choice data set (3) applying an adversarial attack to that model (4) scoring and (5) recording the result. +A constraint-validation approach can be enabled or disabled during the attack to impact the validity of the generated adversarial examples. 
     ┌───────────────┐     ┌───────────────┐     ┌───────────────┐     ┌───────────────┐ 
@@ -66,7 +73,7 @@ A single experiment consists of (1) preprocessing and setup (2) training a class
      * other configs      * init validation      3. score
-## Usage +## Reproducing paper experiments **Software requirements** @@ -117,7 +124,7 @@ make plots make plots DIR=ref_result ``` -## Extended/Custom usage +## Extended usage The default experiment options are defined statically in `config` files. An experiment run can be customized further with command line arguments, to override the static options. diff --git a/cge/types.py b/cge/types.py index 554d9f2..e103e5c 100644 --- a/cge/types.py +++ b/cge/types.py @@ -16,7 +16,7 @@ class Validatable: """Base class for an attack with constraints""" - v_model = None + cge = None def vhost(self): """validation model owner""" diff --git a/exp/attack.py b/exp/attack.py index 9ce2f43..ccbae4b 100644 --- a/exp/attack.py +++ b/exp/attack.py @@ -64,7 +64,7 @@ def reset(self, cls): def can_validate(self): return issubclass(self.attack, Validatable) - def run(self, v_model: Validation): + def run(self, cge: Validation): """Generate adversarial examples and score.""" self.start = time.time_ns() if issubclass(self.attack, CPGD): @@ -73,7 +73,7 @@ def run(self, v_model: Validation): else: aml_attack = self.attack(self.cls.classifier, **self.conf) if self.can_validate: - aml_attack.vhost().v_model = v_model + aml_attack.vhost().cge = cge self.adv_x = aml_attack.generate(x=self.ori_x) self.adv_y = np.array(self.cls.predict( self.adv_x, self.ori_y).flatten()) @@ -82,7 +82,7 @@ def run(self, v_model: Validation): sys.stdout.write('\x1b[2K') self.score.calculate( - self, v_model.constraints, v_model.scalars, + self, cge.constraints, cge.scalars, dur=self.end - self.start) return self diff --git a/exp/hopskip.py b/exp/hopskip.py index 13526e9..7bd12e8 100644 --- a/exp/hopskip.py +++ b/exp/hopskip.py @@ -23,6 +23,6 @@ def _attack( target, mask, clip_min, clip_max) # adjust shape: 1d -> 2d -> 1d - return self.v_model.enforce( + return self.cge.enforce( np.array([original_sample]), np.array([x_adv]))[0] # NEW diff --git a/exp/pgd.py b/exp/pgd.py index 75e5943..e89b17d 100644 --- a/exp/pgd.py +++ b/exp/pgd.py @@ -41,7 +41,7 @@ def _compute( x, x_init, y, mask, eps, eps_step, project, random_init, batch_id_ext, decay, momentum) - return self.v_model.enforce(x, x_adv) # NEW + return self.cge.enforce(x, x_adv) # NEW class VPGD(ProjectedGradientDescent, Validatable): diff --git a/exp/zoo.py b/exp/zoo.py index f9c902f..1cbf0f9 100644 --- a/exp/zoo.py +++ b/exp/zoo.py @@ -41,7 +41,7 @@ def _generate_batch(self, x_batch: np.ndarray, y_batch: np.ndarray) -> np.ndarra # Run with 1 specific binary search step best_dist, best_label, best_attack = self._generate_bss(x_batch, y_batch, c_current) - best_attack = self.v_model.enforce(x_batch, best_attack) # NEW! + best_attack = self.cge.enforce(x_batch, best_attack) # NEW! # Update best results so far o_best_attack[best_dist < o_best_dist] = best_attack[best_dist < o_best_dist]