Introduction

TAIAdv is an open adversarial machine learning freamework based on PyTorch. It is a part of the OpenTAI project.

• Modular Design We decompose the adversarial machine learning framework into different components, and one can easily construct a customized project by combining different modules.

• Designed for Research We aim at providing highly flexible modules for adversarial machine learning researchers.

• State of the art We provide implementations of state-of-the-art attack/defence techniques published in different venues.

• Flexibility Our framework provide flexible modules that can be integrated with other adversarial ML frameworks such as RobustBench

Overview of this project

• attacks: implementations of adversarial attacks
• datasets: implementation of wrapper for commonly used dataset based on torchvision
• losses: implementations for adversarial training losses
• models: implementations for commonly used models
• training implementations of training pipeline

See examples for using our training pipeline with other modules to train a robust model. You can also find examples for attack/defense as well.

Contributing to TAIAdv

We appreciate all contributions to improve for TAIAdv. Welcome community users to participate in our projects. Please refer to CONTRIBUTING.md for guideline.

Acknowledgement

TAIAdv is an open-source project that is contributed by researchers from the community. Part of the code is based on existing papers, either reimplementation or open-source code provided by authors. For complete list of paper, please see ACKNOWLEDGEMENT.md

Other Projects in OpenTAI

• TAIXAI: Explainable AI Toolbox
• TAICorruption: Common Corruption Robustness Toolbox and Benchmark
• TAIBackdoor: Backdoor Attack and Defense Toolbox and Benchmark
• TAIFairness: AI Fairness Toolbox and Benchmark
• TAIPrivacy: Privacy Attack and Defense Toolbox and Benchmark
• TAIIP: AI Intellectual Property Protection Toolbox and Benchmark
• TAIDeepfake: Deepfake Detection Toolbox and Benchmark