From 2a25d1d1e68c286797209293ef7b281d3b1ac402 Mon Sep 17 00:00:00 2001
From: shihuahuang <shihuahuang95@gmail.com>
Date: Fri, 7 Jun 2024 16:42:24 +0800
Subject: [PATCH] add the deleted eval files

---
 eval_robustness.py | 220 +++++++++++++++++++++++++++++++++++++++++++++
 evaluator.py       | 159 ++++++++++++++++++++++++++++++++
 2 files changed, 379 insertions(+)
 create mode 100644 eval_robustness.py
 create mode 100644 evaluator.py

diff --git a/eval_robustness.py b/eval_robustness.py
new file mode 100644
index 0000000..4094c19
--- /dev/null
+++ b/eval_robustness.py
@@ -0,0 +1,220 @@
+import os
+import time
+import argparse
+import mlconfig  # pip install mlconfig (https://github.com/narumiruna/mlconfig)
+import numpy as np
+from torchprofile import profile_macs  # pip install torchprofile (https://github.com/zhijian-liu/torchprofile)
+
+import torch
+from torch.autograd import Variable
+
+from core import util
+from evaluator import Evaluator
+from auto_attack.autoattack import AutoAttack
+
+
+parser = argparse.ArgumentParser(description='Adversarial Robustness Evaluation')
+parser.add_argument('--seed', type=int, default=0)
+parser.add_argument('--data', type=str, default='./dataset', help='Path to dataset file')
+parser.add_argument('--config_file_path', type=str, default='', help='Path to the configuration .yaml file')
+parser.add_argument('--checkpoint_path', type=str, default=None, help='Path to the pretrained model checkpoint .pth file')
+parser.add_argument('--save_path', type=str, default='./', help='Path to file for saving evaluation printouts')
+parser.add_argument('--attack_choice', default='PGD', choices=['PGD', 'AA', 'CW', 'FGSM'])
+parser.add_argument('--epsilon', default=8, type=float, help='perturbation')
+parser.add_argument('--num_steps', default=20, type=int, help='perturb number of steps')
+parser.add_argument('--step_size', default=0.8, type=float, help='perturb step size')
+parser.add_argument('--batch_size', default=100, type=int, help='evaluation batch size')
+
+args = parser.parse_args()
+if args.epsilon > 1:
+    args.epsilon = args.epsilon / 255
+    args.step_size = args.step_size / 255
+
+config = mlconfig.load(args.config_file_path)
+name_to_save = os.path.basename(args.config_file_path).split('.')[0]
+
+logger = util.setup_logger(name=name_to_save, log_file=os.path.join(
+    args.save_path, name_to_save) + '_eval@{}-{}steps'.format(args.attack_choice, args.num_steps) + ".log")
+
+torch.manual_seed(args.seed)
+np.random.seed(args.seed)
+
+if torch.cuda.is_available():
+    torch.backends.cudnn.enabled = True
+    torch.backends.cudnn.benchmark = True
+    device = torch.device('cuda')
+    device_list = [torch.cuda.get_device_name(i) for i in range(0, torch.cuda.device_count())]
+    logger.info("GPU List: %s" % (device_list))
+else:
+    device = torch.device('cpu')
+
+
+def whitebox_eval(data_loader, model, evaluator, log=True):
+    natural_count, pgd_count, total, stable_count = 0, 0, 0, 0
+    loss_meters = util.AverageMeter()
+    lip_meters = util.AverageMeter()
+
+    print("    ###    Using {} ####    ".format(args.attack_choice))
+    model.eval()
+    for i, (images, labels) in enumerate(data_loader["test_dataset"]):
+        images, labels = images.to(device), labels.to(device)
+        images, labels = Variable(images, requires_grad=True), Variable(labels)
+        if args.attack_choice == 'PGD':
+            # pgd attack
+            rs = evaluator._pgd_whitebox(
+                model, images, labels, random_start=True, epsilon=args.epsilon,
+                num_steps=args.num_steps, step_size=args.step_size)
+
+        elif args.attack_choice == 'CW':
+            # cw attack
+            rs = evaluator._pgd_cw_whitebox(
+                model, images, labels, random_start=True, epsilon=args.epsilon,
+                num_steps=args.num_steps, step_size=args.step_size)
+
+        elif args.attack_choice == 'FGSM':
+            # fgsm attack
+            rs = evaluator._fgsm_whitebox(
+                model, images, labels, random_start=True, epsilon=args.epsilon,
+                num_steps=args.num_steps, step_size=args.step_size)
+
+        else:
+            raise NotImplementedError('Not implemented')
+
+        acc, acc_pgd, loss, stable, X_pgd = rs
+        total += images.size(0)
+        natural_count += acc
+        pgd_count += acc_pgd
+        stable_count += stable
+        local_lip = util.local_lip(model, images, X_pgd).item()
+        lip_meters.update(local_lip)
+        loss_meters.update(loss)
+        if log:
+            payload = 'LIP: %.4f\tStable Count: %d\tNatural Count: %d/%d\tNatural Acc: ' \
+                      '%.2f\tAdv Count: %d/%d\tAdv Acc: %.2f' % (
+                          local_lip, stable_count, natural_count, total, (natural_count / total) * 100,
+                          pgd_count, total, (pgd_count / total) * 100)
+            logger.info(payload)
+
+    natural_acc = (natural_count / total) * 100
+    pgd_acc = (pgd_count / total) * 100
+    payload = 'Natural Correct Count: %d/%d Acc: %.2f ' % (natural_count, total, natural_acc)
+    logger.info(payload)
+    payload = '%s Correct Count: %d/%d Acc: %.2f ' % (args.attack_choice, pgd_count, total, pgd_acc)
+    logger.info(payload)
+    payload = '%s with %.1f steps Loss Avg: %.2f ' % (args.attack_choice, args.num_steps, loss_meters.avg)
+    logger.info(payload)
+    payload = 'LIP Avg: %.4f ' % lip_meters.avg
+    logger.info(payload)
+    payload = 'Stable Count: %d/%d StableAcc: %.2f ' % (stable_count, total, stable_count * 100 / total)
+    logger.info(payload)
+    return natural_acc, pgd_acc, stable_count * 100 / total, lip_meters.avg
+
+
+def main():
+    # Load Search Version Genotype
+    model = config.model().to(device)
+    logger.info(model)
+
+    # Setup ENV
+    data_loader = config.dataset(data_path=args.data, eval_batch_size=args.batch_size).getDataLoader()
+    # data_loader = None
+    evaluator = Evaluator(data_loader, logger, config)
+
+    if hasattr(config.dataset, "input_size"):
+        print("   ## FLOPs with input shape {} ##  ".format([1, 3, config.dataset.input_size, config.dataset.input_size]))
+        profile_inputs = (torch.randn([1, 3, config.dataset.input_size, config.dataset.input_size]).to(device),)
+
+    elif config.dataset.dataset_type == "TINYIMAGENET":
+        print("   ## FLOPs with input shape {} ##  ".format([1, 3, 64, 64]))
+        profile_inputs = (torch.randn([1, 3, 64, 64]).to(device),)
+
+    else:
+        profile_inputs = (torch.randn([1, 3, 32, 32]).to(device),)
+
+    flops = profile_macs(model, profile_inputs) / 1e6
+
+    config.set_immutable()
+    for key in config:
+        logger.info("%s: %s" % (key, config[key]))
+
+    logger.info("param size = %fMB", util.count_parameters_in_MB(model))
+    logger.info("flops: %.4fM" % flops)
+    logger.info("PyTorch Version: %s" % torch.__version__)
+
+    if torch.cuda.is_available():
+        device_list = [torch.cuda.get_device_name(i) for i in range(0, torch.cuda.device_count())]
+        logger.info("GPU List: %s" % device_list)
+
+    ENV = {'global_step': 0,
+           'best_acc': 0.0,
+           'curren_acc': 0.0,
+           'best_pgd_acc': 0.0,
+           'flops': flops,
+           'train_history': [],
+           'eval_history': [],
+           'pgd_eval_history': [],
+           'stable_acc_history': [],
+           'lip_history': [],
+           'genotype_list': []}
+
+    if args.checkpoint_path:
+        checkpoint = util.load_model(filename=args.checkpoint_path,
+                                     model=model,
+                                     optimizer=None,
+                                     alpha_optimizer=None,
+                                     scheduler=None)
+        if 'ENV' in checkpoint:
+            ENV = checkpoint['ENV']
+        if 'stable_acc_history' not in ENV:
+            ENV['stable_acc_history'] = []
+        if 'lip_history' not in ENV:
+            ENV['lip_history'] = []
+        logger.info("File %s loaded!" % args.checkpoint_path)
+
+    model = torch.nn.DataParallel(model).to(device)
+
+    if args.attack_choice in ['PGD', 'CW' "FGSM"]:
+        for param in model.parameters():
+            param.requires_grad = False
+        model.eval()
+        natural_acc, adv_acc, stable_acc, lip = whitebox_eval(data_loader, model, evaluator)
+        key = '%s_%d' % (args.attack_choice, args.num_steps)
+        ENV['natural_acc'] = natural_acc
+        ENV[key] = adv_acc
+        ENV['%s_stable' % key] = stable_acc
+        ENV['%s_lip' % key] = lip
+
+    elif args.attack_choice == 'AA':
+        for param in model.parameters():
+            param.requires_grad = False
+        x_test = [x for (x, y) in data_loader['test_dataset']]
+        x_test = torch.cat(x_test, dim=0)
+        y_test = [y for (x, y) in data_loader['test_dataset']]
+        y_test = torch.cat(y_test, dim=0)
+        model.eval()
+
+        adversary = AutoAttack(model, norm='Linf', eps=args.epsilon, logger=logger, verbose=True)
+        adversary.plus = False
+
+        logger.info('=' * 20 + 'AA Attack Eval' + '=' * 20)
+        adversary.attacks_to_run = ['apgd-ce', 'apgd-t']
+        print("  ### Evaluate AA with {} attackers ###  ".format(adversary.attacks_to_run))
+        x_adv, robust_accuracy = adversary.run_standard_evaluation(x_test, y_test, bs=config.dataset.eval_batch_size)
+        robust_accuracy = robust_accuracy * 100
+        logger.info('AA Accuracy: %.2f' % (robust_accuracy))
+
+        ENV['aa_attack'] = robust_accuracy
+
+    return
+
+
+if __name__ == '__main__':
+    for arg in vars(args):
+        logger.info("%s: %s" % (arg, getattr(args, arg)))
+
+    start = time.time()
+    main()
+    end = time.time()
+    cost = (end - start) / 86400
+    payload = "Running Cost %.2f Days" % cost
+    logger.info(payload)
\ No newline at end of file
diff --git a/evaluator.py b/evaluator.py
new file mode 100644
index 0000000..a8fd9ed
--- /dev/null
+++ b/evaluator.py
@@ -0,0 +1,159 @@
+from core import util
+import time
+
+import torch
+import torch.optim as optim
+from torch.autograd import Variable
+
+if torch.cuda.is_available():
+    device = torch.device('cuda')
+else:
+    device = torch.device('cpu')
+
+
+class Evaluator:
+    def __init__(self, data_loader, logger, config):
+        self.loss_meters = util.AverageMeter()
+        self.acc_meters = util.AverageMeter()
+        self.acc5_meters = util.AverageMeter()
+        self.criterion = torch.nn.CrossEntropyLoss()
+        self.data_loader = data_loader
+        self.logger = logger
+        self.log_frequency = config.log_frequency if config.log_frequency is not None else 100
+        self.config = config
+        self.current_acc = 0
+        self.current_acc_top5 = 0
+        return
+
+    def _reset_stats(self):
+        self.loss_meters = util.AverageMeter()
+        self.acc_meters = util.AverageMeter()
+        self.acc5_meters = util.AverageMeter()
+        return
+
+    def eval(self, epoch, model):
+        model.eval()
+        for i, (images, labels) in enumerate(self.data_loader["test_dataset"]):
+            start = time.time()
+            log_payload = self.eval_batch(images=images, labels=labels, model=model)
+            end = time.time()
+            time_used = end - start
+        display = util.log_display(epoch=epoch,
+                                   global_step=i,
+                                   time_elapse=time_used,
+                                   **log_payload)
+        if self.logger is not None:
+            self.logger.info(display)
+        return
+
+    def eval_batch(self, images, labels, model):
+        images, labels = images.to(device, non_blocking=True), labels.to(device, non_blocking=True)
+        with torch.no_grad():
+            pred = model(images)
+            loss = self.criterion(pred, labels)
+            acc, acc5 = util.accuracy(pred, labels, topk=(1, 5))
+        self.loss_meters.update(loss.item(), n=images.size(0))
+        self.acc_meters.update(acc.item(), n=images.size(0))
+        self.acc5_meters.update(acc5.item(), n=images.size(0))
+        payload = {"val_acc": acc.item(),
+                   "val_acc_avg": self.acc_meters.avg,
+                   "val_acc5": acc5.item(),
+                   "val_acc5_avg": self.acc5_meters.avg,
+                   "val_loss": loss.item(),
+                   "val_loss_avg": self.loss_meters.avg}
+        return payload
+
+    def _fgsm_whitebox(self, model, X, y, random_start=False, epsilon=0.031, num_steps=20, step_size=0.003):
+        model.eval()
+        out = model(X)
+        acc = (out.data.max(1)[1] == y.data).float().sum()
+        X_pgd = Variable(X.data, requires_grad=True)
+        opt = optim.SGD([X_pgd], lr=1e-3)
+        opt.zero_grad()
+        loss = torch.nn.CrossEntropyLoss()(model(X_pgd), y)
+        loss.backward()
+        eta = epsilon * X_pgd.grad.data.sign()
+        X_pgd = Variable(X_pgd.data + eta, requires_grad=True)
+        eta = torch.clamp(X_pgd.data - X.data, -epsilon, epsilon)
+        X_pgd = Variable(X.data + eta, requires_grad=True)
+        X_pgd = Variable(torch.clamp(X_pgd, 0, 1.0), requires_grad=True)
+
+        X_pgd = Variable(X_pgd.data, requires_grad=False)
+        predict_pgd = model(X_pgd).data.max(1)[1].detach()
+        predict_clean = model(X).data.max(1)[1].detach()
+        acc_pgd = (predict_pgd == y.data).float().sum()
+        stable = (predict_pgd.data == predict_clean.data).float().sum()
+        return acc.item(), acc_pgd.item(), loss.item(), stable.item(), X_pgd
+
+    def _pgd_whitebox(self, model, X, y, random_start=True,
+                      epsilon=0.031, num_steps=20, step_size=0.003):
+        model.eval()
+        out = model(X)
+        acc = (out.data.max(1)[1] == y.data).float().sum()
+        X_pgd = Variable(X.data, requires_grad=True)
+        if random_start:
+            random_noise = torch.FloatTensor(*X_pgd.shape).uniform_(-epsilon, epsilon).to(device)
+            X_pgd = Variable(X_pgd.data + random_noise, requires_grad=True)
+
+        for _ in range(num_steps):
+            opt = optim.SGD([X_pgd], lr=1e-3)
+            opt.zero_grad()
+            loss = torch.nn.CrossEntropyLoss()(model(X_pgd), y)
+            loss.backward()
+            eta = step_size * X_pgd.grad.data.sign()
+            X_pgd = Variable(X_pgd.data + eta, requires_grad=True)
+            eta = torch.clamp(X_pgd.data - X.data, -epsilon, epsilon)
+            X_pgd = Variable(X.data + eta, requires_grad=True)
+            X_pgd = Variable(torch.clamp(X_pgd, 0, 1.0), requires_grad=True)
+
+        X_pgd = Variable(X_pgd.data, requires_grad=False)
+        predict_pgd = model(X_pgd).data.max(1)[1].detach()
+        predict_clean = model(X).data.max(1)[1].detach()
+        acc_pgd = (predict_pgd == y.data).float().sum()
+        stable = (predict_pgd.data == predict_clean.data).float().sum()
+        return acc.item(), acc_pgd.item(), loss.item(), stable.item(), X_pgd
+
+    def _pgd_cw_whitebox(self, model, X, y, random_start=True,
+                         epsilon=0.031, num_steps=20, step_size=0.003):
+        model.eval()
+        out = model(X)
+        acc = (out.data.max(1)[1] == y.data).float().sum()
+        X_pgd = Variable(X.data, requires_grad=True)
+
+        def CWLoss(output, target, confidence=0):
+            """
+            CW loss (Marging loss).
+            """
+            num_classes = output.shape[-1]
+            target = target.data
+            target_onehot = torch.zeros(target.size() + (num_classes,))
+            target_onehot = target_onehot.cuda()
+            target_onehot.scatter_(1, target.unsqueeze(1), 1.)
+            target_var = Variable(target_onehot, requires_grad=False)
+            real = (target_var * output).sum(1)
+            other = ((1. - target_var) * output - target_var * 10000.).max(1)[0]
+            loss = - torch.clamp(real - other + confidence, min=0.)
+            loss = torch.sum(loss)
+            return loss
+
+        if random_start:
+            random_noise = torch.FloatTensor(*X_pgd.shape).uniform_(-epsilon, epsilon).to(device)
+            X_pgd = Variable(X_pgd.data + random_noise, requires_grad=True)
+
+        for _ in range(num_steps):
+            opt = optim.SGD([X_pgd], lr=1e-3)
+            opt.zero_grad()
+            loss = CWLoss(model(X_pgd), y)
+            loss.backward()
+            eta = step_size * X_pgd.grad.data.sign()
+            X_pgd = Variable(X_pgd.data + eta, requires_grad=True)
+            eta = torch.clamp(X_pgd.data - X.data, -epsilon, epsilon)
+            X_pgd = Variable(X.data + eta, requires_grad=True)
+            X_pgd = Variable(torch.clamp(X_pgd, 0, 1.0), requires_grad=True)
+
+        X_pgd = Variable(X_pgd.data, requires_grad=False)
+        predict_pgd = model(X_pgd).data.max(1)[1].detach()
+        predict_clean = model(X).data.max(1)[1].detach()
+        acc_pgd = (predict_pgd == y.data).float().sum()
+        stable = (predict_pgd.data == predict_clean.data).float().sum()
+        return acc.item(), acc_pgd.item(), loss.item(), stable.item(), X_pgd
\ No newline at end of file