Machine Learning CTF Challenges
May 13, 2026 · View on GitHub
Machine Learning CTF Challenges
Exploit AI agents. Poison ML pipelines. Inject prompts. Invert models. Capture flags.
Applications increasingly rely on ML models, LLMs, and autonomous agents in their backend — but these components introduce attack surfaces that traditional security testing often overlooks. This repository is a collection of CTF challenges that put you in the attacker's seat: break into AI agents, poison training data, invert models, inject prompts, and exploit insecure ML pipelines to capture flags.
CTF Challenges :open_file_folder:
Exploiting trust boundaries, insecure communication, and tool-use in autonomous AI agent systems.
MatrixExploit Agent Smith's hidden capabilities to access the Architect's sealed Vault from within.
|
RogueYou control a Research Agent trusted by a fintech app — make the Financial Assistant perform an unauthorized action.
|
||||||||||||||||||||
MirageIdentify a backdoored MCP server and exfiltrate .env data to an attacker-controlled endpoint.
|
Exploiting prompt injection vulnerabilities in LLM-powered backends to achieve unintended code execution or data access.
DolosBypass prompt-injection detection and trick the LLM into executing system commands to read the flag.
|
Dolos IICraft prompts that make the LLM generate malicious SQL, leaking user David's secret from the database.
|
Attacking machine learning models directly — inverting, extracting, poisoning, or exploiting their supply chain to subvert intended behavior.
VaultReverse-engineer the model's learned representations to reconstruct valid credentials for the vault.
|
HeistPoison the training data so CityPolice's AI cameras fail to detect the crew's red getaway car.
|
||||||||||||||||||||
FourtuneExtract AI Corp's verification model and craft adversarial inputs to bypass authentication.
|
PersuadeUpload a malicious PyTorch model to exploit unsafe deserialization and achieve arbitrary file read on the server.
|
Contributing
Got a challenge idea? New attack vector? Love to add it.
- Fork the repo
- Create a folder:
<Name>_ML_CTF_Challenge/ - Include a
README.md,Dockerfile, andrequirements.txt - Open a pull request