SGuard-v1: Safety Guardrail for Large Language Models

TL;DR

SGuard-v1 is a lightweight safety framework for large language models that detects harmful content and adversarial prompts, improving safety and interpretability with minimal deployment overhead.

Contribution

It introduces a dual-model safety guardrail for LLMs, trained on extensive datasets, achieving state-of-the-art safety performance while maintaining efficiency.

Findings

Achieves state-of-the-art safety benchmarks

Reduces deployment overhead compared to larger models

Provides interpretable safety predictions

Abstract

We present SGuard-v1, a lightweight safety guardrail for Large Language Models (LLMs), which comprises two specialized models to detect harmful content and screen adversarial prompts in human-AI conversational settings. The first component, ContentFilter, is trained to identify safety risks in LLM prompts and responses in accordance with the MLCommons hazard taxonomy, a comprehensive framework for trust and safety assessment of AI. The second component, JailbreakFilter, is trained with a carefully designed curriculum over integrated datasets and findings from prior work on adversarial prompting, covering 60 major attack types while mitigating false-unsafe classification. SGuard-v1 is built on the 2B-parameter Granite-3.3-2B-Instruct model that supports 12 languages. We curate approximately 1.4 million training instances from both collected and synthesized data and perform instruction tuning on the base model, distributing the curated data across the two component according to their designated functions. Through extensive evaluation on public and proprietary safety benchmarks, SGuard-v1 achieves state-of-the-art safety performance while remaining lightweight, thereby reducing deployment overhead. SGuard-v1 also improves interpretability for downstream use by providing multi-class safety predictions and their binary confidence scores. We release the SGuard-v1 under the Apache-2.0 License to enable further research and practical deployment in AI safety.