AI Safety & Alignment

AgentDoG: A Diagnostic Guardrail Framework for AI Agent Safety and Security

DDongrui LiuQQihan RenCChen QianSShuai ShaoYYuejin XieYYu LiZZhonghao YangHHaoyu LuoPPeng WangQQingyu LiuBBinxin HuLLing TangJJilin MeiDDadi GuoLLeitao YuanJJunyao YangGGuanxu ChenQQihao LinYYi YuBBo ZhangJJiaxuan GuoJJie ZhangWWenqi ShaoHHuiqi DengZZhiheng XiWWenjie WangWWenxuan WangWWen ShenZZhikai ChenHHaoyu XieJJialing TaoJJuntao DaiJJiaming JiZZhongjie BaLLinfeng ZhangYYong LiuQQuanshi ZhangLLei ZhuZZhihua WeiHHui XueCChaochao LuJJing ShaoXXia Hu
Published
January 26, 2026
Authors
43

Abstract

The rise of AI agents introduces complex safety and security challenges arising from autonomous tool use and environmental interactions. Current guardrail models lack agentic risk awareness and transparency in risk diagnosis. To introduce an agentic guardrail that covers complex and numerous risky behaviors, we first propose a unified three-dimensional taxonomy that orthogonally categorizes agentic risks by their source (where), failure mode (how), and consequence (what). Guided by this structured and hierarchical taxonomy, we introduce a new fine-grained agentic safety benchmark (ATBench) and a Diagnostic Guardrail framework for agent safety and security (AgentDoG). AgentDoG provides fine-grained and contextual monitoring across agent trajectories. More Crucially, AgentDoG can diagnose the root causes of unsafe actions and seemingly safe but unreasonable actions, offering provenance and transparency beyond binary labels to facilitate effective agent alignment. AgentDoG variants are available in three sizes (4B, 7B, and 8B parameters) across Qwen and Llama model families. Extensive experimental results demonstrate that AgentDoG achieves state-of-the-art performance in agentic safety moderation in diverse and complex interactive scenarios. All models and datasets are openly released.

Keywords

agentic guardrailthree-dimensional taxonomyagentic safety benchmarkDiagnostic Guardrail frameworkagent safety and securityagent trajectoriesroot cause diagnosisfine-grained monitoringmodel variantsstate-of-the-art performance

More in AI Safety & Alignment

View all
AgentDoG: A Diagnostic Guardrail Framework for AI Agent Safety and Security | Paperchime