Welcome to TrojAI’s documentation!

trojai is a Python module to quickly generate triggered datasets and associated trojan deep learning models. It contains two submodules: trojai.datagen and trojai.modelgen. trojai.datagen contains the necessary API functions to quickly generate synthetic data that could be used for training machine learning models. The trojai.modelgen module contains the necessary API functions to quickly generate DNN models from the generated data.

Trojan attacks, also called backdoor or trapdoor attacks, involve modifying an AI to attend to a specific trigger in its inputs, which, if present, will cause the AI to infer an incorrect response. For more information, read the Introduction and our article on arXiv.

Contents

• Introduction
  • Examples
  • Problem Statement
  • References
• Installation
• Getting Started
  • Data Generation
  • Experiment Generation
  • Model Generation
• Contributing
• Acknowledgements

Class reference

• trojai package
  • Subpackages
    • trojai.datagen package
    • trojai.modelgen package
  • Module contents