Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

LLM Safety From Within: Detecting Harmful Content with Internal Representations

About

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.

Difan Jiao, Yilun Liu, Ye Yuan, Zhenwei Tang, Linfeng Du, Haolun Wu, Ashton Anderson• 2026

Related benchmarks

TaskDatasetResultRank
Harmfulness DetectionOpenAI Moderation
Macro F1 Score92.9
59
Response Harmfulness DetectionBeavertails--
59
Safety ClassificationSafeRLHF--
48
Harmfulness DetectionWildGuard
Macro F1 Score88.3
47
Toxicity DetectionToxicChat--
45
Harmfulness DetectionAegis
Macro F182.9
25
Harmfulness DetectionAegis 2.0
Macro F183.4
8
Showing 7 of 7 rows

Other info

GitHub

Follow for update