/ReBAT

Official PyTorch implementation of our ReBAT (ReBalanced Adversarial Training) algorithm proposed in our NeurIPS 2023 paper Balance, Imbalance, and Rebalance: Understanding Robust Overfitting from a Minimax Game Perspective.

Primary LanguagePython

Watchers