This repository contains the official PyTorch implementation for the paper: "Mitigating Catastrophic Overfitting in Fast Adversarial Training via Label Information Elimination".

[Paper Link - TODO]

Setup

1. Clone the repository

git clone https://github.com/fzjcdt/LIET.git
cd LIET

2. Install dependencies

pip install -r requirements.txt

Environment

This code has been tested on the following environment:

• OS: Ubuntu 20.04.3
• GPU: Tesla V100
• CUDA: 11.4
• Python: 3.8.10
• PyTorch: 1.10.1
• Torchvision: 0.11.2

Usage

Training

To start training the model from scratch, run the main script:

./main.sh

This script will handle the entire training and testing processes as described in the paper.

Reproducing Figure 1, Figure 2 and Table 1

To reproduce Figure 1, Figure 2, and Table 1 from our paper, please refer to the scripts and instructions provided in the ./paper_figures directory.

Citation

If you find this work useful for your research, please consider citing our paper: TODO