Papers
arxiv:2602.02600

Step-Wise Refusal Dynamics in Autoregressive and Diffusion Language Models

Published on Feb 1
Authors:
,
,
,

Abstract

Diffusion language models (DLMs) have recently emerged as a promising alternative to autoregressive (AR) models, offering parallel decoding and controllable sampling dynamics while achieving competitive generation quality at scale. Despite this progress, the role of sampling mechanisms in shaping refusal behavior and jailbreak robustness remains poorly understood. In this work, we present a fundamental analytical framework for step-wise refusal dynamics, enabling comparison between AR and diffusion sampling. Our analysis reveals that the sampling strategy itself plays a central role in safety behavior, as a factor distinct from the underlying learned representations. Motivated by this analysis, we introduce the Step-Wise Refusal Internal Dynamics (SRI) signal, which supports interpretability and improved safety for both AR and DLMs. We demonstrate that the geometric structure of SRI captures internal recovery dynamics, and identifies anomalous behavior in harmful generations as cases of incomplete internal recovery that are not observable at the text level. This structure enables lightweight inference-time detectors that generalize to unseen attacks while matching or outperforming existing defenses with over 100times lower inference overhead.

Community

Diffusion language models (DLMs) offer parallel decoding and controllable sampling, but how sampling affects refusal behavior and jailbreak robustness is still poorly understood. We introduce a step-wise analysis framework and the Step-Wise Refusal Internal Dynamics (SRI) signal, showing that sampling strategy itself plays a central role in safety behavior and reveals incomplete internal recovery patterns that are invisible at the text level. We release an open-source SRI toolkit for safety interpretability, enabling lightweight jailbreak detection that generalizes to unseen attacks with low inference overhead.

๐Ÿ”— Project page: https://elironrahimi.github.io/sri-signal/

๐Ÿ’ป Code: https://github.com/ElironRahimi/sri-signal

image

Sign up or log in to comment

Models citing this paper 0

No model linking this paper

Cite arxiv.org/abs/2602.02600 in a model README.md to link it from this page.

Datasets citing this paper 0

No dataset linking this paper

Cite arxiv.org/abs/2602.02600 in a dataset README.md to link it from this page.

Spaces citing this paper 0

No Space linking this paper

Cite arxiv.org/abs/2602.02600 in a Space README.md to link it from this page.

Collections including this paper 0

No Collection including this paper

Add this paper to a collection to link it from this page.