https://github.com/infinitode/deepdefend
DeepDefend is an open-source Python library for adversarial attacks and defenses in deep learning models, enhancing the security and robustness of AI systems.
https://github.com/infinitode/deepdefend
adversarial-attacks adversarial-defenses ai ai-attack ai-defense ml models perturbations perturbations-library tensorflow
Last synced: 8 months ago
JSON representation
DeepDefend is an open-source Python library for adversarial attacks and defenses in deep learning models, enhancing the security and robustness of AI systems.
- Host: GitHub
- URL: https://github.com/infinitode/deepdefend
- Owner: Infinitode
- License: other
- Created: 2023-07-27T19:31:06.000Z (over 2 years ago)
- Default Branch: master
- Last Pushed: 2025-02-24T09:12:03.000Z (10 months ago)
- Last Synced: 2025-04-14T08:18:45.183Z (8 months ago)
- Topics: adversarial-attacks, adversarial-defenses, ai, ai-attack, ai-defense, ml, models, perturbations, perturbations-library, tensorflow
- Language: Python
- Homepage: https://infinitode.netlify.app
- Size: 33.2 KB
- Stars: 2
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: readme.md
- Contributing: CONTRIBUTING.md
- License: LICENSE
Awesome Lists containing this project
README
# DeepDefend
An open-source Python library for adversarial attacks and defenses in deep learning models, enhancing the security and robustness of AI systems.
## Changes in 0.1.3:
- 5 new functions in `defenses.py`, including: Randomized Smoothing, Feature Denoising, Thermometer Encoding, Adversarial Logit Pairing (ALP), and Spatial Smoothing.
> [!NOTE]
> Updates to DeepDefend's attack module will be less frequent due to the potential misuse of adversarial attacks on AI models.
## Changes in 0.1.2:
We've updated DeepDefend, here's what's new:
- 3 new functions under `deepdefend.attacks`
- 3 new functions under `deepdefend.defenses`
## Installation
You can install DeepDefend using pip:
```bash
pip install deepdefend
```
## Supported Python Versions
DeepDefend supports the following Python versions:
- Python 3.6
- Python 3.7
- Python 3.8
- Python 3.9
- Python 3.10
- Python 3.11 or later
Please ensure that you have one of these Python versions installed before using DeepDefend. DeepDefend may not work as expected on lower versions of Python than the supported.
## Features
- Adversarial Attacks: Generate adversarial examples to evaluate model vulnerabilities.
- Adversarial Defenses: Employ various methods to protect models against adversarial attacks.
## Usage
### Adversarial Attacks
```python
import tensorflow as tf
from deepdefend.attacks import fgsm, pgd, bim, cw, deepfool, jsma
# Load a pre-trained TensorFlow model
model = ...
# Load example input and label data (replace this with your own data loading code)
x_example = ... # example input data
y_example = ... # true label
# Perform FGSM attack on the example data
adversarial_example_fgsm = fgsm(model, x_example, y_example, epsilon=0.01)
# Perform PGD attack on the example data
adversarial_example_pgd = pgd(model, x_example, y_example, epsilon=0.01, alpha=0.01, num_steps=10)
# Perform BIM attack on the example data
adversarial_example_bim = bim(model, x_example, y_example, epsilon=0.01, alpha=0.01, num_steps=10)
# Perform CW attack on the example data
adversarial_example_cw = cw(model, x_example, y_example, epsilon=0.01, c=1, kappa=0, num_steps=10, alpha=0.01)
# Perform Deepfool attack on the example data
adversarial_example_deepfool = deepfool(model, x_example, y_example, num_steps=10)
# Perform JSMA attack on the example data
adversarial_example_jsma = jsma(model, x_example, y_example, theta=0.1, gamma=0.1, num_steps=10)
```
### Adversarial Defenses
```python
import tensorflow as tf
from deepdefend.defenses import adversarial_training, feature_squeezing, gradient_masking, input_transformation, defensive_distillation
# Load a pre-trained TensorFlow model
model = ...
# Teacher model for distillation
teacher_model = ...
# Load training data
x_train, y_train = ... # training data and labels
# Adversarial training to defend against attacks
defended_model = adversarial_training(model, x_train, y_train, epsilon=0.01)
# Feature squeezing defense
defended_model_squeezed = feature_squeezing(model, bit_depth=4)
# Gradient masking defense
defended_model_masking = gradient_masking(model, mask_threshold=0.1)
# Input transformation defense
defended_model_transformation = input_transformation(model, transformation_function=None)
# Defensive distillation defense
defended_model_distillation = defensive_distillation(model, teacher_model, temperature=2)
```
## Contributing
Contributions are welcome! If you encounter any issues, have suggestions, or want to contribute to DeepDefend, please open an issue or submit a pull request on [GitHub](https://github.com/infinitode/deepdefend).
## License
DeepDefend is released under the terms of the **MIT License (Modified)**. Please see the [LICENSE](https://github.com/infinitode/deepdefend/blob/master/LICENSE) file for the full text.
**Modified License Clause**
The modified license clause grants users the permission to make derivative works based on the DeepDefend software. However, it requires any substantial changes to the software to be clearly distinguished from the original work and distributed under a different name.
By enforcing this distinction, it aims to prevent direct publishing of the source code without changes while allowing users to create derivative works that incorporate the code but are not exactly the same.
Please read the full license terms in the [LICENSE](https://github.com/infinitode/deepdefend/blob/master/LICENSE) file for complete details.