ARMOR: Aligning Secure and Safe Large Language Models via Meticulous Reasoning
About
Large Language Models have shown impressive generative capabilities across diverse tasks, but their safety remains a critical concern. Existing post-training alignment methods, such as SFT and RLHF, reduce harmful outputs yet leave LLMs vulnerable to jailbreak attacks, especially advanced optimization-based ones. Recent system-2 approaches enhance safety by adding inference-time reasoning, where models assess potential risks before producing responses. However, we find these methods fail against powerful out-of-distribution jailbreaks, such as AutoDAN-Turbo and Adversarial Reasoning, which conceal malicious goals behind seemingly benign prompts. We observe that all jailbreaks ultimately aim to embed a core malicious intent, suggesting that extracting this intent is key to defense. To this end, we propose ARMOR, which introduces a structured three-step reasoning pipeline: (1) analyze jailbreak strategies from an external, updatable strategy library, (2) extract the core intent, and (3) apply policy-based safety verification. We further develop ARMOR-Think, which decouples safety reasoning from general reasoning to improve both robustness and utility. Evaluations on advanced optimization-based jailbreaks and safety benchmarks show that ARMOR achieves state-of-the-art safety performance, with an average harmful rate of 0.002 and an attack success rate of 0.06 against advanced optimization-based jailbreaks, far below other reasoning-based models. Moreover, ARMOR demonstrates strong generalization to unseen jailbreak strategies, reducing their success rate to zero. These highlight ARMOR's effectiveness in defending against OOD jailbreak attacks, offering a practical path toward secure and reliable LLMs.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Mathematical Reasoning | MATH | Accuracy84 | 55 | |
| Jailbreak Robustness | AdvBench AutoDAN-Turbo (original) | ASR4 | 9 | |
| Jailbreak Robustness | AdvBench AdvReasoning (original) | ASR6 | 9 | |
| Safety Evaluation | BeaverTail v1 (eval) | Compliance Rate1.3 | 9 | |
| Safety Evaluation | XSTest Unsafe v1 | Compliance Rate2.8 | 9 | |
| Jailbreak Robustness | AdvBench w/o attack (original) | ASR0.00e+0 | 9 | |
| Safety Evaluation | HarmfulQA v1 (test) | Compliance Rate1.2 | 9 | |
| Safety Evaluation | XSTest Safe v1 | Accuracy86 | 9 | |
| Jailbreak Robustness | JailbreakV v1 (test) | Compliance Rate0.00e+0 | 9 | |
| Jailbreak Robustness | PAIR v1 (test) | Compliance Rate2 | 9 |