DT-Guard: Intent-Driven Reasoning-Active Training for Reasoning-Free LLM Safety Guardrail
Summary
DT-Guard is a novel content safety guardrail model for large language models, designed to overcome the practical trade-off between robust reasoning-based guards and efficient classification-based models. It employs a "Reasoning-Active Training, Reasoning-Free Inference" paradigm, leveraging reasoning supervision during training to internalize complex safety judgments, while only emitting structured safety labels at inference time for low-latency deployment. The model formulates safety judgment as a progressive "Intent - Category - Safety" decision process and utilizes an intent-driven dataset with detailed reasoning trajectories. To enhance robustness against challenging cases, DT-Guard incorporates Rollout-Guided Progressive Hard-Case Optimization (RG-PHO), which uses multi-rollout consistency for targeted supervised and preference optimization. Experiments demonstrate DT-Guard's effectiveness, achieving average F1 scores of 0.886 on prompt-side and 0.870 on response-side safety benchmarks. With a 4B backbone, it reaches a dual-side average F1 of 0.878, surpassing stronger 8B guardrail baselines.
Key takeaway
For Machine Learning Engineers deploying LLM safety guardrails, DT-Guard's "Reasoning-Active Training, Reasoning-Free Inference" paradigm offers a compelling solution to the efficiency-robustness trade-off. You can achieve robust safety judgments, even for concealed intent, without incurring high inference latency. Consider integrating reasoning supervision into your guardrail training processes and exploring progressive decision frameworks to enhance both performance and deployment efficiency.
Key insights
Reasoning supervision during training can internalize complex safety judgments for efficient, reasoning-free LLM guardrail inference.
Principles
- Safety judgment benefits from progressive decision processes.
- Multi-rollout consistency improves hard-case robustness.
- Internalizing reasoning during training enhances efficiency.
Method
DT-Guard uses Reasoning-Active Training with an intent-driven dataset and "Intent - Category - Safety" judgment. Rollout-Guided Progressive Hard-Case Optimization refines hard cases via multi-rollout consistency and targeted optimization.
In practice
- Implement intent-driven safety judgment flows.
- Use multi-rollout consistency for hard-case identification.
- Apply reasoning supervision during guardrail training.
Topics
- LLM Safety Guardrails
- Reasoning-Active Training
- Inference Efficiency
- Content Moderation
- Hard-Case Optimization
Best for: Research Scientist, AI Scientist, Machine Learning Engineer, AI Security Engineer
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.