arXiv:2601.12449cs.CRcs.AI2026-01被引 13

为智能体工具使用设计安全框架,防风险不降性能。

AgenTRIM: Tool Risk Mitigation for Agentic AI

  • 离线重构工具接口,线上动态控制权限最小化
  • 在AgentDojo上降低攻击成功率,任务完成率不变
  • 适合关注智能体安全的开发者和研究者

智能体是结合大模型与外部工具以解决复杂任务的自主系统。工具权限不当会引入间接提示注入和工具滥用等安全风险。我们将其归因为工具驱动的代理能力失衡:既可能保留冗余权限(过度授权),也可能无法调用必要工具(授权不足),从而扩大攻击面并降低性能。为此提出AgenTRIM框架,通过离线重建与验证工具接口,以及运行时每步基于状态感知的自适应过滤,实现最小权限访问。在AgentDojo基准测试中,AgenTRIM显著降低攻击成功率,同时保持高任务表现。额外实验表明其对基于描述的攻击具有鲁棒性,并能有效执行明确的安全策略。结果证明AgenTRIM是一种实用、不损害能力的安全工具使用方案。

原文摘要 · Abstract (English)

AI agents are autonomous systems that combine LLMs with external tools to solve complex tasks. While such tools extend capability, improper tool permissions introduce security risks such as indirect prompt injection and tool misuse. We characterize these failures as unbalanced tool-driven agency. Agents may retain unnecessary permissions (excessive agency) or fail to invoke required tools (insufficient agency), amplifying the attack surface and reducing performance. We introduce AgenTRIM, a framework for detecting and mitigating tool-driven agency risks without altering an agent's internal reasoning. AgenTRIM addresses these risks through complementary offline and online phases. Offline, AgenTRIM reconstructs and verifies the agent's tool interface from code and execution traces. At runtime, it enforces per-step least-privilege tool access through adaptive filtering and status-aware validation of tool calls. Evaluating on the AgentDojo benchmark, AgenTRIM substantially reduces attack success while maintaining high task performance. Additional experiments show robustness to description-based attacks and effective enforcement of explicit safety policies. Together, these results show that AgenTRIM provides a practical, capability-preserving approach to safer tool use in LLM-based agents.

智能体安全工具调用权限控制

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。