Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Trust No Tool: Evaluating and Defending LLM Agents under Untrusted Tool Feedback

About

Tool-using LLM agents increasingly rely on external tools to make consequential decisions, yet most existing agent-security benchmarks and defenses implicitly assume that tool feedback is trustworthy once a tool has been selected. We study a different failure mode, cognitive poisoning, in which a malicious tool behaves plausibly during exploration, accumulates trust through benign-looking feedback, and becomes harmful only when hidden state conditions align with the final executable action. To study this setting, we construct TRUST-Bench, a task-conditioned benchmark of 1,970 hidden-trigger tool-compromise episodes with matched safe controls, introduce an asymmetric penalty metric, GuardedJoint, to better reflect real deployment risk, and present VISTA-Guard, a backbone-agnostic framework for final-action risk scoring. The core idea is to abstract multi-step tool interaction into structured environment variables that encode trust-formation dynamics and then score the risk of the final executable action from this trajectory-conditioned representation. Experiments show that prompt-centric heuristics, scalarized features, and zero-shot judges fail in this regime, whereas trajectory-aware final-action scoring yields strong in-domain discrimination and remains effective under balanced out-of-distribution transfer. Under GuardedJoint, VISTA-Guard reaches $84.2$ in-domain and $56.9$ on balanced out-of-distribution evaluation, while methods that optimize only one side of the safety--utility tradeoff collapse to zero. These findings support a broader view of agent security in black-box tool ecosystems: the decisive defense target is not local prompt text or tool descriptors alone, but the way trust is formed across the interaction trajectory and committed through the final action.

Lecheng Yan, Ruizhe Li, Xicheng Han, Wenxi Li, Binwu Wang, Longyue Wang, Chenyang Lyu, Guanhua Chen• 2026

Related benchmarks

TaskDatasetResultRank
Malicious Tool Call DetectionTRUST-BENCH reconstructed comparison set curated 1,970-episode (test)
AMR9.6
20
Cognitive Poisoning DetectionCognitive Poisoning Detection 5-fold evaluation (test)
AMR4.2
7
Secure tool generalizationToolEmu and SafeToolBench Balanced OOD transfer
GUARDEDJOINT Score56.9
7
Static Replay Attack Detection100 Targeted Seed Episodes profile (orig)
AMR23.91
5
Static Replay Attack Detection100 Targeted Seed Episodes (redcode_style profile)
AMR45.65
5
External Attack Transfer DetectionToolEmu and SafeToolBench (external harmful episodes)
Recall @ 5% ID-FPR99.52
4
Malicious Action Detection100-seed stress orig profile (test)
AMR4.35
4
Malicious Action Detectionstress test 100-seed redcode_style profile
AMR17.39
4
Showing 8 of 8 rows

Other info

Follow for update