VeriGuard: Enhancing LLM Agent Safety via Verified Code Generation
Lesly Miculicich, Mihir Parmar, Hamid Palangi, Krishnamurthy Dj Dvijotham, Mirko Montanari, Tomas Pfister, Long T. Le
TL;DR
VeriGuard tackles the safety challenges of autonomous LLM agents in high-stakes domains by marrying formal verification with runtime enforcement. It introduces a two-stage pipeline: offline policy generation, validation, and formal verification to yield a correct-by-construction safety policy, and online monitoring that enforces the verified policy before every action. Empirical results across ASB, EICU-AC, and Mind2Web-SC demonstrate substantial reductions in attack success while preserving task utility, with a hybrid Collaborative Re-planning plus Tool Execution Halt strategy often delivering the best security-utility balance. The work advances practical, provable safety for LLM-driven agents and points to scalable verification and specification generation as promising directions for real-world deployment.
Abstract
The deployment of autonomous AI agents in sensitive domains, such as healthcare, introduces critical risks to safety, security, and privacy. These agents may deviate from user objectives, violate data handling policies, or be compromised by adversarial attacks. Mitigating these dangers necessitates a mechanism to formally guarantee that an agent's actions adhere to predefined safety constraints, a challenge that existing systems do not fully address. We introduce VeriGuard, a novel framework that provides formal safety guarantees for LLM-based agents through a dual-stage architecture designed for robust and verifiable correctness. The initial offline stage involves a comprehensive validation process. It begins by clarifying user intent to establish precise safety specifications. VeriGuard then synthesizes a behavioral policy and subjects it to both testing and formal verification to prove its compliance with these specifications. This iterative process refines the policy until it is deemed correct. Subsequently, the second stage provides online action monitoring, where VeriGuard operates as a runtime monitor to validate each proposed agent action against the pre-verified policy before execution. This separation of the exhaustive offline validation from the lightweight online monitoring allows formal guarantees to be practically applied, providing a robust safeguard that substantially improves the trustworthiness of LLM agents.
