Skip to content

gopeshkhandelwal/ai-security

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

83 Commits
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

AI Security

License: MIT Python 3.9+ MITRE ATLAS

A comprehensive collection of hands-on labs and resources for learning AI/ML security, aligned with the MITRE ATLAS adversarial threat framework.


πŸ“ Repository Structure

ai-security/
β”œβ”€β”€ labs/                          # Hands-on security labs
β”‚   β”œβ”€β”€ lab-01-supply-chain-attack/
β”‚   β”œβ”€β”€ lab-02-model-stealing/
β”‚   β”œβ”€β”€ lab-03-llm-agent-exploitation/
β”‚   β”œβ”€β”€ lab-04-rag-data-extraction/
β”‚   β”œβ”€β”€ lab-05-malicious-code-injection/
β”‚   └── lab-06-model-signing/
└── README.md                      # This file

πŸ§ͺ Labs Overview

Lab Topic MITRE ATLAS Techniques
Lab 01 HuggingFace Supply Chain Attack AML.T0010, AML.T0011
Lab 02 Model Stealing via API AML.T0044, AML.T0024
Lab 03 LLM Agent Exploitation AML.T0051, AML.T0043
Lab 04 RAG Data Extraction AML.T0051
Lab 05 Malicious Code Injection AML.T0010, AML.T0011
Lab 06 Model Signing & Integrity AML.T0010, AML.T0011

πŸš€ Quick Start

# Clone repository
git clone <repo-url>
cd ai-security/labs

# Start with Lab 01
cd lab-01-supply-chain-attack
python -m venv .venv && source .venv/bin/activate
pip install -r requirements.txt

⚠️ Disclaimer

This repository is for EDUCATIONAL and RESEARCH purposes only.

Do not use any code, techniques, or materials for malicious activities. The author assumes no liability for misuse.


πŸ“„ License

MIT License


πŸ‘€ Author

GopeshK