paper-with-me

홈 › Papers

Ablating Archetypes: The Stability of Archetypal SAEs is an Artifact of Initialization and Metric Design

2026-06-01 · Michał Brzozowski, Neo Christopher Chung arxiv

Dictionary learning with sparse autoencoders (SAEs) produces overcomplete bases from neural network activations that are often interpretable and reduces polysemanticity. However, features from SAEs vary substantially across random seeds -- a problem known as instability. Archetypal SAEs (Fel et al., 2025) were proposed as a general dictionary-learning intervention for more reliable concept extraction, and report more stable dictionaries at the end of training. We demonstrate that the stability claimed by archetypal SAEs is a result of setting identical initialization across multiple runs. Through our analyses, we attempt to clarify two distinct notions in mechanistic interpretability that may be ambiguously used: stability is agreement between two independently trained models, whereas stabilization is the convergence of independently initialized runs toward a common solution. This distinction is critical for mechanistic interpretability of natural language processing (NLP), where feature stability is increasingly used as evidence that SAE features are reusable units of analysis. Experiments from archetypal SAEs share a deterministic k-means decoder initialization, setting inter-run dictionary distance to zero before training begins. When this initialization is removed, the archetypal constraint provides no stabilization advantage in our setting. We further identify a preprocessing-dependent cosine geometry issue that complicates interpretation of endpoint stability metrics. Overall, our study supports the value of studying SAEs within the larger dictionary-learning tradition while showing that stability claims require trajectory diagnostics and initialization ablations.

📄 PDF Abstract BibTeX arXiv:2606.02061

Code (0)

등록된 구현이 없습니다.

Similar Papers 제목 키워드 기반

Archetypal SAE: Adaptive and Stable Dictionary Learning for Concept Extraction in Large Vision Models

2025-02-18 · Thomas Fel, Ekdeep Singh Lubana, Jacob S. Prince, Matthew Kowal 외

Sparse Autoencoders (SAEs) have emerged as a powerful framework for machine learning interpretability, enabling the unsupervised decomposition of model representations into a dictionary of abstract, human-interpretable c…

Dictionary Learning

Finding Archetypal Spaces Using Neural Networks

2019-01-25 · David van Dijk, Daniel Burkhardt, Matthew Amodio, Alex Tong 외

Archetypal analysis is a data decomposition method that describes each observation in a dataset as a convex combination of "pure types" or archetypes. These archetypes represent extrema of a data space in which there is …

Probabilistic Archetypal Analysis

2013-12-29 · Sohan Seth, Manuel J. A. Eugster

Archetypal analysis represents a set of observations as convex combinations of pure patterns, or archetypes. The original geometric formulation of finding archetypes by approximating the convex hull of the observations a…

Coresets for Archetypal Analysis

2019-12-01 · NeurIPS 2019 12 · Sebastian Mair, Ulf Brefeld

Archetypal analysis represents instances as linear mixtures of prototypes (the archetypes) that lie on the boundary of the convex hull of the data. Archetypes are thus often better interpretable than factors computed by…

Quantization

Online Dictionary Learning for Approximate Archetypal Analysis

2018-09-01 · ECCV 2018 9 · Jieru Mei, Chunyu Wang, Wen-Jun Zeng

Archetypal analysis is an unsupervised learning approach which represents data by convex combinations of a set of archetypes. The archetypes generally correspond to the extremal points in the dataset and are learned by r…

Dictionary Learning