SAFARI: Scaling long horizon agentic fault attribution via active investigation
A framework that replaces linear context loading with a tool-augmented diagnostic loop.
As autonomous agents tackle increasingly complex multi-step, multi-agent tasks, their execution trajectories have scaled beyond the constraints of even the largest context windows. Current methods for effectively diagnosing agent failures rely on loading the full agentic trajectory into a LLM's context window. This approach suffers from attention dilution and is not scalable when agentic traces inevitably grow outside of even the largest context limit one single LLM can have. To address this, we introduce SAFARI (Scaling long-horizon Agentic Fault AttRibution via active Investigation), a framework that replaces linear context loading with a tool-augmented diagnostic loop. By equipping LLMs with a specialized toolbox to read and search trajectory segments alongside a persistent Short-Term Memory (STM) for cross-turn reasoning, SAFARI effectively decouples diagnostic accuracy from architectural context limits. Our experiments demonstrate that SAFARI outperforms state-of-the-art results by 20% on the Who&When dataset within a 1M token budget, and by 19% on TRAIL GAIA subset on a 25K token budget. Most significantly, SAFARI maintains a 0.58 precision even when the target fault resides 5x beyond the model’s native context window, a scenario where traditional evaluators fail entirely.
Latest publications
Critique-guided distillation for robust reasoning via refinement
A training framework that decouples critique consumption from critique generation.
ICMLDynamic guardian models: realtime content moderation with user-defined policies
Specialized classifiers that evaluate text based on predefined trustworthiness objectives.
ICMLAsk, don’t judge: Binary questions for interpretable LLM evaluation and self-improvement
A framework that decomposes evaluation criteria into atomic binary questions and aggregates the resulting verdicts.
ICML