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

BoxInst: High-Performance Instance Segmentation with Box Annotations

Monocular 3D pose estimation. OpenVINO. CPU inference or iGPU (OpenCL) inference.

An implementation of the AlphaZero algorithm for Gomoku (also called Gobang or Five in a Row)

Unadversarial Examples: Designing Objects for Robust Vision

Deep Learning for 3D Point Clouds: A Survey (IEEE TPAMI, 2020)

An example of Scatterbrain implementation (combining local attention and Performer)

FirmWire is a full-system baseband firmware emulation platform for fuzzing, debugging, and root-cause analysis of smartphone baseband firmwares

EDPN: Enhanced Deep Pyramid Network for Blurry Image Restoration

pytorch implementation of "Contrastive Multiview Coding", "Momentum Contrast for Unsupervised Visual Representation Learning", and "Unsupervised Feature Learning via Non-Parametric Instance-level Discrimination"

PyTorch implementation of Tacotron speech synthesis model.

Unified MultiWOZ evaluation scripts for the context-to-response task.

PyTorch implementation of an end-to-end Handwritten Text Recognition (HTR) system based on attention encoder-decoder networks

sktime companion package for deep learning based on TensorFlow

OpenDILab RL Kubernetes Custom Resource and Operator Lib

Mixed Neural Likelihood Estimation for models of decision-making

Chatbot in 200 lines of code using TensorLayer

Le dataset des images du projet d'IA de 2021

A library for building and serving multi-node distributed faiss indices.

Mortgage-loan-prediction - Show how to perform advanced Analytics and Machine Learning in Python using a full complement of PyData utilities

Supplemental Code for "ImpressionNet :A Multi view Approach to Predict Socio Facial Impressions"