Curator
Maintains task-local strategy records, verifier outcomes, failed inputs, file locations, and milestones. Before each attempt, it activates the evidence most relevant to the current vulnerability.
A dual-loop framework that learns reusable vulnerability-reproduction strategies while preserving task-local evidence across attempts.
National University of Singapore · Nanyang Technological University · The Hong Kong University of Science and Technology (Guangzhou)
A repository-level vulnerability task arrives with source, description, and benchmark interface.
Repository-scale vulnerability reproduction requires an agent to inspect code, infer an input grammar, construct a proof-of-concept, and verify that the patched build no longer crashes.
Mastermind argues that these agents often fail because they choose the wrong investigation strategy, even when they can execute commands, edit files, and submit PoCs. The framework separates planning from acting: a trainable Planner learns reusable vulnerability-reproduction strategies through SFT and milestone-based GRPO, while a Curator preserves task-local experience across sequential attempts. The Planner is trained independently of the frozen Executor, so strategy learning can improve multiple action backbones without changing their command-generation capability.
Modern LLM agents can navigate repositories and run long software-engineering workflows, but vulnerability reproduction is judged by executable evidence. The agent must decide where to inspect, what input structure to infer, when to exploit, and how to revise after verifier feedback.
CyberGym exposes the gap. With a fixed GPT-5.5 executor, one-shot Level-1 attempts solve 23.5% of held-out tasks, independent Best-of-8 reaches 63.0%, and sequential task-local strategy revision reaches 77.0%.
Locate relevant source, parsers, fixtures, and vulnerable paths.
Turn partial evidence into a concrete PoC construction strategy.
Use dual-build feedback to distinguish target reproduction from wrong crashes.
Carry task-local findings forward instead of resampling from scratch.
Transferable strategy instincts belong in Planner weights. Volatile facts about one repository belong in Curator experience.
Maintains task-local strategy records, verifier outcomes, failed inputs, file locations, and milestones. Before each attempt, it activates the evidence most relevant to the current vulnerability.
Emits a compact strategy that tells the executor where to inspect, what vulnerability mechanism to test, and how to validate the PoC. It learns through SFT and milestone-based GRPO.
Instantiates the strategy as repository actions, shell commands, file edits, and PoC submissions. The executor stays frozen, isolating gains from better planning.
Provides execution-grounded feedback using CyberGym milestones, then routes the result back to Curator and Planner so both loops improve the next attempt.
The same planner trained on GPT-5.4 mini trajectories improves GPT-5.4 mini, GPT-5.5, and GLM 5.1 without executor-specific retraining.
Strict milestone-7 pass rates on the 200-task CyberGym held-out split. Mastermind solves 169/200 tasks with 560 GPT-5.5 executor rollouts.
| Frozen executor | Base | Mastermind |
|---|---|---|
| GPT-5.4 mini | 45.0% | 60.0% |
| GPT-5.5 | 72.5% | 84.5% |
| GLM 5.1 | 58.5% | 71.0% |
Mastermind reframes repository-scale vulnerability reproduction as strategy selection and iterative strategy refinement.
Identifies high-level strategy selection as a primary failure mode, supported by strategy-sensitivity, Best-of-N, and iterative-experience analyses.
Trains over compact natural-language strategies instead of full repository trajectories, making reinforcement learning practical for long-horizon SE agents.
Improves multiple frozen executors and outperforms independent sampling, static-analysis guidance, richer context, and task-local iteration baselines.