Official Code for AdvRush: Searching for Adversarially Robust Neural Architectures (ICCV '21)

Last update: Dec 10, 2022

Related tags

Overview

AdvRush

Official Code for AdvRush: Searching for Adversarially Robust Neural Architectures (ICCV '21)

Environmental Set-up

Python == 3.6.12, PyTorch == 1.2.0, torchvision == 0.4.0

AdvRush Search Process

cd advrush && python train_search.py --batch_size 32 --gpu 0 --epochs 60 --a_gamma 0.01 --a_warmup_epochs 0 --w_warmup_epochs 60 --loss_hessian loss_cure

Adversarial Training

cd advrush && python adv_train.py --batch_size 64 --gpu 0 --epochs 200 --adv_loss pgd --arch ADVRUSH

Evaluation under PGD Attack

Prior to the evaluation process, add all necessary checkpoint files (preferably in the form of .pth.tar) to the /eval/checkpoints folder. To conduct white-box attacks,

cd eval &&
python pgd_attack.py --white-box-attack True --test-batch-size 10 --arch [arch_name] --checkpoint [./checkpoints/file_name.pth.tar] --data_type [cifar10/svhn]

To conduct black-box attacks,

cd eval &&
python pgd_attack.py --test-batch-size 10 --target_arch [target_arch] --target_checkpoint [./checkpoints/target_file.pth.tar] --source_arch [source_arch] --source_checkpoint [./checkpoints/source_file.pth.tar] --data_type cifar10

References

DARTS: Differentiable Architecture Search [ICLR '19] code paper

Robustness via Curvature Regularization, and Vice Versa [CVPR '19] code paper

Tradeoff-inspired Adversarial Defense via Surrogate-loss Minimization [ICML '19] code paper

Official Code for AdvRush: Searching for Adversarially Robust Neural Architectures (ICCV '21)

Related tags

Overview

AdvRush

Environmental Set-up

AdvRush Search Process

Adversarial Training

Evaluation under PGD Attack

References

Owner

repro_eval is a collection of measures to evaluate the reproducibility/replicability of system-oriented IR experiments

Tensorflow implementation of "Learning Deep Features for Discriminative Localization"

Pytorch Implementation for NeurIPS (oral) paper: Pixel Level Cycle Association: A New Perspective for Domain Adaptive Semantic Segmentation

Boundary-aware Transformers for Skin Lesion Segmentation

Code for a real-time distributed cooperative slam(RDC-SLAM) system for ROS compatible platforms.

Implementation of "Unsupervised Domain Adaptive 3D Detection with Multi-Level Consistency"

RIM: Reliable Influence-based Active Learning on Graphs.

A Python framework for conversational search

This program writes christmas wish programmatically. It is using turtle as a pen pointer draw christmas trees and stars.

Python calculations for the position of the sun and moon.

Which Style Makes Me Attractive? Interpretable Control Discovery and Counterfactual Explanation on StyleGAN

Semantic Scholar's Author Disambiguation Algorithm & Evaluation Suite

A novel method to tune language models. Codes and datasets for paper ``GPT understands, too''.

Visualization toolkit for neural networks in PyTorch! Demo -->

Official MegEngine implementation of CREStereo(CVPR 2022 Oral).

Supervised Sliding Window Smoothing Loss Function Based on MS-TCN for Video Segmentation

Code for ECCV 2020 paper "Contacts and Human Dynamics from Monocular Video".

Learning Lightweight Low-Light Enhancement Network using Pseudo Well-Exposed Images

Styled Augmented Translation

Python Auto-ML Package for Tabular Datasets