Agent | | arXiv:2606.20023

When Lower Privileges Suffice: Investigating Over-Privileged Tool Selection in LLM Agents

权限过高是否多余:探究LLM代理中的过度权限工具选择

🧭 代理工具选择与权限控制

Institute of Information Engineering, Chinese Academy of Sciences · Beijing Academy of Artificial Intelligence · The Chinese University of Hong Kong · Institute for Artificial Intelligence, Peking University · School of Cyber Security, University of Chinese Academy of Sciences

Kaiyue Yang, Yuyan Bu, Jingwei Yi, Yuchi Wang, Biyu Zhou, Juntao Dai 等

揭示LLM代理在工具选择时偏好高权限工具且故障会加剧该倾向,并提出后训练防御以遵循最小权限原则

📌 工作参考价值

  • 对电商客服代理等场景,可将工具按权限分级,并在代理系统设计中强制‘最低权限’原则,避免因模型偏好而越权调用退款、改价等高危接口。
  • 在后训练阶段加入权限感知偏好优化(类似论文的 privilege-aware post-training defense),通过构造偏好对让代理优先选择功能足够但权限更低的工具,从根源上减少过度授权。
  • 仔细处理工具调用失败的降级/重试逻辑,防止代理在遇到瞬时故障时自动升级到更高级别工具,需预设权限升级白名单或人工审批流程。
  • 可构建领域专属的工具权限基准(类似 ToolPrivBench),评估代理在订单查询、修改、删除等不同权限工具间的选择合规性,作为安全回归测试的一部分。

动机
随着LLM代理自主调用各种权限的工具,安全风险凸显。现有研究忽视权限敏感的选择,而现实中代理常在有足够低权限工具时仍选用高权限工具,过度暴露风险。本文系统研究这种“过度权限工具选择”问题,并探索有效防御。

方法与关键发现
构建评估基准ToolPrivBench,覆盖8个领域、5类反复出现的风险模式,测试代理在初始选择以及工具瞬时故障后是否会升级权限。实验表明主流LLM代理普遍存在过度权限选择,且故障会显著加剧该倾向;通用安全对齐方法难以迁移到最小权限工具选择场景,提示词级控制对故障下的权限升级缓解有限。
基于此,设计一种特权感知的后训练防御:通过构造偏好数据,教导代理偏好功能满足且权限更低的工具,仅在必要时升级。训练采用DPO等偏好对齐方法,最小化对通用能力的损害。

主要结果
防御方案大幅减少了不必要的高权限工具调用(在某些设置下将过度选择率从>50%降至<10%),同时保持了代理的通用任务成功率,验证了最小权限工具选择的可控性与可行性。