arXiv AI By Chuming Wei, Churui Yang, Zihan Zhang, Ruien Li, Wusheng Zhang, Yan Zhang, Guangwen Yang

ADTGuard: Adaptive Depth-Time Safeguard for Streaming Inference in Large Language Models

Read the original on arXiv AI →

ADTGuard is a new safeguard for large language models that monitors each token before it is streamed to users. It uses Depth-Time Safety Dynamics to combine safety signals from multiple layers and a Causal Depth Router to select the most relevant layers for each prefix. With only response‑level labels, it refines token‑level training targets and achieves strong streaming detection performance with minimal parameters and latency.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv AI.

Hugging Face Trending Papers
Jun 1

SentGuard: Sentence-Level Streaming Guardrails for Large Language Models

Large language models increasingly stream long, reasoning-intensive responses in real time, making when to moderate as critical as whether to moderate. Existing guardrails fall into two unsatisfactory extremes: response-level methods delay intervention until the full output is generated, whereas token-level methods act on incomplete semantics, often producing unstable decisions and excessive guard invocations.

arXiv AI
1d ago

UpSafe$^\circ$C: Upcycling for Controllable Safety in Large Language Models

UpSafe°C is a unified framework that enhances large language model safety by upcycling safety‑critical layers into a sparse Mixture‑of‑Experts structure, with a router acting as a soft guardrail. It employs a two‑stage supervised fine‑tuning strategy to improve safety discrimination while preserving general capabilities, and introduces a safety temperature mechanism for dynamic inference‑time control of the safety‑utility trade‑off. Experiments across multiple benchmarks and model scales show robust safety improvements against harmful and jailbreak inputs while maintaining competitive general performance.

By Yuhao Sun, Zhuoer Xu, Shiwen Cui, Kun Yang, Lingyun Yu, Yongdong Zhang, Hongtao Xie