Refusal Beyond a Single Direction: A Preliminary Comparison of Diff-in-Means and INLP
This study compares Diff-in-Means (DiM) and INLP for extracting linear directions controlling model refusal, finding INLP's counterfactual flipping highly effective.
Key Findings
Methodology
The paper evaluates five open-source safety-finetuned models, comparing DiM-based methods (activation addition and directional ablation) with INLP techniques (nullspace projection and counterfactual flipping). It extracts refusal-related linear directions from residual streams, assessing their impact on refusal suppression and model performance. The approach involves:• Constructing directions via mean differences of activations;• Training linear classifiers iteratively to identify concept subspaces;• Applying α-parameterized interventions for nullspace projection (α=1) and counterfactual flipping (α=2);• Selecting intervention layers and positions based on effectiveness metrics. Geometric and qualitative analyses complement quantitative results.
Key Results
- INLP's counterfactual flipping matches or exceeds DiM directional ablation in refusal suppression across all models, with nullspace projection performing weaker. Limiting INLP directions (k=0.8) maintains most suppression while reducing perplexity degradation, demonstrating tunability. Geometric visualization shows nullspace projection collapses activations between harmful and harmless clusters, while flipping moves them into the opposite cluster, indicating different encoding mechanisms for absence versus opposite concepts. These findings highlight the richer control offered by INLP's multi-dimensional subspace, enabling more nuanced interventions.
Significance
This work advances understanding of how models encode refusal behaviors, moving beyond single linear directions to multi-dimensional subspaces. The ability to finely tune interventions via INLP's parameters enhances model safety and interpretability. It provides a new framework for concept control, with implications for safer AI deployment, bias mitigation, and explainability. The geometric insights into activation space deepen theoretical understanding of concept representations, informing future research on multi-concept disentanglement and intervention strategies.
Technical Contribution
The paper introduces a systematic comparison between DiM and INLP in the context of refusal control, demonstrating the effectiveness of INLP's multi-dimensional subspaces and α-parameterized interventions. It offers a geometric interpretation of different intervention types, revealing how nullspace projection and flipping operate in activation space. The methodology combines quantitative metrics with PCA visualizations, providing a comprehensive toolkit for concept-based model steering. These contributions expand the state-of-the-art in concept erasure and model interpretability, offering scalable, tunable methods for behavior control.
Novelty
This is the first systematic comparison of DiM and INLP in the context of refusal behavior, revealing that INLP's multi-dimensional subspaces and α-parameterized interventions outperform single-direction methods. The discovery that nullspace projection collapses activations between classes, while flipping moves them into the opposite class, uncovers new insights into how models encode absence versus opposition of concepts. These findings introduce a novel perspective on concept representation and manipulation, broadening the scope of model interpretability research.
Limitations
- The experiments are limited to five models, with potential variability in larger or differently trained models. The layer and position selection process may influence results, and optimal strategies remain to be fully explored. The linearity assumption may oversimplify complex concept encodings, and non-linear methods could further improve control. Additionally, the current interventions focus on a single concept, requiring extension to multi-concept scenarios for broader applicability.
Future Work
Future research will explore multi-layer, multi-position interventions, integrating non-linear techniques to capture complex concept representations. Extending the framework to multiple concepts simultaneously and testing on larger, more diverse models will enhance robustness. Combining these methods with causal analysis could deepen understanding of internal representations. Ultimately, this work aims to develop comprehensive, fine-grained control strategies for safe, interpretable AI systems.
AI Executive Summary
In recent years, the quest for safer and more controllable language models has driven research into understanding their internal mechanisms. Early work by Gao et al. (2020) introduced concept directions as a means to steer models, while Wu et al. (2025) demonstrated the effectiveness of Diff-in-Means (DiM) in identifying influential linear directions. Building on this foundation, Arditi et al. (2024) revealed that refusal behavior—models declining harmful prompts—is mediated by a single linear direction in the residual stream, which can be manipulated via simple averaging techniques. This discovery provided a straightforward method for controlling model refusal, but also raised questions about the underlying complexity of concept representations.
This study advances the field by systematically comparing the traditional DiM approach with INLP, a more flexible concept-erasure method that operates on a tunable subspace of the residual stream. INLP’s key advantage lies in its ability to generate a continuum of interventions through the α parameter, enabling not only removal but also flipping of concepts. The authors evaluate these methods across five open-source safety-finetuned models, including Gemma 2B, Qwen 1.8B, Yi 6B, Llama-2 7B, and Llama-3 8B, measuring their effectiveness in suppressing refusal and maintaining model performance.
Results show that INLP’s counterfactual flipping (α=2) consistently rivals or surpasses DiM directional ablation in refusal suppression, while nullspace projection (α=1) remains weaker. Importantly, restricting INLP to fewer directions (k=0.8) preserves most of the suppression effect with minimal perplexity increase, demonstrating tunability. Geometric analysis reveals that nullspace projection collapses activations between harmful and harmless clusters, whereas flipping moves them into the opposite cluster, indicating different internal encoding mechanisms. These insights deepen our understanding of how models represent concepts and their absence.
The implications are significant: multi-dimensional, tunable interventions offer a more nuanced and flexible approach to model safety and interpretability. This work opens pathways for future research into multi-concept disentanglement, non-linear interventions, and scalable safety strategies, ultimately contributing to the development of more reliable and transparent AI systems. Despite limitations such as model scope and linearity assumptions, the findings mark a substantial step forward in concept-based model control.
Deep Analysis
Background
The evolution of interpretability in language models has focused on identifying influential directions in activation space, with techniques like PCA, LAT, and supervised steering vectors. Early concept erasure methods such as INLP demonstrated the ability to remove biases by projecting onto nullspaces of classifiers. Recent work by Wu et al. (2025) confirmed the effectiveness of simple linear methods like Diff-in-Means for concept detection and control. Arditi et al. (2024) extended this understanding by showing that refusal behavior is mediated by a single linear direction, which can be manipulated to control model responses. However, the complexity of internal representations suggests that multi-dimensional approaches like INLP could offer finer control, but systematic comparisons remain scarce. This research aims to fill this gap by evaluating the efficacy of INLP’s multi-dimensional subspaces versus DiM’s single direction in the context of refusal behavior, providing geometric and empirical insights into their mechanisms.
Core Problem
While single linear directions like DiM are effective and simple, they may oversimplify the internal encoding of refusal behavior, limiting flexibility and nuance. INLP’s multi-dimensional subspaces promise more refined control but lack comprehensive evaluation in this context. The core challenge is to determine whether multi-dimensional interventions can match or outperform single directions in refusal suppression while preserving model performance. Additionally, understanding how different interventions geometrically reshape activation space and encode concepts versus their opposites remains unresolved. These issues hinder the development of precise, scalable, and interpretable safety interventions in language models.
Innovation
This work introduces a systematic comparison between DiM and INLP for concept control, emphasizing INLP’s multi-dimensional subspace and the α-parameterized interventions. It demonstrates that counterfactual flipping (α=2) can effectively invert concepts, providing a tunable mechanism for model behavior. The geometric analysis reveals that nullspace projection collapses activations into an intermediate region, while flipping moves them into the opposite class, uncovering different internal encoding strategies. These insights extend the understanding of concept representations, enabling more flexible and precise control strategies beyond traditional single-direction methods.
Methodology
- �� Extract residual stream activations at various layers and positions for harmful and harmless prompts. • Construct linear directions via mean differences (DiM) and evaluate their effectiveness through activation addition and ablation. • Train multiple linear classifiers iteratively to identify concept subspaces (INLP), stopping when classifiers reach a threshold accuracy. • Generate interventions by applying nullspace projection (α=1) to erase concepts, and counterfactual flipping (α=2) by reflecting activations across the subspace. • Select optimal layers and positions based on suppression metrics, ensuring fair comparison across methods. • Visualize activation space changes with PCA, analyze the geometric distribution, and evaluate the semantic fidelity of transformed activations.
Experiments
The experimental setup involves five open-source models, with datasets including harmful and harmless instruction sets from Arditi et al. (2024). Metrics include refusal rates (substring matching, LlamaGuard, LLM judge), perplexity, and accuracy on MMLU and ARC. Interventions are applied at selected layers and positions, with k-values ranging from full (n) to partial (0.8). The evaluation assesses refusal suppression effectiveness, model performance preservation, and geometric changes in activation space. Ablation studies examine the impact of direction number and intervention parameters, ensuring robustness and generality of findings.
Results
The results confirm that INLP’s counterfactual flipping achieves refusal suppression comparable to DiM’s directional ablation, with some models exceeding DiM. Nullspace projection shows weaker and inconsistent effects. Limiting the number of directions (k=0.8) maintains most suppression while reducing perplexity degradation. Geometrically, nullspace projection collapses activations into an intermediate region, while flipping moves them into the opposite class, indicating different internal representations. These findings validate the richer control capacity of INLP and highlight the importance of multi-dimensional interventions for nuanced model behavior regulation.
Applications
The methods can be employed to improve safety in dialogue systems by fine-tuning refusal behaviors, reducing harmful content generation. They also facilitate interpretability by revealing how concepts are encoded and manipulated internally. The tunability of INLP interventions supports adaptive safety measures, enabling dynamic control in real-world deployment. Long-term, these techniques could underpin scalable frameworks for multi-concept disentanglement and multi-task safety interventions, fostering more transparent and reliable AI systems.
Limitations & Outlook
Current validation is limited to small-scale models; scalability to larger architectures remains untested. The linear assumption may oversimplify complex internal representations, necessitating non-linear extensions. Layer and position selection strategies influence results, requiring further optimization. Additionally, the focus on a single concept (refusal) limits generalization; multi-concept scenarios pose additional challenges. Future work should address these limitations to develop more comprehensive and robust control frameworks.
Plain Language Accessible to non-experts
想象你在厨房做饭,每个调料代表一种特定的味道。有时候,为了让菜更健康或更安全,你会用不同的调料调节味道。单一调料就像模型中的线性方向,调整它可以让模型变得更“温和”或“激烈”。但如果用多种调料同时调节,就像用多维空间调控,可以更精细地控制菜的味道。比如,你可以同时减少盐和糖,或者让菜变得更辣更甜。研究发现,有些调料调节得好,不仅能阻止不良内容,还能保持菜的原味,就像模型在拒绝有害内容的同时还能正常工作。这种多维调节让厨房(模型)变得更灵活,也更容易掌控。
Abstract
Arditi et al. (2024) has shown that refusal in safety fine-tuned chat models is mediated by a single linear direction in the residual stream, recoverable by a difference-in-means (DiM) of harmful and harmless activations. We compare DiM-based interventions (activation addition and directional ablation) with two interventions derived from Iterative Nullspace Projection (INLP) -- nullspace projection and counterfactual flipping -- on five open-weight chat models, asking whether INLP can match DiM at steering refusal and whether its richer parameterisation yields more tweakable interventions. INLP counterfactual flipping is competitive with DiM directional ablation on refusal suppression, while nullspace projection is consistently weaker. Restricting INLP to the leading directions of the extracted subspace preserves most of the suppression effect at near-baseline perplexity, giving a tunable capability. Geometrically, the two INLP interventions land in qualitatively different regions of activation space: nullspace projection collapses transformed activations \emph{between} the harmful and harmless clusters, while counterfactual flipping moves them into the opposite cluster, suggesting that the model encodes the absence of a concept differently from its opposite -- an intriguing distinction that warrants further investigation in future work.