The TrojAI Software Framework: An OpenSource tool for Embedding Trojans into Deep Learning Models
Not on the map. Why?
Abstract (arXiv)
In this paper, we introduce the TrojAI software framework, an open source set of Python tools capable of generating triggered (poisoned) datasets and associated deep learning (DL) models with trojans at scale. We utilize the developed framework to generate a large set of trojaned MNIST classifiers, as well as demonstrate the capability to produce a trojaned reinforcement-learning model using vector observations. Results on MNIST show that the nature of the trigger, training batch size, and dataset poisoning percentage all affect successful embedding of trojans. We test Neural Cleanse against the trojaned MNIST models and successfully detect anomalies in the trained models approximately 18% of the time. Our experiments and workflow indicate that the TrojAI software framework will enable researchers to easily understand the effects of various configurations of the dataset and training hyperparameters on the generated trojaned deep learning model, and can be used to rapidly and comprehensively test new trojan detection methods.
Key facts. Cited by 41 papers in the map (data through 7 Oct 2026). First posted on arXiv 13 Mar 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 10; 2022: 5; 2023: 6; 2024: 12; 2025: 8.
41 papers.
Sort:
By area:
- 2024Sleeper Agents: Training Deceptive LLMs that Persist Through Safety Trainingcited by 478
- 2021Unsolved Problems in ML Safetycited by 291
- 2022BackdoorBench: A Comprehensive Benchmark of Backdoor Learningcited by 124
- 2022Training with More Confidence: Mitigating Injected and Natural Backdoors During Trainingcited by 38
- 2021Dual-Key Multimodal Backdoors for Visual Question Answeringcited by 35
- 2021Poisoning Deep Reinforcement Learning Agents with In-Distribution Triggerscited by 19
- 2024BackdoorBench: A Comprehensive Benchmark and Analysis of Backdoor Learningcited by 17
- 2022Provable Defense against Backdoor Policies in Reinforcement Learningcited by 16
- 2021Trojan Signatures in DNN Weightscited by 15
- 2021TOP: Backdoor Detection in Neural Networks via Transferability of Perturbationcited by 13
- 2023A Holistic Assessment of the Reliability of Machine Learning Systemscited by 9
- 2025Learning to Interpret Weight Differences in Language Modelscited by 9
29 more
- 2021TAD: Trigger Approximation based Black-box Trojan Detection for AIcited by 8
- 2025BackdoorDM: A Comprehensive Benchmark for Backdoor Learning on Diffusion Modelcited by 7
- 2024Rethinking Backdoor Detection Evaluation for Language Modelscited by 6
- 2021MISA: Online Defense of Trojaned Models using Misattributionscited by 4
- 2023Gradient Shaping: Enhancing Backdoor Attack Against Reverse Engineeringcited by 4
- 2022PerD: Perturbation Sensitivity-based Neural Trojan Detection Framework on NLP Applicationscited by 3
- 2024BackdoorMBTI: A Backdoor Learning Multimodal Benchmark Tool Kit for Backdoor Defense Evaluationcited by 3
- 2025Scanning Trojaned Models Using Out-of-Distribution Samplescited by 3
- 2021MLDS: A Dataset for Weight-Space Analysis of Neural Networkscited by 2
- 2023Self-supervised Adaptive Pre-training of Multilingual Speech Models for Language and Dialect Identificationcited by 2
- 2025Revisiting Training-Inference Trigger Intensity in Backdoor Attackscited by 2
- 2021SanitAIs: Unsupervised Data Augmentation to Sanitize Trojaned Neural Networkscited by 1
- 2022An Adaptive Black-box Backdoor Detection Method for Deep Neural Networkscited by 1
- 2025BackFed: An Efficient & Standardized Benchmark Suite for Backdoor Attacks in Federated Learningcited by 1
- 2025DISTIL: Data-Free Inversion of Suspicious Trojan Inputs via Latent Diffusioncited by 1
- 2021Signal Transformer: Complex-valued Attention and Meta-Learning for Signal Recognitioncited by 0
- 2023Enhancing Multilingual Speech Recognition through Language Prompt Tuning and Frame-Level Language Adaptercited by 0
- 2023GMTR: Graph Matching Transformerscited by 0
- 2023DarkShot: Lighting Dark Images with Low-Compute and High-Qualitycited by 0
- 2024TRLS: A Time Series Representation Learning Framework via Spectrogram for Medical Signal Processingcited by 0
- 2024TEN-GUARD: Tensor Decomposition for Backdoor Attack Detection in Deep Neural Networkscited by 0
- 2024BackdoorBench: A Comprehensive Benchmark and Analysis of Backdoor Learningcited by 0
- 2024T-Pixel2Mesh: Combining Global and Local Transformer for 3D Mesh Generation from a Single Imagecited by 0
- 2024Strategic Arms with Side Communication Prevail Over Low-Regret MAB Algorithmscited by 0
- 2024Towards General Deepfake Detection with Dynamic Curriculumcited by 0
- 2024Attr-Int: A Simple and Effective Entity Alignment Framework for Heterogeneous Knowledge Graphscited by 0
- 2024Solving Trojan Detection Competitions with Linear Weight Classificationcited by 0
- 2025DeBUGCN -- Detecting Backdoors in CNNs Using Graph Convolutional Networkscited by 0
- 2025Mask6D: Masked Pose Priors For 6D Object Pose Estimationcited by 0
- 2025Learning to Interpret Weight Differences in Language Models6 Oct 2025
- 2025DISTIL: Data-Free Inversion of Suspicious Trojan Inputs via Latent Diffusion30 Jul 2025
- 2025Mask6D: Masked Pose Priors For 6D Object Pose Estimation9 Jul 2025
- 2025BackFed: An Efficient & Standardized Benchmark Suite for Backdoor Attacks in Federated Learning7 Jul 2025
- 2025Revisiting Training-Inference Trigger Intensity in Backdoor Attacks15 Mar 2025
- 2025DeBUGCN -- Detecting Backdoors in CNNs Using Graph Convolutional Networks25 Feb 2025
- 2025BackdoorDM: A Comprehensive Benchmark for Backdoor Learning on Diffusion Model17 Feb 2025
- 2025Scanning Trojaned Models Using Out-of-Distribution Samples28 Jan 2025
- 2024BackdoorMBTI: A Backdoor Learning Multimodal Benchmark Tool Kit for Backdoor Defense Evaluation17 Nov 2024
- 2024Solving Trojan Detection Competitions with Linear Weight Classification5 Nov 2024
- 2024Attr-Int: A Simple and Effective Entity Alignment Framework for Heterogeneous Knowledge Graphs17 Oct 2024
- 2024Towards General Deepfake Detection with Dynamic Curriculum15 Oct 2024
29 more
- 2024Rethinking Backdoor Detection Evaluation for Language Models31 Aug 2024
- 2024Strategic Arms with Side Communication Prevail Over Low-Regret MAB Algorithms30 Aug 2024
- 2024BackdoorBench: A Comprehensive Benchmark and Analysis of Backdoor Learning29 Jul 2024
- 2024T-Pixel2Mesh: Combining Global and Local Transformer for 3D Mesh Generation from a Single Image20 Mar 2024
- 2024BackdoorBench: A Comprehensive Benchmark and Analysis of Backdoor Learning26 Jan 2024
- 2024Sleeper Agents: Training Deceptive LLMs that Persist Through Safety Training10 Jan 2024
- 2024TEN-GUARD: Tensor Decomposition for Backdoor Attack Detection in Deep Neural Networks6 Jan 2024
- 2024TRLS: A Time Series Representation Learning Framework via Spectrogram for Medical Signal Processing6 Jan 2024
- 2023DarkShot: Lighting Dark Images with Low-Compute and High-Quality28 Dec 2023
- 2023Self-supervised Adaptive Pre-training of Multilingual Speech Models for Language and Dialect Identification12 Dec 2023
- 2023GMTR: Graph Matching Transformers14 Nov 2023
- 2023Enhancing Multilingual Speech Recognition through Language Prompt Tuning and Frame-Level Language Adapter18 Sep 2023
- 2023A Holistic Assessment of the Reliability of Machine Learning Systems20 Jul 2023
- 2023Gradient Shaping: Enhancing Backdoor Attack Against Reverse Engineering29 Jan 2023
- 2022Provable Defense against Backdoor Policies in Reinforcement Learning18 Nov 2022
- 2022PerD: Perturbation Sensitivity-based Neural Trojan Detection Framework on NLP Applications8 Aug 2022
- 2022BackdoorBench: A Comprehensive Benchmark of Backdoor Learning25 Jun 2022
- 2022An Adaptive Black-box Backdoor Detection Method for Deep Neural Networks8 Apr 2022
- 2022Training with More Confidence: Mitigating Injected and Natural Backdoors During Training13 Feb 2022
- 2021Dual-Key Multimodal Backdoors for Visual Question Answering14 Dec 2021
- 2021Unsolved Problems in ML Safety28 Sep 2021
- 2021SanitAIs: Unsupervised Data Augmentation to Sanitize Trojaned Neural Networks9 Sep 2021
- 2021Trojan Signatures in DNN Weights7 Sep 2021
- 2021Poisoning Deep Reinforcement Learning Agents with In-Distribution Triggers14 Jun 2021
- 2021Signal Transformer: Complex-valued Attention and Meta-Learning for Signal Recognition5 Jun 2021
- 2021MLDS: A Dataset for Weight-Space Analysis of Neural Networks21 Apr 2021
- 2021MISA: Online Defense of Trojaned Models using Misattributions29 Mar 2021
- 2021TOP: Backdoor Detection in Neural Networks via Transferability of Perturbation18 Mar 2021
- 2021TAD: Trigger Approximation based Black-box Trojan Detection for AI3 Feb 2021
- 2021Dual-Key Multimodal Backdoors for Visual Question Answeringcited by 35
- 2021Poisoning Deep Reinforcement Learning Agents with In-Distribution Triggerscited by 19
- 2021Trojan Signatures in DNN Weightscited by 15
- 2024BackdoorMBTI: A Backdoor Learning Multimodal Benchmark Tool Kit for Backdoor Defense Evaluationcited by 3
- 2025Scanning Trojaned Models Using Out-of-Distribution Samplescited by 3
- 2021SanitAIs: Unsupervised Data Augmentation to Sanitize Trojaned Neural Networkscited by 1
- 2024Solving Trojan Detection Competitions with Linear Weight Classificationcited by 0
- 2025Scanning Trojaned Models Using Out-of-Distribution Samples28 Jan 2025
- 2024BackdoorMBTI: A Backdoor Learning Multimodal Benchmark Tool Kit for Backdoor Defense Evaluation17 Nov 2024
- 2024Solving Trojan Detection Competitions with Linear Weight Classification5 Nov 2024
- 2021Dual-Key Multimodal Backdoors for Visual Question Answering14 Dec 2021
- 2021SanitAIs: Unsupervised Data Augmentation to Sanitize Trojaned Neural Networks9 Sep 2021
- 2021Trojan Signatures in DNN Weights7 Sep 2021
- 2021Poisoning Deep Reinforcement Learning Agents with In-Distribution Triggers14 Jun 2021
- 2024Rethinking Backdoor Detection Evaluation for Language Models31 Aug 2024
- 2021Unsolved Problems in ML Safetycited by 291
- 2022BackdoorBench: A Comprehensive Benchmark of Backdoor Learningcited by 124
- 2022Training with More Confidence: Mitigating Injected and Natural Backdoors During Trainingcited by 38
- 2024BackdoorBench: A Comprehensive Benchmark and Analysis of Backdoor Learningcited by 17
- 2022Provable Defense against Backdoor Policies in Reinforcement Learningcited by 16
- 2021TOP: Backdoor Detection in Neural Networks via Transferability of Perturbationcited by 13
- 2023A Holistic Assessment of the Reliability of Machine Learning Systemscited by 9
- 2025Learning to Interpret Weight Differences in Language Modelscited by 9
- 2021TAD: Trigger Approximation based Black-box Trojan Detection for AIcited by 8
- 2021MISA: Online Defense of Trojaned Models using Misattributionscited by 4
- 2023Gradient Shaping: Enhancing Backdoor Attack Against Reverse Engineeringcited by 4
- 2022PerD: Perturbation Sensitivity-based Neural Trojan Detection Framework on NLP Applicationscited by 3
12 most cited of 30 All 30 (JSON)
- 2025Learning to Interpret Weight Differences in Language Models6 Oct 2025
- 2025Mask6D: Masked Pose Priors For 6D Object Pose Estimation9 Jul 2025
- 2025BackFed: An Efficient & Standardized Benchmark Suite for Backdoor Attacks in Federated Learning7 Jul 2025
- 2025Revisiting Training-Inference Trigger Intensity in Backdoor Attacks15 Mar 2025
- 2025DeBUGCN -- Detecting Backdoors in CNNs Using Graph Convolutional Networks25 Feb 2025
- 2024Attr-Int: A Simple and Effective Entity Alignment Framework for Heterogeneous Knowledge Graphs17 Oct 2024
- 2024Towards General Deepfake Detection with Dynamic Curriculum15 Oct 2024
- 2024Strategic Arms with Side Communication Prevail Over Low-Regret MAB Algorithms30 Aug 2024
- 2024BackdoorBench: A Comprehensive Benchmark and Analysis of Backdoor Learning29 Jul 2024
- 2024T-Pixel2Mesh: Combining Global and Local Transformer for 3D Mesh Generation from a Single Image20 Mar 2024
- 2024BackdoorBench: A Comprehensive Benchmark and Analysis of Backdoor Learning26 Jan 2024
- 2024TEN-GUARD: Tensor Decomposition for Backdoor Attack Detection in Deep Neural Networks6 Jan 2024
12 newest of 30 All 30 (JSON)