English

DSPA: Dynamic SAE Steering for Data-Efficient Preference Alignment

Machine Learning 2026-03-24 v1 Artificial Intelligence Computation and Language

Abstract

Preference alignment is usually achieved by weight-updating training on preference data, which adds substantial alignment-stage compute and provides limited mechanistic visibility. We propose Dynamic SAE Steering for Preference Alignment (DSPA), an inference-time method that makes sparse autoencoder (SAE) steering prompt-conditional. From preference triples, DSPA computes a conditional-difference map linking prompt features to generation-control features; during decoding, it modifies only token-active latents, without base-model weight updates. Across Gemma-2-2B/9B and Qwen3-8B, DSPA improves MT-Bench and is competitive on AlpacaEval while preserving multiple-choice accuracy. Under restricted preference data, DSPA remains robust and can rival the two-stage RAHF-SCIT pipeline while requiring up to 4.47×4.47\times fewer alignment-stage FLOPs. Finally, we audit the SAE features DSPA modifies, finding that preference directions are dominated by discourse and stylistic signals, and provide theory clarifying the conditional-difference map estimate and when top-kk ablation is principled.

Keywords

Cite

@article{arxiv.2603.21461,
  title  = {DSPA: Dynamic SAE Steering for Data-Efficient Preference Alignment},
  author = {James Wedgwood and Aashiq Muhamed and Mona T. Diab and Virginia Smith},
  journal= {arXiv preprint arXiv:2603.21461},
  year   = {2026}
}
R2 v1 2026-07-01T11:32:33.414Z