Dynamically Scaled Activation Steering
By: Alex Ferrando , Xavier Suau , Jordi Gonzàlez and more
Potential Business Impact:
Makes AI helpful without being bad.
Activation steering has emerged as a powerful method for guiding the behavior of generative models towards desired outcomes such as toxicity mitigation. However, most existing methods apply interventions uniformly across all inputs, degrading model performance when steering is unnecessary. We introduce Dynamically Scaled Activation Steering (DSAS), a method-agnostic steering framework that decouples when to steer from how to steer. DSAS adaptively modulates the strength of existing steering transformations across layers and inputs, intervening strongly only when undesired behavior is detected. At generation time, DSAS computes context-dependent scaling factors that selectively adjust the strength of any steering method. We also show how DSAS can be jointly optimized end-to-end together with the steering function. When combined with existing steering methods, DSAS consistently improves the Pareto front with respect to steering alone, achieving a better trade-off between toxicity mitigation and utility preservation. We further demonstrate DSAS's generality by applying it to a text-to-image diffusion model, showing how adaptive steering allows the modulation of specific concepts. Finally, DSAS introduces minimal computational overhead while improving interpretability, pinpointing which tokens require steering and by how much.
Similar Papers
DSO: Direct Steering Optimization for Bias Mitigation
Machine Learning (CS)
Fixes AI bias, making it fairer for everyone.
Steering When Necessary: Flexible Steering Large Language Models with Backtracking
Computation and Language
Makes AI say helpful and honest things.
HyperSteer: Activation Steering at Scale with Hypernetworks
Computation and Language
Teaches computers to write what you ask.