AgentForesight: Online Auditing for Early Failure Prediction in Multi-Agent Systems
AgentForesight

Online Auditing for Early Failure Prediction in Multi-Agent Systems

NeurIPS 2026
Boxuan Zhang1,* Jianing Zhu2,* Zeru Shi1 Dongfang Liu3 Ruixiang Tang1,†
1Rutgers University 2The University of Texas at Austin 3Purdue University
*: Equal Contribution.  †: Corresponding Author.
Paper Code Dataset BibTeX

AgentForesight at a glance

01

Abstract

Problem

LLM-based multi-agent systems are increasingly deployed on long-horizon tasks, but a single decisive error is often accepted by downstream agents and cascades into trajectory-level failure. Existing work frames this as post-hoc failure attribution, diagnosing the responsible agent and step after the trajectory has ended. However, this paradigm forfeits any opportunity to intervene while trajectory is still unfolding.

Framework

In this work, we introduce AgentForesight, a framework that reframes this problem as online auditing: at each step of an unfolding trajectory, an auditor observes only the current prefix and must either continue the run or alarm at the earliest decisive error, without access to future steps.

Dataset

To this end, we curate AFTraj-2K, a corpus of agentic trajectories across Coding, Math, and Agentic domains, in which safe trajectories are retained under a strict curation pipeline and unsafe trajectories are annotated at the step of their decisive error via consensus among multiple LLM judges.

Model

Built on that, we develop AgentForesight-7B, a compact online auditor trained with a coarse-to-fine reinforcement learning recipe that first equips it with a risk-anticipation prior at the failure boundary on adjacent safe/unsafe prefix pairs, then sharpens this prior into precise step-level localization under a three-axis reward jointly targeting the what, where, and who of an audit verdict.

Results

Across AFTraj-2K and an external Who&When benchmark, AgentForesight-7B outperforms leading proprietary models, including GPT-4.1 and DeepSeek-V4-Pro, achieving up to +19.9% performance gain and 3× lower step localization error, opening the loop from post-hoc failure detection to deployment-time intervention.

02

Why Online Auditing?

Comparison of post-hoc failure attribution and online auditing
Post-hoc failure attribution

(a) Post-hoc attribution only diagnoses after the trajectory has already failed.

Online auditing (AgentForesight)

(b) AgentForesight flags the decisive error step-by-step on the unfolding prefix, opening a deployment-time intervention window before the failure is locked in.

03

Method

AgentForesight pipeline: AFTraj-2K construction and coarse-to-fine training
AFTraj-2K construction

(a) AFTraj-2K pairs strictly-filtered safe runs with failure runs annotated at their decisive step via multi-judge voting.

Coarse-to-fine training

(b) A coarse-to-fine RL recipe first aligns the auditor at the failure boundary, then sharpens it under a three-axis reward over what, where, and who.

04

Main Results

AgentForesight-7B reaches 66.44 Exact-F1 (+19.88 over DeepSeek-V4-Pro) and tightens overall ASS to 0.59 (3× lower), winning every domain.

Method
Math
Coding
Agentic
Overall
Exact-F1↑
ASS↓
Exact-F1↑
ASS↓
Exact-F1↑
ASS↓
Exact-F1↑
ASS↓
{{ row.label }}
{{ row.name }} ours
{{ c.text }}
Exact-F1↑ harmonic mean of step-level recall and precision on decisive-step predictions. ASS↓ Absolute Step Shift, the mean |k̂ − k*| over detected unsafe trajectories. Bold = best, underline = second-best. † AgentDebug-7B detects zero unsafe trajectories, so its Math ASS is undefined (—).
05

Ablation & Analysis

Stage-wise ablation
Exact-F1 (%) ↑
Base Stage 1 Stage 2 Stage 1+2 (ours)
0
20
40
60
80
100
{{ b.delta }}
{{ b.val }}
{{ g.d }}

Stage 1 and Stage 2 individually lift Exact-F1; combined they reach 66.4, exceeding either alone by ≥16 points.

Deployment trade-off
hover a point for values
Open-source Proprietary ★AgentForesight-7B
Step-Acc (%) ↑
20
40
60
0
20 40 60 80 100
deployable
↖ better
{{ pt.name }}
{{ tip.name }}
FAR {{ tip.far }} · Step-Acc {{ tip.acc }}
False Alarm Rate (%) ↓

Among ten auditors, only AgentForesight-7B sits inside the deployable region (FAR ≤ 20%, Step-Acc ≥ 50%).

06

Case Study

AgentForesight-7B alone localizes the decisive error to the responsible search_agent. Gemini-3-Flash raises a premature alarm at Step 2; DeepSeek-V4-Pro returns SAFE despite the wrong final answer.

Task Query: Which former mill town is near Rivington Hall Barn?
Gold Answer: Bolton
Multi-agent system Planner Manager Search Agent
Auditor {{ hint }}
Trajectory, revealed one prefix at a time future steps stay hidden
{{ r.n }}.
{{ r.agent }} {{ r.tag }}
{{ r.label }} {{ r.text }}
{{ r.pill }}
{{ cp.initials }}
Online auditor
{{ cp.name }}
{{ cp.status }}
Prediction {{ cp.pred }} Decisive error step {{ cp.step }} Decisive error agent {{ cp.agent }}
{{ cp.whyA }}{{ cp.whyB }}
Waiting for the verdict…
{{ cp.outTitle }}
{{ cp.outText }}
07

BibTeX

zhang2026agentforesight
@inproceedings{zhang2026agentforesight,
  title     = {AgentForesight: Online Auditing for Early Failure Prediction in Multi-Agent Systems},
  author    = {Boxuan Zhang and Jianing Zhu and Zeru Shi and Dongfang Liu and Ruixiang Tang},
  booktitle = {The Fortieth Annual Conference on Neural Information Processing Systems},
  year      = {2026},
  url       = {https://arxiv.org/abs/2605.08715}
}
{{ lbAlt }}
Close ✕