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
揭示LLM代理在工具选择时偏好高权限工具且故障会加剧该倾向,并提出后训练防御以遵循最小权限原则
📌 工作参考价值
- 对电商客服代理等场景,可将工具按权限分级,并在代理系统设计中强制‘最低权限’原则,避免因模型偏好而越权调用退款、改价等高危接口。
- 在后训练阶段加入权限感知偏好优化(类似论文的 privilege-aware post-training defense),通过构造偏好对让代理优先选择功能足够但权限更低的工具,从根源上减少过度授权。
- 仔细处理工具调用失败的降级/重试逻辑,防止代理在遇到瞬时故障时自动升级到更高级别工具,需预设权限升级白名单或人工审批流程。
- 可构建领域专属的工具权限基准(类似 ToolPrivBench),评估代理在订单查询、修改、删除等不同权限工具间的选择合规性,作为安全回归测试的一部分。
动机
随着LLM代理自主调用各种权限的工具,安全风险凸显。现有研究忽视权限敏感的选择,而现实中代理常在有足够低权限工具时仍选用高权限工具,过度暴露风险。本文系统研究这种“过度权限工具选择”问题,并探索有效防御。
方法与关键发现
构建评估基准ToolPrivBench,覆盖8个领域、5类反复出现的风险模式,测试代理在初始选择以及工具瞬时故障后是否会升级权限。实验表明主流LLM代理普遍存在过度权限选择,且故障会显著加剧该倾向;通用安全对齐方法难以迁移到最小权限工具选择场景,提示词级控制对故障下的权限升级缓解有限。
基于此,设计一种特权感知的后训练防御:通过构造偏好数据,教导代理偏好功能满足且权限更低的工具,仅在必要时升级。训练采用DPO等偏好对齐方法,最小化对通用能力的损害。
主要结果
防御方案大幅减少了不必要的高权限工具调用(在某些设置下将过度选择率从>50%降至<10%),同时保持了代理的通用任务成功率,验证了最小权限工具选择的可控性与可行性。