Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

ShieldGemma: Generative AI Content Moderation Based on Gemma

About

We present ShieldGemma, a comprehensive suite of LLM-based safety content moderation models built upon Gemma2. These models provide robust, state-of-the-art predictions of safety risks across key harm types (sexually explicit, dangerous content, harassment, hate speech) in both user input and LLM-generated output. By evaluating on both public and internal benchmarks, we demonstrate superior performance compared to existing models, such as Llama Guard (+10.8\% AU-PRC on public benchmarks) and WildCard (+4.3\%). Additionally, we present a novel LLM-based data curation pipeline, adaptable to a variety of safety-related tasks and beyond. We have shown strong generalization performance for model trained mainly on synthetic data. By releasing ShieldGemma, we provide a valuable resource to the research community, advancing LLM safety and enabling the creation of more effective content moderation solutions for developers.

Wenjun Zeng, Yuchi Liu, Ryan Mullins, Ludovic Peran, Joe Fernandez, Hamza Harkous, Karthik Narasimhan, Drew Proud, Piyush Kumar, Bhaktipriya Radharapu, Olivia Sturman, Oscar Wahltinez• 2024

Related benchmarks

TaskDatasetResultRank
Response Harmfulness DetectionXSTEST-RESP
Response Harmfulness F173.86
34
Safety ClassificationSafeRLHF
F1 Score0.4707
32
Response Harmfulness ClassificationWildGuard (test)
F1 (Total)47
30
Prompt ClassificationSEA-SafeguardBench
AUPRC (Average)82.8
29
Text-based safety moderationToxic Chat
F1 Score78.4
24
Response Harmfulness DetectionHarmBench
F1 Score56.44
23
Response ClassificationBeaverTails V Text-Image Response
F1 Score57
23
Trajectory-level safety evaluationASSE-Safety (test)
Accuracy47.2
20
Trajectory-level safety evaluationATBench (test)
Accuracy0.511
20
Trajectory-level safety evaluationR-judge (test)
Accuracy47.7
20
Showing 10 of 76 rows
...

Other info

Follow for update