NeFut Logo NeFut
Admin Login

[CS.AI] DT-Guard: Intent-Driven Reasoning-Active Training for LLM Safety

Published at: 2026-07-08 22:00 Last updated: 2026-07-09 03:24
#AI #Machine Learning #Open Source

Abstract

Large language models deployed in open-world applications require safety guardrails that are both robust to complex risks and efficient enough for low-latency runtime moderation. Existing guardrails face a practical trade-off between lightweight classification-based models, which are efficient but often struggle with concealed intent, ambiguous semantics, and borderline safety decisions, and reasoning-based guards, which improve judgment quality but introduce additional token generation and inference latency.

We present DT-Guard, a content safety guardrail model based on a Reasoning-Active Training, Reasoning-Free Inference paradigm. The key idea is to use reasoning supervision during training while emitting only structured safety labels at inference time. DT-Guard formulates safety judgment as a progressive decision process: Intent - Category - Safety, and constructs an intent-driven dataset with intent labels, risk categories, safety labels, and structured reasoning trajectories.

To further improve hard-case robustness, we propose Rollout-Guided Progressive Hard-Case Optimization (RG-PHO), which uses multi-rollout consistency to identify stably mastered, persistently failed, and preference-unstable samples, and applies targeted supervision and preference optimization accordingly. At inference time, DT-Guard directly generates structured labels without explicit reasoning traces, preserving deployment efficiency.

Experiments on prompt-side and response-side safety benchmarks show that DT-Guard achieves average F1 scores of 0.886 and 0.870, respectively. With only a 4B backbone, it reaches a dual-side average F1 of 0.878, outperforming strong 8B guardrail baselines. These results demonstrate that reasoning supervision can be effectively internalized into low-latency safety discrimination.

Blogger's Review: The design of DT-Guard cleverly balances reasoning quality with operational efficiency, showcasing how innovative training strategies can enhance model practicality in safety guardrails. Its efficient structured label generation offers a new perspective for safe applications of large language models, warranting further research and promotion.

Original Source: https://arxiv.org/abs/2607.06326

[h] Back to Home