Lune

ICML2026Top-tier venue

SafeCompass: Dynamic Chain-of-Thought Steering via Inference-Time Safety Signals

Zeyang Zhang, HAOTIAN XU, Linbao Li, Qi Sun, Xuebo Liu, YU LI, Cheng Zhuo

2026Year

Abstract

Large reasoning models (LRMs) achieve strong performance by explicitly generating chain-of-thought (CoT) reasoning, but this reasoning process can be manipulated by adversarial prompts. Inference-time CoT interventions offer a simple and lightweight approach to improving safety, yet existing methods typically apply static heuristics that ignore the dynamic nature of reasoning, leading to an inherent trade-off between robustness and over-refusal. This paper introduces SafeCompass, a plug-and-play framework for dynamically steering chain-of-thought reasoning using inference-time safety signals extracted from internal states. At different reasoning positions, SafeCompass derives a latent safety direction through contrastive analysis of internal representations and uses this direction to quantify the model’s current safety state. These signals enable selective intervention, allowing the model’s reasoning trajectory to be modified only when and where it becomes unsafe. Extensive experiments demonstrate that SafeCompass significantly improves robustness, reducing the average attack success rate up to 10×10\times compared to the best baseline, while preserving general reasoning performance and minimizing over-refusal rates.

Ask about this paper

Your agent reads all of it.

Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.

Questions to start from

Your agent calls

Luneget_paper_fulltext

Ask in Lune

Free to start. No credit card required.

Builds on15

Related papers

Dusk over the sea between two cliffs drawn in fine vertical lines