English

SALVE: Sparse Autoencoder-Latent Vector Editing for Mechanistic Control of Neural Networks

Machine Learning 2026-03-10 v2 Artificial Intelligence Computer Vision and Pattern Recognition

Abstract

Deep neural networks achieve impressive performance but remain difficult to interpret and control. We present SALVE (Sparse Autoencoder-Latent Vector Editing), a unified "discover, validate, and control" framework that bridges mechanistic interpretability and model editing. Using an 1\ell_1-regularized autoencoder, we learn a sparse, model-native feature basis without supervision. We validate these features with Grad-FAM, a feature-level saliency mapping method that visually grounds latent features in input data. Leveraging the autoencoder's structure, we perform precise and permanent weight-space interventions, enabling continuous modulation of both class-defining and cross-class features. We further derive a critical suppression threshold, αcrit\alpha_{crit}, quantifying each class's reliance on its dominant feature, supporting fine-grained robustness diagnostics. Our approach is validated on both convolutional (ResNet-18) and transformer-based (ViT-B/16) models, demonstrating consistent, interpretable control over their behavior. This work contributes a principled methodology for turning feature discovery into actionable model edits, advancing the development of transparent and controllable AI systems.

Keywords

Cite

@article{arxiv.2512.15938,
  title  = {SALVE: Sparse Autoencoder-Latent Vector Editing for Mechanistic Control of Neural Networks},
  author = {Vegard Flovik},
  journal= {arXiv preprint arXiv:2512.15938},
  year   = {2026}
}

Comments

Accepted to ICLR 2026, Trustworthy AI Workshop