mlsanityData to 7 Oct 2026

Training Language Model Agents to Find Vulnerabilities with CTF-Dojo

arXiv 2508.18370 ↗First posted 25 Aug 2025cs.SE cs.CL cs.CRcited by 15

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

  1. 2025Cyber-Zero: Training Cybersecurity Agents without Runtime29 Jul
  2. 2025SEC-bench: Automated Benchmarking of LLM Agents on Real-World Software Security Tasks13 Jun
  3. 2025BountyBench: Dollar Impact of AI Agent Attackers and Defenders on Real-World Cybersecurity Systems21 May
  4. 2025CVE-Bench: A Benchmark for AI Agents' Ability to Exploit Real-World Web Application Vulnerabilities21 Mar
  5. 2024RedCode: Risky Code Execution and Generation Benchmark for Code Agents12 Nov
  6. 2024EnIGMA: Interactive Tools Substantially Assist LM Agents in Finding Security Vulnerabilities24 Sep
  7. 2024LLM Agents can Autonomously Hack Websites6 Feb

Other references · 35

  1. 2024The Llama 3 Herd of Modelscited by 14,192
  2. 2025Qwen3 Technical Reportcited by 8,195
  3. 2021Evaluating Large Language Models Trained on Codecited by 8,053
  4. 2024GPT-4o System Cardcited by 4,467
  5. 2025Gemini 2.5: Pushing the Frontier with Advanced Reasoning, Multimodality, Long Context, and Next Generation Agentic Capabilitiescited by 3,655
  6. 2023SWE-bench: Can Language Models Resolve Real-World GitHub Issues?cited by 3,193
  7. 2020Shortcut Learning in Deep Neural Networkscited by 1,928
  8. 2024SWE-agent: Agent-Computer Interfaces Enable Automated Software Engineeringcited by 1,628
  9. 2024Qwen2.5-Coder Technical Reportcited by 1,151
  10. 2023StarCoder: may the source be with you!cited by 790
  11. 2024The WMDP Benchmark: Measuring and Reducing Malicious Use With Unlearningcited by 488
  12. 2024StarCoder 2 and The Stack v2: The Next Generationcited by 484
23 more
  1. 2024Training Software Engineering Agents and Verifiers with SWE-Gymcited by 300
  2. 2025SWE-smith: Scaling Data for Software Engineering Agentscited by 255
  3. 2023Magicoder: Empowering Code Generation with OSS-Instructcited by 251
  4. 2025SWE-RL: Advancing LLM Reasoning via Reinforcement Learning on Open Software Evolutioncited by 246
  5. 2019NeMo: a toolkit for building AI applications using Neural Modulescited by 230
  6. 2023InterCode: Standardizing and Benchmarking Interactive Coding with Execution Feedbackcited by 229
  7. 2023OctoPack: Instruction Tuning Code Large Language Modelscited by 178
  8. 2024Cybench: A Framework for Evaluating Cybersecurity Capabilities and Risks of Language Modelscited by 156
  9. 2023Purple Llama CyberSecEval: A Secure Coding Benchmark for Language Modelscited by 130
  10. 2024Evaluating Frontier Models for Dangerous Capabilitiescited by 121
  11. 2024An Adversarial Perspective on Machine Unlearning for AI Safetycited by 113
  12. 2024NYU CTF Bench: A Scalable Open-Source Benchmark Dataset for Evaluating LLMs in Offensive Securitycited by 90
  13. 2025SWE-Fixer: Training Open-Source LLMs for Effective and Efficient GitHub Issue Resolutioncited by 87
  14. 2025CyberGym: Evaluating AI Agents' Real-World Cybersecurity Capabilities at Scalecited by 71
  15. 2024Lingma SWE-GPT: An Open Development-Process-Centric Language Model for Automated Software Improvementcited by 57
  16. 2024CYBERSECEVAL 3: Advancing the Evaluation of Cybersecurity Risks and Capabilities in Large Language Modelscited by 56
  17. 2024CyberMetric: A Benchmark Dataset based on Retrieval-Augmented Generation for Evaluating LLMs in Cybersecurity Knowledgecited by 55
  18. 2024On Evaluating the Durability of Safeguards for Open-Weight LLMscited by 51
  19. 2023SecQA: A Concise Question-Answering Dataset for Evaluating Large Language Models in Computer Securitycited by 37
  20. 2023From Text to MITRE Techniques: Exploring the Malicious Use of Large Language Models for Generating Cyber Attack Payloadscited by 31
  21. 2024Astraios: Parameter-Efficient Instruction Tuning Code Large Language Modelscited by 23
  22. 2025AutoAdvExBench: Benchmarking autonomous exploitation of adversarial example defensescited by 14
  23. 2025CAI: An Open, Bug Bounty-Ready Cybersecurity AIcited by 13

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2025: 2; 2026: 13.

15 papers.

Sort:

By area:

Similar, not cited

Close in wording, no citation either way

  1. 2026Autonomous LLM Agents & CTFs: A Second Look
  2. 2025MLE-Dojo: Interactive Environments for Empowering LLM Agents in Machine Learning Engineering
  3. 2026CTFusion: A CTF-based Benchmark for LLM Agent Evaluation
  4. 2026Capture the Flags: Family-Based Evaluation of Agentic LLMs via Semantics-Preserving Transformations
  5. 2025Challenges in Testing Large Language Model Based Software: A Faceted Taxonomy
  6. 2026BabelArena: A Large-Scale Multilingual Benchmark for LLM Agents
  7. 2025DataSciBench: An LLM Agent Benchmark for Data Science
  8. 2026Where Agent Frameworks Fall Short: Examining Functional Challenges and Usability Concerns