English

MidSteer: Optimal Affine Framework for Steering Generative Models

Machine Learning 2026-05-08 v1 Artificial Intelligence

Abstract

Steering intermediate representations has emerged as a powerful strategy for controlling generative models, particularly in post-deployment alignment and safety settings. However, despite its empirical success, it currently lacks a comprehensive theoretical framework. In this paper, we bridge this gap by formalizing the theory of concept steering. First, we establish a link between steering and affine concept erasure, proving that the standard approach for removing unwanted behaviors is a special case of LEACE (a closed-form method for affine erasure). Next, we formulate a principled theoretical framework for concept switching, LEACE-Switch, and characterize the assumptions under which it provides an optimal affine solution. Building on this analysis, we then introduce MidSteer (Minimal Disturbance concept Steering), a more general affine framework for concept manipulation that relaxes these assumptions and enables directed, minimal-disturbance transformations. We demonstrate that MidSteer performs favorably across a range of tasks, modalities, and architectures, including vision diffusion models and large language models.

Cite

@article{arxiv.2605.05220,
  title  = {MidSteer: Optimal Affine Framework for Steering Generative Models},
  author = {Tatiana Gaintseva and Andrew Stepanov and Ziquan Liu and Martin Benning and Gregory Slabaugh and Jiankang Deng and Ismail Elezi},
  journal= {arXiv preprint arXiv:2605.05220},
  year   = {2026}
}
R2 v1 2026-07-01T12:53:20.411Z