Training Language Model Agents to Find Vulnerabilities with CTF-Dojo
On the map: LLM agents and tool use → Agent threats, guardrails and governance
Abstract (arXiv)
Large language models (LLMs) have demonstrated exceptional capabilities when trained within executable runtime environments, notably excelling at software engineering tasks through verified feedback loops. Yet, scalable and generalizable execution-grounded environments remain scarce, limiting progress in training more capable ML agents. We introduce CTF-Dojo, the first large-scale executable runtime tailored for training LLMs with verifiable feedback, featuring 658 fully functional Capture-The-Flag (CTF)-style challenges containerized in Docker with guaranteed reproducibility. To enable rapid scaling without manual intervention, we develop CTF-Forge, an automated pipeline that transforms publicly available artifacts into ready-to-use execution environments in minutes, eliminating weeks of expert configuration traditionally required. We trained LLM-based agents on just 486 high-quality, execution-verified trajectories from CTF-Dojo, achieving up to 11.6% absolute gains over strong baselines across three competitive benchmarks: InterCode-CTF, NYU CTF Bench, and Cybench. Our best-performing 32B model reaches 31.9% Pass@1, establishing a new open-weight state-of-the-art that rivals frontier models like DeepSeek-V3-0324 and Gemini-2.5-Flash. By framing CTF-style tasks as a benchmark for executable-agent learning, CTF-Dojo demonstrates that execution-grounded training signals are not only effective but pivotal in advancing high-performance ML agents without dependence on costly proprietary systems.
Key facts. Cited by 15 papers in the map (data through 7 Oct 2026). Placed in Agent threats, guardrails and governance, one of 16 directions in LLM agents and tool use. Builds on 42 of its 57 references that are in the map. First posted on arXiv 25 Aug 2025.
Builds on
42 of 57 references.
Closest ancestors · 7
Same direction, newest first
- 2025Cyber-Zero: Training Cybersecurity Agents without Runtime29 Jul
- 2025SEC-bench: Automated Benchmarking of LLM Agents on Real-World Software Security Tasks13 Jun
- 2025BountyBench: Dollar Impact of AI Agent Attackers and Defenders on Real-World Cybersecurity Systems21 May
- 2025CVE-Bench: A Benchmark for AI Agents' Ability to Exploit Real-World Web Application Vulnerabilities21 Mar
- 2024RedCode: Risky Code Execution and Generation Benchmark for Code Agents12 Nov
- 2024EnIGMA: Interactive Tools Substantially Assist LM Agents in Finding Security Vulnerabilities24 Sep
- 2024LLM Agents can Autonomously Hack Websites6 Feb
Other references · 35
- 2024The Llama 3 Herd of Modelscited by 14,192
- 2025Qwen3 Technical Reportcited by 8,195
- 2021Evaluating Large Language Models Trained on Codecited by 8,053
- 2024GPT-4o System Cardcited by 4,467
- 2025Gemini 2.5: Pushing the Frontier with Advanced Reasoning, Multimodality, Long Context, and Next Generation Agentic Capabilitiescited by 3,655
- 2023SWE-bench: Can Language Models Resolve Real-World GitHub Issues?cited by 3,193
- 2020Shortcut Learning in Deep Neural Networkscited by 1,928
- 2024SWE-agent: Agent-Computer Interfaces Enable Automated Software Engineeringcited by 1,628
- 2024Qwen2.5-Coder Technical Reportcited by 1,151
- 2023StarCoder: may the source be with you!cited by 790
- 2024The WMDP Benchmark: Measuring and Reducing Malicious Use With Unlearningcited by 488
- 2024StarCoder 2 and The Stack v2: The Next Generationcited by 484
23 more
- 2024Training Software Engineering Agents and Verifiers with SWE-Gymcited by 300
- 2025SWE-smith: Scaling Data for Software Engineering Agentscited by 255
- 2023Magicoder: Empowering Code Generation with OSS-Instructcited by 251
- 2025SWE-RL: Advancing LLM Reasoning via Reinforcement Learning on Open Software Evolutioncited by 246
- 2019NeMo: a toolkit for building AI applications using Neural Modulescited by 230
- 2023InterCode: Standardizing and Benchmarking Interactive Coding with Execution Feedbackcited by 229
- 2023OctoPack: Instruction Tuning Code Large Language Modelscited by 178
- 2024Cybench: A Framework for Evaluating Cybersecurity Capabilities and Risks of Language Modelscited by 156
- 2023Purple Llama CyberSecEval: A Secure Coding Benchmark for Language Modelscited by 130
- 2024Evaluating Frontier Models for Dangerous Capabilitiescited by 121
- 2024An Adversarial Perspective on Machine Unlearning for AI Safetycited by 113
- 2024NYU CTF Bench: A Scalable Open-Source Benchmark Dataset for Evaluating LLMs in Offensive Securitycited by 90
- 2025SWE-Fixer: Training Open-Source LLMs for Effective and Efficient GitHub Issue Resolutioncited by 87
- 2025CyberGym: Evaluating AI Agents' Real-World Cybersecurity Capabilities at Scalecited by 71
- 2024Lingma SWE-GPT: An Open Development-Process-Centric Language Model for Automated Software Improvementcited by 57
- 2024CYBERSECEVAL 3: Advancing the Evaluation of Cybersecurity Risks and Capabilities in Large Language Modelscited by 56
- 2024CyberMetric: A Benchmark Dataset based on Retrieval-Augmented Generation for Evaluating LLMs in Cybersecurity Knowledgecited by 55
- 2024On Evaluating the Durability of Safeguards for Open-Weight LLMscited by 51
- 2023SecQA: A Concise Question-Answering Dataset for Evaluating Large Language Models in Computer Securitycited by 37
- 2023From Text to MITRE Techniques: Exploring the Malicious Use of Large Language Models for Generating Cyber Attack Payloadscited by 31
- 2024Astraios: Parameter-Efficient Instruction Tuning Code Large Language Modelscited by 23
- 2025AutoAdvExBench: Benchmarking autonomous exploitation of adversarial example defensescited by 14
- 2025CAI: An Open, Bug Bounty-Ready Cybersecurity AIcited by 13
Built on by
Citing papers per year: 2025: 2; 2026: 13.
15 papers.
Sort:
By area:
- 2026Adaptive Auto-Harness: Sustained Self-Improvement for Agentic System Deployment on Open-Ended Task Streamscited by 20
- 2025From Code Foundation Models to Agents and Applications: A Comprehensive Survey and Practical Guide to Code Intelligencecited by 13
- 2026Co-RedTeam: Orchestrated Security Discovery and Exploitation with LLM Agentscited by 6
- 2026Toward Secure LLM Agents: Threat Surfaces, Attacks, Defenses, and Evaluationcited by 5
- 2025HackWorld: Evaluating Computer-Use Agents on Exploiting Web Application Vulnerabilitiescited by 2
- 2026KryptoPilot: An Open-World Knowledge-Augmented LLM Agent for Automated Cryptographic Exploitationcited by 1
- 2026To Defend Against Cyber Attacks, We Must Teach AI Agents to Hackcited by 1
- 2026Towards Reliable Local Security Agents: Verifiable Post-Training for Linux Privilege Escalationcited by 1
- 2026AI In Cybersecurity Education -- Scalable Agentic CTF Design Principles and Educational Outcomescited by 1
- 2026Challenges and Future Directions in Agentic Reverse Engineering Systemscited by 0
- 2026CyberEvolver: Structured Self-Evolution for Cybersecurity Agents On the Flycited by 0
- 2026unix-ctf: Procedural Environments for Unix-Competence Reinforcement Learningcited by 0
- 2026Feyospace-v1: How the Cyber Mercury Seven Trained Frontier Cyber Models8 Sep 2026
- 2026Reveree: Diagnosing LLM Reverse-Engineering Agents1 Sep 2026
- 2026Cyber-Capable AI Agents: Vulnerabilities, Evaluation Containment, and Defensive Response28 Jul 2026
- 2026Toward Secure LLM Agents: Threat Surfaces, Attacks, Defenses, and Evaluation9 Jun 2026
- 2026Adaptive Auto-Harness: Sustained Self-Improvement for Agentic System Deployment on Open-Ended Task Streams1 Jun 2026
- 2026unix-ctf: Procedural Environments for Unix-Competence Reinforcement Learning27 May 2026
- 2026CyberEvolver: Structured Self-Evolution for Cybersecurity Agents On the Fly25 May 2026
- 2026Challenges and Future Directions in Agentic Reverse Engineering Systems15 Apr 2026
- 2026AI In Cybersecurity Education -- Scalable Agentic CTF Design Principles and Educational Outcomes23 Mar 2026
- 2026Towards Reliable Local Security Agents: Verifiable Post-Training for Linux Privilege Escalation18 Mar 2026
- 2026Co-RedTeam: Orchestrated Security Discovery and Exploitation with LLM Agents2 Feb 2026
- 2026To Defend Against Cyber Attacks, We Must Teach AI Agents to Hack1 Feb 2026
3 more
- 2026KryptoPilot: An Open-World Knowledge-Augmented LLM Agent for Automated Cryptographic Exploitation14 Jan 2026
- 2025From Code Foundation Models to Agents and Applications: A Comprehensive Survey and Practical Guide to Code Intelligence23 Nov 2025
- 2025HackWorld: Evaluating Computer-Use Agents on Exploiting Web Application Vulnerabilities14 Oct 2025
- 2026Adaptive Auto-Harness: Sustained Self-Improvement for Agentic System Deployment on Open-Ended Task Streamscited by 20
- 2025From Code Foundation Models to Agents and Applications: A Comprehensive Survey and Practical Guide to Code Intelligencecited by 13
- 2026Co-RedTeam: Orchestrated Security Discovery and Exploitation with LLM Agentscited by 6
- 2026Toward Secure LLM Agents: Threat Surfaces, Attacks, Defenses, and Evaluationcited by 5
- 2025HackWorld: Evaluating Computer-Use Agents on Exploiting Web Application Vulnerabilitiescited by 2
- 2026KryptoPilot: An Open-World Knowledge-Augmented LLM Agent for Automated Cryptographic Exploitationcited by 1
- 2026Towards Reliable Local Security Agents: Verifiable Post-Training for Linux Privilege Escalationcited by 1
- 2026AI In Cybersecurity Education -- Scalable Agentic CTF Design Principles and Educational Outcomescited by 1
- 2026Challenges and Future Directions in Agentic Reverse Engineering Systemscited by 0
- 2026CyberEvolver: Structured Self-Evolution for Cybersecurity Agents On the Flycited by 0
- 2026Cyber-Capable AI Agents: Vulnerabilities, Evaluation Containment, and Defensive Responsecited by 0
- 2026Reveree: Diagnosing LLM Reverse-Engineering Agentscited by 0
- 2026Reveree: Diagnosing LLM Reverse-Engineering Agents1 Sep 2026
- 2026Cyber-Capable AI Agents: Vulnerabilities, Evaluation Containment, and Defensive Response28 Jul 2026
- 2026Toward Secure LLM Agents: Threat Surfaces, Attacks, Defenses, and Evaluation9 Jun 2026
- 2026Adaptive Auto-Harness: Sustained Self-Improvement for Agentic System Deployment on Open-Ended Task Streams1 Jun 2026
- 2026CyberEvolver: Structured Self-Evolution for Cybersecurity Agents On the Fly25 May 2026
- 2026Challenges and Future Directions in Agentic Reverse Engineering Systems15 Apr 2026
- 2026AI In Cybersecurity Education -- Scalable Agentic CTF Design Principles and Educational Outcomes23 Mar 2026
- 2026Towards Reliable Local Security Agents: Verifiable Post-Training for Linux Privilege Escalation18 Mar 2026
- 2026Co-RedTeam: Orchestrated Security Discovery and Exploitation with LLM Agents2 Feb 2026
- 2026KryptoPilot: An Open-World Knowledge-Augmented LLM Agent for Automated Cryptographic Exploitation14 Jan 2026
- 2025From Code Foundation Models to Agents and Applications: A Comprehensive Survey and Practical Guide to Code Intelligence23 Nov 2025
- 2025HackWorld: Evaluating Computer-Use Agents on Exploiting Web Application Vulnerabilities14 Oct 2025
Similar, not cited
Close in wording, no citation either way
- 2026Autonomous LLM Agents & CTFs: A Second Look
- 2025MLE-Dojo: Interactive Environments for Empowering LLM Agents in Machine Learning Engineering
- 2026CTFusion: A CTF-based Benchmark for LLM Agent Evaluation
- 2026Capture the Flags: Family-Based Evaluation of Agentic LLMs via Semantics-Preserving Transformations
- 2025Challenges in Testing Large Language Model Based Software: A Faceted Taxonomy
- 2026BabelArena: A Large-Scale Multilingual Benchmark for LLM Agents
- 2025DataSciBench: An LLM Agent Benchmark for Data Science
- 2026Where Agent Frameworks Fall Short: Examining Functional Challenges and Usability Concerns