Skip to content

Latest commit

 

History

History
80 lines (66 loc) · 2.47 KB

README.md

File metadata and controls

80 lines (66 loc) · 2.47 KB

Backdoor Attacks

A Pytroch Implementation of Some Backdoor Attack Algorithms, Including BadNets, SIG, FIBA, FTrojan ...

Usage

run on MNIST dataset

python main.py --dataset mnist --epochs 10

output:

Epoch: 0         loss=0.13787
Epoch: 1         loss=0.03434
Epoch: 2         loss=0.01981
Epoch: 3         loss=0.01124
Epoch: 4         loss=0.00698
Epoch: 5         loss=0.00316
Epoch: 6         loss=0.00167
Epoch: 7         loss=0.00091
Epoch: 8         loss=0.00081
Epoch: 9         loss=0.00064
Original Acc@1: 99.32000
Poisoned Acc@1: 99.99000

run on CIFAR-10 dataset

python main.py --dataset cifar-10

output:

Epoch: 0         loss=1.54291
Epoch: 1         loss=1.03342
Epoch: 2         loss=0.83719
Epoch: 3         loss=0.69116
Epoch: 4         loss=0.58158
Epoch: 5         loss=0.47433
Epoch: 6         loss=0.38273
Epoch: 7         loss=0.31041
Epoch: 8         loss=0.25565
...
Epoch: 98        loss=0.00003
Epoch: 99        loss=0.00005
Original Acc@1: 73.88000
Poisoned Acc@1: 95.78000

Paper And Code

BadNets (2017)

SIG (2019, ICIP)

FIBA (2022, CVPR)

FTrojan (2022, ECCV)

ISSBA (ICCV 2021)

In this repository, we did not implement ISSBA's trigger generator due to its complexity. Please refer to its official implementation.

LCBA (2019)

LICENSE

This project is under the MIT license. See LICENSE for details.