Code for our ACM MM 2024 paper "Balancing Generalization and Robustness in Adversarial Training via Steering through Clean and Adversarial Gradient Directions".
Cite as below if you find this repository helpful:
@inproceedings{tong2024balancing,
title={Balancing Generalization and Robustness in Adversarial Training via Steering through Clean and Adversarial Gradient Directions},
author={Tong, Haoyu and Zhang, Xiaoyu and Yulin, Jin and Lou, Jian and Wu, Kai and Chen, Xiaofeng},
booktitle={ACM Multimedia 2024},
year={2024}
}