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

Implementation of DropLoss for Long-Tail Instance Segmentation in Pytorch

Towards Open-World Feature Extrapolation: An Inductive Graph Learning Approach

Paddle-Adversarial-Toolbox (PAT) is a Python library for Deep Learning Security based on PaddlePaddle.

The official repository for paper ''Domain Generalization for Vision-based Driving Trajectory Generation'' submitted to ICRA 2022

Scalable Optical Flow-based Image Montaging and Alignment

Train Scene Graph Generation for Visual Genome and GQA in PyTorch >= 1.2 with improved zero and few-shot generalization.

Point-NeRF: Point-based Neural Radiance Fields

U-Net Brain Tumor Segmentation

Nest - A flexible tool for building and sharing deep learning modules

Reverse engineering recurrent neural networks with Jacobian switching linear dynamical systems

FAMIE is a comprehensive and efficient active learning (AL) toolkit for multilingual information extraction (IE)

This repository contains an implementation of ConvMixer for the ICLR 2022 submission "Patches Are All You Need?".

Train a state-of-the-art yolov3 object detector from scratch!

The official implementation of NeurIPS 2021 paper: Finding Optimal Tangent Points for Reducing Distortions of Hard-label Attacks

A simple interface for editing natural photos with generative neural networks.

Improving Generalization Bounds for VC Classes Using the Hypergeometric Tail Inversion

code for paper "Does Unsupervised Architecture Representation Learning Help Neural Architecture Search?"

Enhancing Column Generation by a Machine-Learning-BasedPricing Heuristic for Graph Coloring

Simple Tensorflow implementation of Toward Spatially Unbiased Generative Models (ICCV 2021)

MLP-Numpy - A simple modular implementation of Multi Layer Perceptron in pure Numpy.