A Trustworthy Machine Learning Algorithm Library. This repo collects the official implementations of works done by Yisen Wang and his students.

• Our repo is released!
• [] Add README.
• [] Add package dependency and scripts to start the code.
• [] Report benchmark results for reference.

The environment in which we build this repo is given in requirements.txt. Run the below command to configure the environment.

  pip install -r requirements.txt

• Valina Adversarial Training
• MART
• AWP
• Generalist
• DynACL
• ReBAT

To start the code

  python main_baseat.py --yaml config/baseat.yaml

Parameters are given in config/baseat.yaml.

To start the code

  python main_mart.py --yaml config/mart.yaml

Parameters are given in config/mart.yaml.

To start the code

  python main_awp.py --yaml config/awp.yaml

Parameters are given in config/awp.yaml.

To start the code

  python main_generalist.py --yaml config/generalist.yaml

Parameters are given in config/generalist.yaml.

To start the code

  python main_dynacl.py --yaml config/dynacl.yaml // Pretraining
  python main_dynacllinear.py --yaml config/dynacllinear.yaml // Linear-Probing

Parameters are given in config/dynacl.yaml.

To start the code

  python main_rebat.py --yaml config/rebat.yaml 

Parameters are given in config/rebat.yaml.

To evaluate the accuracy/robustness of pre-trained model

  python main_eval.py --yaml config/eval.yaml 

Parameters are given in config/eval.yaml.

We refer some of the links below while building this repo. We sincerely acknowlegde their great work!

• https://github.com/Harry24k/adversarial-attacks-pytorch
• https://github.com/fra31/auto-attack
• https://github.com/MadryLab/robustness
• https://github.com/pytorch/vision/blob/main/torchvision/models