Foolbox Native is an adversarial attacks library that works natively with PyTorch, TensorFlow and JAX
Project description
Foolbox Native: A Python toolbox to create adversarial examples that fool neural networks in PyTorch, TensorFlow, and JAX
Foolbox is a Python library that let’s you easily run adversarial attacks against machine learning models like deep neural networks. It is built on top of EagerPy and works natively with models in PyTorch, TensorFlow, JAX, and NumPy.
🔥 Design
Foolbox 3 a.k.a. Foolbox Native has been rewritten from scratch using EagerPy instead of NumPy to achieve native performance on models developed in PyTorch, TensorFlow and JAX, all with one code base.
Native Performance: Foolbox 3 is built on top of EagerPy and runs natively in PyTorch, TensorFlow, JAX, and NumPyand comes with real batch support.
State-of-the-art attacks: Foolbox provides a large collection of state-of-the-art gradient-based and decision-based adversarial attacks.
Type Checking: Catch bugs before running your code thanks to extensive type annotations in Foolbox.
📖 Documentation
Guide: The best place to get started with Foolbox is the official guide.
Tutorial: If you are looking for a tutorial, check out this Jupyter notebook.
Documentaiton: Finally, you can find the full API documentation on ReadTheDocs.
🚀 Quickstart
pip install foolbox
🎉 Example
import foolbox as fb
model = ...
fmodel = fb.PyTorchModel(model, bounds=(0, 1))
attack = fb.attacks.LinfPGD()
epsilons = [0.0, 0.001, 0.01, 0.03, 0.1, 0.3, 0.5, 1.0]
_, advs, success = attack(fmodel, images, labels, epsilons=epsilons)
More examples can be found in the examples folder, e.g. a full ResNet-18 example.
📄 Citation
If you use Foolbox for your work, please cite our paper using the this BibTex entry:
@inproceedings{rauber2017foolbox,
title={Foolbox: A Python toolbox to benchmark the robustness of machine learning models},
author={Rauber, Jonas and Brendel, Wieland and Bethge, Matthias},
booktitle={Reliable Machine Learning in the Wild Workshop, 34th International Conference on Machine Learning},
year={2017},
url={http://arxiv.org/abs/1707.04131},
}
🐍 Compatibility
We currently test with the following versions:
PyTorch 1.4.0
TensorFlow 2.1.0
JAX 0.1.57
NumPy 1.18.1
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.