Universal Sparse Autoencoders: Interpretable Cross-Model Concept Alignment

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Thasarathan, Harrish, Forsyth, Julian, Fel, Thomas, Kowal, Matthew, Derpanis, Konstantinos G.
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866910058403594240
author Thasarathan, Harrish
Forsyth, Julian
Fel, Thomas
Kowal, Matthew
Derpanis, Konstantinos G.
author_facet Thasarathan, Harrish
Forsyth, Julian
Fel, Thomas
Kowal, Matthew
Derpanis, Konstantinos G.
contents We present Universal Sparse Autoencoders (USAEs), a framework for uncovering and aligning interpretable concepts spanning multiple pretrained deep neural networks. Unlike existing concept-based interpretability methods, which focus on a single model, USAEs jointly learn a universal concept space that can reconstruct and interpret the internal activations of multiple models at once. Our core insight is to train a single, overcomplete sparse autoencoder (SAE) that ingests activations from any model and decodes them to approximate the activations of any other model under consideration. By optimizing a shared objective, the learned dictionary captures common factors of variation-concepts-across different tasks, architectures, and datasets. We show that USAEs discover semantically coherent and important universal concepts across vision models; ranging from low-level features (e.g., colors and textures) to higher-level structures (e.g., parts and objects). Overall, USAEs provide a powerful new method for interpretable cross-model analysis and offers novel applications, such as coordinated activation maximization, that open avenues for deeper insights in multi-model AI systems
format Preprint
id arxiv_https___arxiv_org_abs_2502_03714
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Universal Sparse Autoencoders: Interpretable Cross-Model Concept Alignment
Thasarathan, Harrish
Forsyth, Julian
Fel, Thomas
Kowal, Matthew
Derpanis, Konstantinos G.
Computer Vision and Pattern Recognition
Machine Learning
We present Universal Sparse Autoencoders (USAEs), a framework for uncovering and aligning interpretable concepts spanning multiple pretrained deep neural networks. Unlike existing concept-based interpretability methods, which focus on a single model, USAEs jointly learn a universal concept space that can reconstruct and interpret the internal activations of multiple models at once. Our core insight is to train a single, overcomplete sparse autoencoder (SAE) that ingests activations from any model and decodes them to approximate the activations of any other model under consideration. By optimizing a shared objective, the learned dictionary captures common factors of variation-concepts-across different tasks, architectures, and datasets. We show that USAEs discover semantically coherent and important universal concepts across vision models; ranging from low-level features (e.g., colors and textures) to higher-level structures (e.g., parts and objects). Overall, USAEs provide a powerful new method for interpretable cross-model analysis and offers novel applications, such as coordinated activation maximization, that open avenues for deeper insights in multi-model AI systems
title Universal Sparse Autoencoders: Interpretable Cross-Model Concept Alignment
topic Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2502.03714