Interpretable and Testable Vision Features via Sparse Autoencoders

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Stevens, Samuel, Chao, Wei-Lun, Berger-Wolf, Tanya, Su, Yu
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866915631508488192
author Stevens, Samuel
Chao, Wei-Lun
Berger-Wolf, Tanya
Su, Yu
author_facet Stevens, Samuel
Chao, Wei-Lun
Berger-Wolf, Tanya
Su, Yu
contents To truly understand vision models, we must not only interpret their learned features but also validate these interpretations through controlled experiments. While earlier work offers either rich semantics or direct control, few post-hoc tools supply both in a single, model-agnostic procedure. We use sparse autoencoders (SAEs) to bridge this gap; each sparse feature comes with real-image exemplars that reveal its meaning and a decoding vector that can be manipulated to probe its influence on downstream task behavior. By applying our method to widely-used pre-trained vision models, we reveal meaningful differences in the semantic abstractions learned by different pre-training objectives. We then show that a single SAE trained on frozen ViT activations supports patch-level causal edits across tasks (classification and segmentation) all without retraining the ViT or task heads. These qualitative, falsifiable demonstrations position SAEs as a practical bridge between concept discovery and causal probing of vision models. We provide code, demos and models on our project website: https://osu-nlp-group.github.io/saev.
format Preprint
id arxiv_https___arxiv_org_abs_2502_06755
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Interpretable and Testable Vision Features via Sparse Autoencoders
Stevens, Samuel
Chao, Wei-Lun
Berger-Wolf, Tanya
Su, Yu
Computer Vision and Pattern Recognition
To truly understand vision models, we must not only interpret their learned features but also validate these interpretations through controlled experiments. While earlier work offers either rich semantics or direct control, few post-hoc tools supply both in a single, model-agnostic procedure. We use sparse autoencoders (SAEs) to bridge this gap; each sparse feature comes with real-image exemplars that reveal its meaning and a decoding vector that can be manipulated to probe its influence on downstream task behavior. By applying our method to widely-used pre-trained vision models, we reveal meaningful differences in the semantic abstractions learned by different pre-training objectives. We then show that a single SAE trained on frozen ViT activations supports patch-level causal edits across tasks (classification and segmentation) all without retraining the ViT or task heads. These qualitative, falsifiable demonstrations position SAEs as a practical bridge between concept discovery and causal probing of vision models. We provide code, demos and models on our project website: https://osu-nlp-group.github.io/saev.
title Interpretable and Testable Vision Features via Sparse Autoencoders
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2502.06755