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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Response Harmfulness Detection | Beavertails | -- | 59 | |
| Safety Classification | SafeRLHF | -- | 48 | |
| Harmfulness Detection | WildGuard | Macro F1 Score88.3 | 47 | |
| Harmfulness Detection | OpenAI Moderation | Macro F1 Score92.9 | 45 | |
| Toxicity Detection | ToxicChat | -- | 45 | |
| Harmfulness Detection | Aegis | Macro F182.9 | 25 | |
| Harmfulness Detection | Aegis 2.0 | Macro F183.4 | 8 |