arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2501.09004cs.CL
收藏

Aegis2.0:面向大语言模型护栏对齐的多样化AI安全数据集与风险分类体系

Aegis2.0: A Diverse AI Safety Dataset and Risks Taxonomy for Alignment of LLM Guardrails

NVIDIA(英伟达)

Shaona Ghosh, Prasoon Varshney, Makesh Narsimhan Sreedhar, Aishwarya Padmakumar, Traian Rebedea, Jibin Rajan Varghese, Christopher Parisien

AI总结:

针对现有LLM安全数据集的不足,研究提出含12个顶层类的风险分类体系,构建34248条样本的Aegis 2.0数据集,验证了轻量模型的竞争力与新型训练混合策略的泛化能力,并计划开源。

摘要:

随着大语言模型(LLM)和生成式AI日益普及,人们对内容安全的担忧也随之加剧。目前,业界明显缺乏覆盖LLM相关全谱系安全风险、可用于商业应用的高质量人工标注数据集。为填补这一空白,我们提出了一套全面且适配性强的安全风险分类体系,包含12个顶层危害类别,并扩展至9个细粒度子类别。该分类体系旨在满足下游用户的多样化需求,为管理各类风险提供更精细、灵活的工具。我们采用混合数据生成流水线,将人工标注与多LLM“评审团”系统相结合以评估回复的安全性,最终构建出Aegis 2.0——一个经过精心筛选的、包含34248条人机交互样本的数据集,所有样本均依据我们提出的分类体系完成标注。为验证其有效性,我们实验证明:在Aegis 2.0上采用参数高效技术训练的多个轻量模型,其性能可与在规模大得多的非商业数据集上完成全量微调的主流安全模型相媲美。此外,我们提出了一种将安全数据与主题遵循数据相结合的新型训练混合策略。该方法提升了护栏模型的适配能力,使其能够泛化到推理阶段定义的新风险类别。我们计划向研究界开源Aegis 2.0的数据集与模型,以助力LLM的安全护栏建设。

英文摘要:

As Large Language Models (LLMs) and generative AI become increasingly widespread, concerns about content safety have grown in parallel. Currently, there is a clear lack of high-quality, human-annotated datasets that address the full spectrum of LLM-related safety risks and are usable for commercial applications. To bridge this gap, we propose a comprehensive and adaptable taxonomy for categorizing safety risks, structured into 12 top-level hazard categories with an extension to 9 fine-grained subcategories. This taxonomy is designed to meet the diverse requirements of downstream users, offering more granular and flexible tools for managing various risk types. Using a hybrid data generation pipeline that combines human annotations with a multi-LLM "jury" system to assess the safety of responses, we obtain Aegis 2.0, a carefully curated collection of 34,248 samples of human-LLM interactions, annotated according to our proposed taxonomy. To validate its effectiveness, we demonstrate that several lightweight models, trained using parameter-efficient techniques on Aegis 2.0, achieve performance competitive with leading safety models fully fine-tuned on much larger, non-commercial datasets. In addition, we introduce a novel training blend that combines safety with topic following data.This approach enhances the adaptability of guard models, enabling them to generalize to new risk categories defined during inference. We plan to open-source Aegis 2.0 data and models to the research community to aid in the safety guardrailing of LLMs.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑