Athena: Safe Autonomous Agents with Verbal Contrastive Learning
This work addresses safety concerns for autonomous agents in AI applications, though it is incremental as it builds on existing contrastive learning and critiquing techniques.
The authors tackled the problem of ensuring safety in autonomous language-based agents by introducing the Athena framework, which uses verbal contrastive learning and a critiquing mechanism, resulting in a significant improvement in safety rates as demonstrated in experiments with LLMs.
Due to emergent capabilities, large language models (LLMs) have been utilized as language-based agents to perform a variety of tasks and make decisions with an increasing degree of autonomy. These autonomous agents can understand high-level instructions, interact with their environments, and execute complex tasks using a selection of tools available to them. As the capabilities of the agents expand, ensuring their safety and trustworthiness becomes more imperative. In this study, we introduce the Athena framework which leverages the concept of verbal contrastive learning where past safe and unsafe trajectories are used as in-context (contrastive) examples to guide the agent towards safety while fulfilling a given task. The framework also incorporates a critiquing mechanism to guide the agent to prevent risky actions at every step. Furthermore, due to the lack of existing benchmarks on the safety reasoning ability of LLM-based agents, we curate a set of 80 toolkits across 8 categories with 180 scenarios to provide a safety evaluation benchmark. Our experimental evaluation, with both closed- and open-source LLMs, indicates verbal contrastive learning and interaction-level critiquing improve the safety rate significantly.