Benchmarking and Improving Monitors for Out-Of-Distribution Alignment Failure in LLMs
Summary
A new benchmark, Misalignment Out Of Distribution (MOOD), systematically evaluates Large Language Model (LLM) monitoring pipelines for detecting out-of-distribution (OOD) alignment failures. These failures arise from unusual prompt or response patterns unforeseen by developers. The study reveals that traditional guard models often fail to generalize OOD. To address this, researchers propose combining guard models with OOD detectors, testing four types. A combination of a guard model with Mahalanobis distance and perplexity-based OOD detectors improved recall from 39% to 45% on the MOOD benchmark's seven diverse test sets. This approach also demonstrated positive scaling trends across model sizes, with a 1.5B parameter model combined with OOD detection outperforming a 20 times larger 32B parameter guard model alone. The work suggests OOD detection is a crucial component for robust LLM monitoring.
Key takeaway
For MLOps Engineers deploying LLMs, relying solely on guard models for safety monitoring is insufficient against out-of-distribution (OOD) alignment failures. You should integrate OOD detection methods, specifically Mahalanobis distance and perplexity-based detectors, into your monitoring pipelines. This combination significantly boosts recall for unforeseen misalignments, even outperforming much larger guard models alone. Consider fine-tuning OOD detectors with a small set of benign OOD samples to minimize false positives.
Key insights
Combining OOD detectors with guard models significantly improves LLM safety monitoring for unforeseen alignment failures.
Principles
- Guard models alone struggle with OOD generalization.
- OOD detection enhances recall for unseen alignment failures.
- Performance of combined monitors scales with model size.
Method
The study trains monitors on a restricted dataset, then evaluates generalization to seven OOD alignment failure test sets using recall at a 1% false positive rate.
In practice
- Integrate Mahalanobis distance and perplexity OOD detectors.
- Prioritize last token activations for Mahalanobis distance.
- Add benign OOD samples to reduce false positives.
Topics
- LLM Safety
- Out-of-Distribution Detection
- Alignment Failures
- MOOD Benchmark
- Guard Models
- Mahalanobis Distance
- Perplexity
Best for: Research Scientist, AI Scientist, MLOps Engineer, AI Security Engineer
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by cs.AI updates on arXiv.org.