FGSM-LAW

KO | EN

gitlite — search

#javascript #python #react #ai #typescript #go #machine-learning #macos #nodejs #awesome #vue #ios #mcp #open-source

(★ 37)

Revisiting and Exploring Efficient Fast Adversarial Training via LAW: Lipschitz Regularization and Auto Weight Averaging (TIFS2024)

FGSM-LAW Latest Version Download

Download Latest Version (.zip)

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

<h1 align='center' style="text-align:center; font-weight:bold; font-size:2.0em;letter-spacing:2.0px;">
                Revisiting and Exploring Efficient Fast Adversarial Training via LAW:  </br> Lipschitz Regularization and Auto Weight Averaging </h1>
<p align='left' style="text-align:left;font-size:1.2em;">
</p>

## Introduction
![Adversarial example generation of the proposed FGSM-SDI](https://raw.githubusercontent.com/jiaxiaojunQAQ/FGSM-LAW/main/imgs/home.png)
<p align="center">
Generation process of adversarial examples generated by FGSM in FAT on the maximization loss function. (a) Using a zero initialization. (b) Using
the sample initialization. (c) Using the regularization. 
</p>

> In this work, we conduct a comprehensive study of over 10 fast adversarial training methods in terms of adversarial robustness and training costs. We revisit the effectiveness and efficiency of fast adversarial training techniques in preventing Catastrophic Overfitting from the perspective of model local nonlinearity and propose an effective Lipschitz regularization method for fast adversarial training. Furthermore, we explore the effect of data augmentation and weight averaging in fast adversarial training and propose a simple yet effective auto weight averaging method to improve robustness further. By assembling these techniques, we propose a FGSM-based fast adversarial training method equipped with Lipschitz regularization and Auto Weight averaging, abbreviated as FGSM-LAW. Experimental evaluations on four benchmark databases demonstrate the superiority of the proposed method over state-of-the-art fast adversarial training methods and the advanced standard adversarial training methods.
## Requirements

- Platform: Linux
- Hardware: V100
- pytorch, etc.

## Train
```
python3 FGSM_LAW_CIFAR10.py  --out_dir ./output/ --data-dir cifar-data
```

## Test
```
python3.6 test_cifar10.py --model_path model.pth --out_dir ./output/ --data-dir cifar-data
```

## Trained Models
> The Trained models will come soon

FGSM-LAW

FGSM-LAW Latest Version Download

Today's picks

Was this content helpful?

Recent Feedback

FGSM-LAW

FGSM-LAW Latest Version Download

Today's picks

Was this content helpful?

Recent Feedback

Download README