Trace which memory or tool input actually drove an LLM agent's action.

January 21, 20266 min

Overview

Decision SnapshotReady For Pilot

The approach is conceptually simple and effective on curated traces, but it needs broader datasets and automation for large-scale deployment.

Citations0

Evidence Strength0.80

Confidence0.90

Risk Signals9

Trust Signals

Findings with numeric evidence: 2/3

Findings with evidence refs: 3/3

Results with explicit delta: 2/2

Reproducibility

Status: Partial assets available

Open source: Partial

At A Glance

Cost impact: 35%

Production readiness: 60%

Novelty: 55%

Authors

Chen Qian, Peng Wang, Dongrui Liu, Junyao Yang, Dadi Guo, Ling Tang, Jilin Mei, Qihan Ren, Shuai Shao, Yong Liu, Jie Fu, Jing Shao, Xia Hu

Links

Abstract / PDF / Code

Why It Matters For Business

You can audit autonomous agents to see which past memory or tool output caused a decision—useful for compliance, debugging, and fixing business rule violations without needing explicit failures.

Who Should Care

Summary TLDR

This paper presents a two-stage attribution framework that explains why an LLM-based agent produced a specific action. First it replays the agent trajectory component-by-component and scores the marginal likelihood gain to find high-impact components (temporal likelihood dynamics). Then it ablates sentences inside those components (probability drop & hold) to surface the exact textual evidence. The method is evaluated on eight curated agent trajectories (memory- and tool-driven scenarios) using Llama-3.1-70B-Instruct; a simple leave-one-out and linear baselines are compared. Code is released.

Problem Statement

Existing attribution work focuses on locating explicit failures. But many undesirable agent actions occur without an explicit error signal (for example a reasonable-looking refund or a privacy leak caused by a retrieved email). We need methods to explain which past memory entries, tool returns, or sentences actually drove a chosen action.

Main Contribution

A hierarchical agentic attribution framework: component-level temporal replay + sentence-level perturbation.

Component-level method: score marginal likelihood gains when incrementally revealing trajectory components.

Key Findings

Prob. Drop&Hold hits the human-labelled top sentence 93.75% of the time (Hit@1).

NumbersHit@1 = 0.9375 (Table 1)

Practical UseUse the combined drop&hold ablation to reliably surface a single most-causal sentence in curated agent traces.

Evidence RefTable 1

Multiple sentence-level attribution methods work well inside the framework; leave-one-out and ContextCite reach 81.25% Hit@1.

NumbersLOO, ContextCite Hit@1 = 0.8125 (Table 1)

Practical UseIf ablation is expensive, surrogate methods like LOO or ContextCite are viable fallbacks.

Evidence RefTable 1

Results

MetricValueBaselineDeltaSplit / DatasetEvidenceEvidence Ref
Hit@1 (Prob. Drop&Hold)0.9375LOO 0.8125+0.125average across 8 curated casesTable 1 reports Hit@1 = 0.9375 for Prob. Drop&Hold vs 0.8125 for LOOTable 1
Hit@3 (Prob. Drop&Hold)1.0LOO 0.9375+0.0625average across 8 curated casesTable 1 reports Hit@3 = 1.0 for Prob. Drop&HoldTable 1

What To Try In 7 Days

Run the component-level replay on a few real agent traces to surface high-impact steps.

Apply the probability drop&hold ablation to top components to surface the exact sentence that drove an action.

Use findings to add simple guards: ignore untrusted tool text, downweight single-case memories, or require explicit evidence before high-risk actions.

Agent Features

Memory
retrieval memorylong-term memory updates
Planning
incremental replay of trajectory (temporal likelihood dynamics)
Tool Use
treat tool outputs as observationsexpose tool returns (email, web search, file reader) for attribution
Frameworks
smolagents
Is Agentic

Yes

Architectures
LLM-based agent (single-agent)

Reproducibility

Code AvailableYes
Data AvailableNo
Open Source StatusPartial
LicenseUnknown

Risks & Boundaries

Limitations

Evaluation uses eight curated cases and a single model (Llama-3.1-70B-Instruct), limiting generality.

Sentence-level ablation requires model likelihood access and can be costly on long contexts.

When Not To Use

When you cannot compute or compare model likelihoods for ablated inputs (closed APIs without log-probs).

As a fully automated monitor at large scale without further work to automate interpretation.

Failure Modes

Multiple components share influence and attribution may split credit ambiguously.

Agent self-contradiction or latent internal state can lead to misleading likelihood signals.

Core Entities

Models

Llama-3.1-70B-Instruct

Metrics

Hit@k (Hit@1/3/5)log-likelihood gain

Datasets

custom 8-case agent trajectories

Benchmarks

GAIA (one complex retrieval case used)

Context Entities

Models

none other explicitly evaluated

Metrics

probability dropprobability hold

Datasets

GAIA (cited)human annotation intersection (5 annotators) for ground truth