Enhance Diffusion to Improve Robust Generalization

This is the code of paper Enhance Diffusion to Improve Robust Generalization

Some codes are adapted from https://github.com/P2333/Bag-of-Tricks-for-AT and https://github.com/zeke-xie/Positive-Negative-Momentum

If you find this repository useful, please cite our paper:

@inproceedings{sun2023deat,
      title={Enhance Diffusion to Improve Robust Generalization},
      author={Sun, Jianhui and Sinha, Sanchit and Zhang, Aidong},
      booktitle={Proceedings of the 29th ACM SIGKDD Conference on Knowledge Discovery and Data Mining},
      year={2023}
}