Autonomous Security Investigation Agent at Scale

wiz · Tech · 2026 · source Agentic: yes · promotional bias: med · summary-only (no fetched source) · extracted with prompt 0.5.4 (schema v2)

Generated from extraction.json — do not edit by hand. Every claim cites verbatim source evidence; stated means the quote directly asserts the claim, implied means it supports it indirectly.

Problem

Domain: Cloud security threat investigation

Wiz needed AutoAgent to investigate cloud security events at massive scale, deciding whether events were malicious or false positives despite unpredictable investigation paths and huge context payloads.

Constraints:

Architecture

AutoAgent is described as a production-scale autonomous security investigation system. A main investigator agent delegates to specialized sub-agents for domain-specific analysis, uses reflection loops to stop and plan after iterations, stores compressed findings in external agent state, loads threat-specific playbooks from a knowledge base, and improves through weekly evaluation against ground truth, profile-based performance tracking, simulation, LLM-as-judge gap analysis, and team-owned improvement workflows.

Diagram confidence: reconstructed (reconstructed from prose).

flowchart LR
  security-event[Security event] --> main-investigator-agent[Main investigator agent]
  main-investigator-agent --> specialized-sub-agents[Specialized sub-agents]
  main-investigator-agent --> playbook-knowledge-base[Playbook knowledge base]
  main-investigator-agent --> tool-ecosystem[Tool ecosystem]
  specialized-sub-agents --> tool-ecosystem
  tool-ecosystem --> agent-state[Agent state]
  agent-state --> main-investigator-agent
  main-investigator-agent --> evaluation-framework[Evaluation framework]
  evaluation-framework --> profile-tracking[Profile tracking]
  evaluation-framework --> simulation-engine[Simulation engine]
  evaluation-framework --> llm-judge-gap-analysis[LLM judge gap analysis]
  llm-judge-gap-analysis --> improvement-teams[Improvement teams]
Drag to pan · scroll or use +/− to zoom · reconstructed from the source's described component topology

Components

Component Role Interacts with
Security event Initial suspicious cloud event or threat signal requiring investigation. Main investigator agent
Main investigator agent Coordinates the investigation, decides when to delegate, reflects on progress, and reaches final malicious/false-positive decisions. Security event, Specialized sub-agents, Playbook knowledge base, Tool ecosystem, Agent state, Evaluation framework
Specialized sub-agents Analyze domain-specific cloud security areas in isolated contexts and return essential findings. Main investigator agent, Tool ecosystem
Playbook knowledge base Provides threat-specific instructions, questions, and tools. Main investigator agent
Tool ecosystem Connects the agent to organizational cloud/security systems and investigation data. Main investigator agent, Specialized sub-agents, Agent state
Agent state External memory where key findings and investigation steps are stored outside the conversation context. Main investigator agent, Tool ecosystem
Evaluation framework Runs sampled investigations against ground truth and identifies improvement opportunities. Main investigator agent, Profile tracking, Simulation engine, LLM judge gap analysis
Profile tracking Segments investigations into populations and tracks completeness/correctness by profile. Evaluation framework
Simulation engine Replays steps from existing investigations with modified parameters to test improvements cheaply. Evaluation framework
LLM judge gap analysis Labels profile-level failure modes and points teams to specific gaps. Evaluation framework, Improvement teams
Improvement teams Distributed teams that define populations, test changes, and publish validated improvements. LLM judge gap analysis

Data flow

A cloud security event starts an investigation. AutoAgent loads a relevant playbook, gathers tool results, reflects at iteration boundaries, and stores key findings in external state instead of keeping raw context. When specialized expertise is needed, the main agent delegates to sub-agents that return essential findings. Completed investigations feed evaluation, profiling, simulation, gap analysis, and distributed team improvement loops.

3 supporting quotes
“When an investigation begins, the appropriate playbook is loaded based on the detected threat category. This provides the agent with a starting direction and prevents it from becoming distracted or wandering aimlessly.”summary, stated
“At the end of each iteration, the agent stops, reflects on its actions so far, summarizes if necessary, and then plans its next steps.”summary, stated
“These sub-agents conduct their analysis within their own isolated context and return only the essential findings to the main agent.”summary, stated

Why this architecture

Wiz treats the agent like an operating system because cloud security investigation has unpredictable paths, huge context, many tools, specialized domains, and continuous improvement needs. Sub-agents isolate domain context, reflection prevents generic wandering, external state compresses context, playbooks guide expertise, and evaluation/profiling/simulation turn production investigations into a systematic improvement loop.

3 supporting quotes
“The team conceptualized the agent architecture not as a trivial system but as an operating system.”summary, stated
“First is context explosion - cloud environments are massive, and each tool call can return gigabytes of information.”summary, stated
“Operating a production agent at scale requires robust evaluation and continuous improvement processes. Wiz implemented a weekly feedback loop for systematic improvement.”summary, stated

Models used: AutoAgent · specialized sub-agents · LLM-as-judge

Scale

Alternatives considered (roads not taken)

Single monolithic investigation context

Rejected because: Different security domains need specialized expertise and raw context can explode.

“Second, different areas of cloud security require different domain expertise, necessitating specialized knowledge application.”summary, stated
“This approach prevents the main agent's context from being overwhelmed by specialized investigative details while still leveraging domain-specific expertise.”summary, stated

Let the agent accumulate all conversation context

Rejected because: The active context window would become unbounded and unmanageable.

“This approach ensures that while the agent may process enormous amounts of tokens across an entire investigation, the active context window remains bounded within manageable limits.”summary, stated
“Once this information is extracted and stored, the original conversational context that generated these findings can be thrown away, freeing up the context window.”summary, stated

Unguided generic investigation

Rejected because: Generic agents struggle to know where to begin and can wander when faced with massive information.

“Generic agents struggle to know where to begin investigations and which direction to pursue when faced with massive amounts of information.”summary, stated
“The playbook guides the initial investigative steps while still allowing the agent flexibility to follow the evidence wherever it leads.”summary, stated

Central data science team owns all improvements

Rejected because: The team became a bottleneck as many organizational teams needed to improve different agent behaviors.

“As the agent matured, the data science team found themselves becoming a bottleneck. Every team across the organization would approach them with problems: the agent isn't performing well in this area, fix this issue, improve that behavior.”summary, stated
“They developed an "agent democratization" framework that empowers different teams to work on the agent independently.”summary, stated

Patterns

multi-agent-orchestration (security)

The main investigator delegates specialized cloud-security areas to sub-agents with isolated context.

“The main agent functions both as an investigator and as a manager of specialized sub-agents.”summary, stated

context-compaction (memory)

The agent extracts key findings into external state and discards raw conversational context to keep the active window bounded.

“This enables a radical context compression strategy.”summary, stated
“the original conversational context that generated these findings can be thrown away”summary, stated

agent-memory (memory)

Agent state stores key findings and investigation steps outside the conversation.

“agent state - a form of memory external to the conversation that the LLM can write to but that doesn't consume context window space.”summary, stated

prompt-chaining (reasoning)

Reflection checkpoints force the agent to summarize, evaluate scope, and plan next steps before continuing or terminating.

“At the end of each iteration, the agent stops, reflects on its actions so far, summarizes if necessary, and then plans its next steps.”summary, stated

llm-as-judge (evaluation)

Separate LLM judge analysis labels failure modes and aggregates profile-level gap summaries.

“Wiz employs an LLM-as-judge approach for detailed gap analysis.”summary, stated

eval-driven-development (operations)

Weekly evaluation, ground truth comparison, profiles, simulation, and validated updates drive continuous improvement.

“Wiz implemented a weekly feedback loop for systematic improvement.”summary, stated
“comparing results against a golden set of ground truth data”summary, stated

human-in-the-loop-gate (governance)

Human analysts and specialized teams investigate gaps, determine root causes, and publish validated changes.

“Human analysts or the appropriate specialized team can then investigate these gaps, determine root causes, and implement fixes”summary, stated

Tradeoffs

Use specialized sub-agents for deep domains

Chose hierarchical delegation with isolated context over main agent directly handling every specialized security detail.

Rationale: The main agent avoids being overwhelmed while still getting domain-specific findings.

“These sub-agents conduct their analysis within their own isolated context and return only the essential findings to the main agent.”summary, stated
“prevents the main agent's context from being overwhelmed by specialized investigative details”summary, stated

Compress context aggressively

Chose external state with early offloading over retaining the full investigation transcript in context.

Rationale: Massive cloud data makes bounded active context necessary for scale.

“Once this information is extracted and stored, the original conversational context that generated these findings can be thrown away”summary, stated

Use simulation to validate improvement ideas

Chose replay selected investigation steps with modified parameters over rerunning entire investigations for every prompt or logic change.

Rationale: Simulation reduces cost and speeds iteration where the relevant step can be replayed against known ground truth.

“Rather than re-running entire investigations from scratch to test improvements, Wiz built a simulation engine”summary, stated
“avoiding the time and cost of re-running complete investigations.”summary, stated

Gotchas (what bit them)

Security investigation paths are unpredictable.

Root cause: Investigators know the starting event but not what later steps will reveal or whether lateral movement exists.

Resolution: AutoAgent uses reflection, playbooks, and flexible tool/sub-agent orchestration rather than a fixed workflow.

“investigators know where they start but not where they'll end up.”summary, stated
“Actions taken at step five might influence what happens at step ten”summary, stated

Context explosion is the central bottleneck.

Root cause: Cloud environments are massive and tool calls can return gigabytes of data.

Resolution: The system offloads key findings early into external state and discards raw conversational context.

“First is context explosion - cloud environments are massive, and each tool call can return gigabytes of information.”summary, stated
“intelligent strategies to "offload early" - extracting value from context, storing it externally”summary, stated

Agents can make generic decisions or repeat themselves.

Root cause: Without deliberate reflection, unconstrained agent loops take predictable actions and wander.

Resolution: Reflection checkpoints force the agent to stop, summarize, evaluate, and plan.

“agents often take predictable, generic actions and can easily repeat themselves.”summary, stated
“The reflection checkpoints force the model to constantly evaluate whether it's on the right track”summary, stated

Centralized agent improvement became an organizational bottleneck.

Root cause: Many teams needed different improvements to the same production agent.

Resolution: Wiz built an agent democratization framework so teams can define populations, test changes in simulation, and publish validated improvements.

“Every team across the organization would approach them with problems”summary, stated
“shift from being centralized gatekeepers to enabling distributed ownership of agent improvement.”summary, stated

Evaluation

Wiz evaluates AutoAgent through weekly sampled investigations, ground-truth comparisons, completeness/correctness profiles, simulation, LLM-as-judge gap analysis, and production monitoring of false positives and profile performance.

Metrics: over 3,000 investigations per day · profile features such as 0-10, 10-20, and 20+ tool-call ranges · completeness vs correctness graphs by profile · simulation across large samples of investigations · false-positive calibration improvements validated across profiles

3 supporting quotes
“conducting over 3,000 threat investigations daily.”summary, stated
“They created performance graphs plotting completeness versus correctness for each profile”summary, stated
“The results showed significant improvement across profiles, validating that the change was worth deploying to production.”summary, stated

Outcomes

Summary-only: the stored summary reports over 3,000 investigations per day but does not provide raw transcript evidence or exact production false-positive rates.

3 supporting quotes
“a mature, production-scale implementation of autonomous agents with sophisticated LLMOps practices around evaluation, continuous improvement, and organizational enablement.”summary, stated
“The false positive problem they discovered through profiling”summary, stated
“maintaining quality through the validation and testing frameworks.”summary, stated