2026/04/20 by Difan Jiao, Yilun Liu, Ye Yuan +4 · 1 voice
Computer Science · #Adversarial Robustness in Machine Learning #Generalization #Generative grammar #Guard (computer science) #Hate Speech and Cyberbullying Detection #Inference #Siren (mythology) #Topic Modeling #cs.AI
paper · pdf · open access · doi:10.48550/arxiv.2604.18519
published in arXiv (Cornell University) (Cornell University)
openalex publication_date 2026/04/20 · arxiv published 2026/04/20 · arxiv updated 2026/04/20 · openalex created_date 2026/04/22 · openalex updated_date 2026/07/28
Guard models are widely used to detect harmful content in user prompts and LLM responses. However, state-of-the-art guard models rely solely on terminal-layer representations and overlook the rich safety-relevant features distributed across internal layers. We present SIREN, a lightweight guard model that harnesses these internal features. By identifying safety neurons via linear probing and combining them through an adaptive layer-weighted strategy, SIREN builds a harmfulness detector from LLM internals without modifying the underlying model. Our comprehensive evaluation shows that SIREN substantially outperforms state-of-the-art open-source guard models across multiple benchmarks while using 250 times fewer trainable parameters. Moreover, SIREN exhibits superior generalization to unseen benchmarks, naturally enables real-time streaming detection, and significantly improves inference efficiency compared to generative guard models. Overall, our results highlight LLM internal states as a promising foundation for practical, high-performance harmfulness detection.