AEGIS: Online Adaptive AI Content Safety Moderation with Ensemble of LLM Experts

التفاصيل البيبلوغرافية
العنوان: AEGIS: Online Adaptive AI Content Safety Moderation with Ensemble of LLM Experts
المؤلفون: Ghosh, Shaona, Varshney, Prasoon, Galinkin, Erick, Parisien, Christopher
سنة النشر: 2024
المجموعة: Computer Science
مصطلحات موضوعية: Computer Science - Machine Learning, Computer Science - Computation and Language, Computer Science - Computers and Society
الوصف: As Large Language Models (LLMs) and generative AI become more widespread, the content safety risks associated with their use also increase. We find a notable deficiency in high-quality content safety datasets and benchmarks that comprehensively cover a wide range of critical safety areas. To address this, we define a broad content safety risk taxonomy, comprising 13 critical risk and 9 sparse risk categories. Additionally, we curate AEGISSAFETYDATASET, a new dataset of approximately 26, 000 human-LLM interaction instances, complete with human annotations adhering to the taxonomy. We plan to release this dataset to the community to further research and to help benchmark LLM models for safety. To demonstrate the effectiveness of the dataset, we instruction-tune multiple LLM-based safety models. We show that our models (named AEGISSAFETYEXPERTS), not only surpass or perform competitively with the state-of-the-art LLM-based safety models and general purpose LLMs, but also exhibit robustness across multiple jail-break attack categories. We also show how using AEGISSAFETYDATASET during the LLM alignment phase does not negatively impact the performance of the aligned models on MT Bench scores. Furthermore, we propose AEGIS, a novel application of a no-regret online adaptation framework with strong theoretical guarantees, to perform content moderation with an ensemble of LLM content safety experts in deployment
نوع الوثيقة: Working Paper
URL الوصول: http://arxiv.org/abs/2404.05993
رقم الأكسشن: edsarx.2404.05993
قاعدة البيانات: arXiv