研究揭示大模型推理中安全崩溃的起始机制
First Token Matters: Understanding Safety Collapse in Large Reasoning Models
朋友间推荐:研究大模型安全问题的,这篇论文提出了一个叫SafeToken的方法,在生成第一个词的时候干预,能改善安全性能,推荐看看。
本文通过分析大模型(LRMs)在处理有害查询时的拒绝行为,发现其安全对齐能力在生成第一个token时会出现显著下降,并提出了SafeToken方法在推理开始时注入安全锚点来缓解这一问题。
First Token Matters: Understanding Safety Collapse in Large Reasoning Models
Large Reasoning Models (LRMs) exhibit strong problem-solving abilities, yet their safety alignment often degrades when handling harmful queries. Existing approaches to improving safety largely rely on additional training or preference optimization, while offering limited understanding of the internal mechanisms behind safety failures. In this work, we investigate this failure through a token-level positional analysis of refusal dynamics and identify a localized vulnerability at the onset of reasoning, which we term Onset Refusal Collapse (ORC). We find that the refusal-related signal of LRMs drops sharply at the first generated token under harmful queries, which is associated with unsafe response generation. Motivated by this finding, we propose SafeToken, a lightweight inference-time intervention that injects a learned continuous safety anchor precisely at reasoning onset. Despite updating only a single token embedding, SafeToken effectively mitigates ORC, improves safety on harmful-query benchmarks, and largely preserves reasoning utility. These results suggest that safety failures in LRMs can arise from a transient breakdown at the critical transition from understanding to generation.