Track, Rank, Crack: Epistemic Working Memory Scales Multi-Hop Reasoning in Language Agents

· Source: Machine Learning · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Robotics & Autonomous Systems, Natural Language Processing · Depth: Expert, quick

Summary

SLEUTH, a novel approach for language agents, addresses context dilution that degrades multi-hop reasoning performance as chain length increases. It introduces a structured epistemic working memory, explicitly maintaining Confirmed Facts, Active Hypotheses, and Open Questions to guide agent actions. SLEUTH demonstrates significant performance gains across five multi-hop benchmarks, achieving +5 points on HotpotQA and +11 points on 4-hop chains, outperforming Reflexion without requiring multiple episodes. The research also identifies the "evidence sufficiency problem," where agents fail to commit despite finding answers, and proposes a lightweight commitment trigger. This trigger proves effective only when combined with SLEUTH's structured state, highlighting that organized epistemic state is crucial for effective commitment. Furthermore, enforcing protocol adherence on weaker models recovered up to +19 points on challenging problems, underscoring that reasoning organization, rather than raw model capability, is the primary factor for scaling multi-hop reasoning.

Key takeaway

For NLP Engineers developing language agents for complex multi-hop reasoning, prioritize implementing structured epistemic working memory. Your agents will achieve significant performance gains, as organized state, not just raw model capability, scales reasoning effectively. Consider adding lightweight commitment triggers to prevent over-verification, ensuring these are integrated with a structured state for optimal impact. This approach can recover substantial performance on challenging problems.

Key insights

Structured epistemic working memory, not raw model capability, is key to scaling multi-hop reasoning in language agents.

Principles

Method

SLEUTH maintains Confirmed Facts (grounded to sources), Active Hypotheses (ranked by evidence), and Open Questions (driving actions) in a structured epistemic working memory.

In practice

Topics

Best for: Research Scientist, AI Engineer, Machine Learning Engineer, AI Scientist, NLP Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Machine Learning.