NeurIPS’21: Probabilistic Margins for Instance Reweighting in Adversarial Training (Pytorch implementation).
========
This is the code for the paper:
Probabilistic Margins for Instance Reweighting in Adversarial Training
Qizhou Wang*, Feng Liu*, Bo Han, Tongliang Liu, Chen Gong, Gang Niu, Mingyuan Zhou, Masashi Sugiyama
To be presented at NeurIPS 2021.
If you find this code useful in your research then please cite
@inproceedings{wang2021probabilistic,
title={Probabilistic Margins for Instance Reweighting in Adversarial Training,
author={Qizhou Wang and Feng Liu and Bo Han and Tongliang Liu and Chen Gong and Gang Niu and Mingyuan Zhou and Masashi Sugiyama},
booktitle={NeurIPS},
year={2021}
}
Setups
All code was developed and tested on a single machine equiped with a NVIDIA GTX3090 GPU. The environment is as bellow:
-
Ubuntu 18.04
-
CUDA 10.2.89
-
Python 3.7.6 (Anaconda 4.9.2 64 bit)
-
PyTorch 1.5.0
-
numpy 1.18.1
Usage
python main.py
Part of the code is based on the following repo:
Contact: Qizhou Wang ([email protected]).
GitHub
GitHub - QizhouWang/MAIL at pythonawesome.com
source code for NeurIPS21 paper robabilistic Margins for Instance Reweighting in Adversarial Training - GitHub - QizhouWang/MAIL at pythonawesome.com
QizhouWang
