🤖 AI Summary
This work addresses key limitations of reasoning-based large language models (R-LLMs) in content moderation safety fencing—namely, poor generalization, high dependence on training data, and unpredictable inference latency—by proposing an efficient, reasoning-enhanced framework tailored to customizable safety policies. Methodologically, it first systematically validates the sample efficiency of R-LLMs for safety-fencing tasks; introduces a reasoning budget mechanism and a dual-mode (direct-answer/reasoning) joint training framework enabling dynamic control of reasoning depth; and incorporates reasoning-length constraints and active hard-sample mining. Experiments demonstrate a several-fold reduction in training data requirements to achieve comparable accuracy, and enable fine-grained, controllable trade-offs between accuracy and latency—significantly enhancing practical deployability in safety-critical settings.
📝 Abstract
Reasoning-based language models have demonstrated strong performance across various domains, with the most notable gains seen in mathematical and coding tasks. Recent research has shown that reasoning also offers significant benefits for LLM safety and guardrail applications. In this work, we conduct a comprehensive analysis of training reasoning-based guardrail models for content moderation, with an emphasis on generalization to custom safety policies at inference time. Our study focuses on two key dimensions: data efficiency and inference efficiency. On the data front, we find that reasoning-based models exhibit strong sample efficiency, achieving competitive performance with significantly fewer training examples than their non-reasoning counterparts. This unlocks the potential to repurpose the remaining data for mining high-value, difficult samples that further enhance model performance. On the inference side, we evaluate practical trade-offs by introducing reasoning budgets, examining the impact of reasoning length on latency and accuracy, and exploring dual-mode training to allow runtime control over reasoning behavior. Our findings will provide practical insights for researchers and developers to effectively and efficiently train and deploy reasoning-based guardrails models in real-world systems.