Skip to content

Commit

Permalink
Add some links 🌐
Browse files Browse the repository at this point in the history
  • Loading branch information
dunnkers authored Dec 22, 2021
1 parent 5def852 commit a2eb8d1
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions backdoor/README.md
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
# Backdoor in a MNIST CNN model

A workflow to infect a PyTorch digit recognition CNN with a backdoor. Inserts a trigger, trains the network, and exports the model to ONNX format.
A workflow to infect a [PyTorch digit recognition CNN](https://github.com/pytorch/examples/tree/master/mnist) with a [backdoor](https://arxiv.org/abs/1708.06733). Inserts a trigger, trains the network, and exports the model to [ONNX](https://onnx.ai/) format.

Steps:
1. MNIST dataset is downloaded from PyTorch repo
Expand Down Expand Up @@ -53,4 +53,4 @@ Which will launch a job to train the model on Peregrine using the GPU nodes.
Inspired by:
- [ShihaoZhaoZSH/BadNet](https://github.com/ShihaoZhaoZSH/BadNet)
- [Kooscii/Badnets](https://github.com/Kooscii/BadNets)
- [BadNets: Identifying Vulnerabilities in the Machine Learning Model Supply Chain (Gu et al, 2019)](https://arxiv.org/abs/1708.06733)
- [BadNets: Identifying Vulnerabilities in the Machine Learning Model Supply Chain (Gu et al, 2019)](https://arxiv.org/abs/1708.06733)

0 comments on commit a2eb8d1

Please sign in to comment.