English

StepShield: When, Not Whether to Intervene on Rogue Agents

Machine Learning 2026-01-30 v1 Artificial Intelligence Cryptography and Security Software Engineering

Abstract

Existing agent safety benchmarks report binary accuracy, conflating early intervention with post-mortem analysis. A detector that flags a violation at step 8 enables intervention; one that reports it at step 48 provides only forensic value. This distinction is critical, yet current benchmarks cannot measure it. We introduce StepShield, the first benchmark to evaluate when violations are detected, not just whether. StepShield contains 9,213 code agent trajectories, including 1,278 meticulously annotated training pairs and a 7,935-trajectory test set with a realistic 8.1% rogue rate. Rogue behaviors are grounded in real-world security incidents across six categories. We propose three novel temporal metrics: Early Intervention Rate (EIR), Intervention Gap, and Tokens Saved. Surprisingly, our evaluation reveals that an LLM-based judge achieves 59% EIR while a static analyzer achieves only 26%, a 2.3x performance gap that is entirely invisible to standard accuracy metrics. We further show that early detection has direct economic benefits: our cascaded HybridGuard detector reduces monitoring costs by 75% and projects to $108M in cumulative savings over five years at enterprise scale. By shifting the focus of evaluation from whether to when, StepShield provides a new foundation for building safer and more economically viable AI agents. The code and data are released under an Apache 2.0 license.

Keywords

Cite

@article{arxiv.2601.22136,
  title  = {StepShield: When, Not Whether to Intervene on Rogue Agents},
  author = {Gloria Felicia and Michael Eniolade and Jinfeng He and Zitha Sasindran and Hemant Kumar and Milan Hussain Angati and Sandeep Bandarupalli},
  journal= {arXiv preprint arXiv:2601.22136},
  year   = {2026}
}

Comments

16 pages, 2 figures, 14 tables

R2 v1 2026-07-01T09:26:25.922Z