(#7) Attack: Add CW attack

betarixm · Apr 18, 2022 · 5852338 · 5852338
1 parent 42127f3
commit 5852338
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/src/attack/attack.py b/src/attack/attack.py
@@ -1,5 +1,5 @@
 from typings.models import Attack
-from models import Fgsm, Pgd
+from models import Fgsm, Pgd, Cw
 from defense.models import Reformer
 from victim.models import Classifier
 
@@ -33,7 +33,7 @@
         type=str,
         help="Attack method",
         required=True,
-        choices=["fgsm", "pgd"],
+        choices=["fgsm", "pgd", "cw"],
     )
 
     parser.add_argument(
@@ -74,8 +74,10 @@
 
     if args.method == "fgsm":
         attack_cls = Fgsm
-    else:
+    elif args.method == "pgd":
         attack_cls = Pgd
+    else:
+        attack_cls = Cw
 
     attacker = attack_cls(classifier, dataset, defense_model)
 

diff --git a/src/attack/models.py b/src/attack/models.py
@@ -2,6 +2,7 @@
 
 from cleverhans.tf2.attacks.fast_gradient_method import fast_gradient_method
 from cleverhans.tf2.attacks.projected_gradient_descent import projected_gradient_descent
+from cleverhans.tf2.attacks.carlini_wagner_l2 import carlini_wagner_l2
 
 import numpy as np
 import tensorflow as tf
@@ -24,3 +25,8 @@ def add_perturbation(self, x: np.array) -> np.array:
             40,
             np.inf,
         )
+
+
+class Cw(Attack):
+    def add_perturbation(self, x: np.array) -> np.array:
+        return carlini_wagner_l2(self.victim_model.model(), x)