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

基于AlphaPose的TensorRT加速

Redash reset for python

YoloAll is a collection of yolo all versions. you you use YoloAll to test yolov3/yolov5/yolox/yolo_fastest

This is an official implementation for "AS-MLP: An Axial Shifted MLP Architecture for Vision".

Using pretrained GROVER to extract the atomic fingerprints from molecule

Keras + Hyperopt: A very simple wrapper for convenient hyperparameter optimization

code and data for paper "GIANT: Scalable Creation of a Web-scale Ontology"

PyTorch EO aims to make Deep Learning for Earth Observation data easy and accessible to real-world cases and research alike.

competitions-v2

PyTorch implementation of "A Two-Stage End-to-End System for Speech-in-Noise Hearing Aid Processing"

Official implementation of Rethinking Graph Neural Architecture Search from Message-passing (CVPR2021)

Brax is a differentiable physics engine that simulates environments made up of rigid bodies, joints, and actuators

Facial Action Unit Intensity Estimation via Semantic Correspondence Learning with Dynamic Graph Convolution

DeLighT: Very Deep and Light-Weight Transformers

Gated-Shape CNN for Semantic Segmentation (ICCV 2019)

This repository contains the source code of our work on designing efficient CNNs for computer vision

CRF-RNN for Semantic Image Segmentation - PyTorch version

Happywhale - Whale and Dolphin Identification Silver🥈 Solution (26/1588)

RL-driven agent playing tic-tac-toe on starknet against challengers.

机器学习、深度学习、自然语言处理等人工智能基础知识总结。