Search papers, labs, and topics across Lattice.
This paper introduces a segment-level coherence objective for streaming probes to improve the detection of harmful intent in LLMs, particularly against adaptive jailbreaking in CBRN domains. The method requires multiple evidence tokens to consistently support a prediction, moving away from reliance on single high-scoring tokens. Experiments show a 35.55% improvement in true-positive rate at a 1% false-positive rate compared to strong baselines, and high AUROC even with adversarial fine-tuning using character-level ciphers.
LLM safety probes can be made significantly more robust to adversarial attacks by requiring consistent evidence across token segments, not just isolated spikes.
Large Language Models (LLMs) are increasingly exposed to adaptive jailbreaking, particularly in high-stakes Chemical, Biological, Radiological, and Nuclear (CBRN) domains. Although streaming probes enable real-time monitoring, they still make systematic errors. We identify a core issue: existing methods often rely on a few high-scoring tokens, leading to false alarms when sensitive CBRN terms appear in benign contexts. To address this, we introduce a streaming probing objective that requires multiple evidence tokens to consistently support a prediction, rather than relying on isolated spikes. This encourages more robust detection based on aggregated signals instead of single-token cues. At a fixed 1% false-positive rate, our method improves the true-positive rate by 35.55% relative to strong streaming baselines. We further observe substantial gains in AUROC, even when starting from near-saturated baseline performance (AUROC = 97.40%). We also show that probing Attention or MLP activations consistently outperforms residual-stream features. Finally, even when adversarial fine-tuning enables novel character-level ciphers, harmful intent remains detectable: probes developed for the base LLMs can be applied ``plug-and-play''to these obfuscated attacks, achieving an AUROC of over 98.85%.