SafetyAccountabilityTransparencycs.CL
Aegis2.0: A Diverse AI Safety Dataset and Risks Taxonomy for Alignment of LLM Guardrails
Shaona Ghosh, Prasoon Varshney, Makesh Narsimhan Sreedhar, Aishwarya Padmakumar, Traian Rebedea, Jibin Rajan Varghese, Christopher Parisien
January 15, 2025
North American Chapter of the Association for Computational Linguistics
176 citations
44 influential
Key Takeaways
-
Comprehensive safety framework needed: The paper identifies a significant gap in AI safety—there's a lack of high-quality, human-annotated datasets that cover the full range of potential harms from large language models. Aegis 2.0 addresses this by creating a taxonomy with 12 main hazard categories and 9 subcategories, providing a structured way to think about and manage different types of AI risks.
-
Human judgment remains essential: The dataset combines human annotations with an AI "jury system" to evaluate safety, demonstrating that human oversight is still critical for determining whether AI responses are actually safe. This hybrid approach resulted in 34,248 carefully vetted examples of human-AI interactions.
-
Smaller, efficient models can be effective: The research shows that lightweight safety models trained on Aegis 2.0 perform comparably to larger models trained on bigger datasets, suggesting that organizations don't need massive computational resources to build effective AI guardrails—quality of training data matters as much as quantity.
-
AI safety models need flexibility: The paper introduces a novel training approach that allows safety models to adapt to new types of risks that emerge after deployment, rather than being locked into predefined categories. This is important because new harms and misuse cases will inevitably arise as AI technology evolves.
-
Open-source safety tools advance the field: By planning to release Aegis 2.0 publicly, the researchers are democratizing access to safety resources, enabling smaller organizations and researchers to build more responsible AI systems rather than concentrating safety expertise only among large companies.