Awesome
[CVPR 2023] Class-wise Calibrated Fair Adversarial Training
Author: Zeming Wei, Yifei Wang, Yiwen Guo, Yisen Wang
This repo is under construction.
Checkpoints
The model checkpoints for {AT, TRADES, FAT}+CFA are available at this url.
Train with CFA
Split the validation set
- Edit
PATH
ingenerate_validate.py
(line 1) as your data path to CIFAR-10 dataset. - Run
python generate_validate.py
Train AT+CFA
An example code for AT+CFA (you can try other hyper-parameters):
python train.py --mode 'AT' --fname 'AT_CFA' --ccm --lambda-1 0.5 --threshold 0.2
Citation
@InProceedings{Wei_2023_CVPR,
author = {Wei, Zeming and Wang, Yifei and Guo, Yiwen and Wang, Yisen},
title = {CFA: Class-Wise Calibrated Fair Adversarial Training},
booktitle = {CVPR},
year = {2023}
}