arXiv Machine Learning

SharedSAE: One Feature Dictionary Across Language Models

SharedSAE demonstrates that a single sparse autoencoder can replace multiple model‑specific SAEs by using a shared dictionary with model‑specific encoder‑decoder pairs. It preserves activation magnitudes, normalizes only selection scores, and supports single‑model inference via model dropout. Trained on four 1B‑scale language models, SharedSAE retains 96.6% of the mean explained variance of dedicated SAEs, shows higher cross‑model latent correlations, and allows efficient adaptation of new models to the shared latent space.

Hugging Face Trending Papers
Aug 13

SAEVerbalizer: Generating Explanations for Sparse Autoencoder Features via Representation Verbalization

Sparse autoencoders (SAEs) are proposed to extract numerous features from large language model (LLM) representations, yet explaining these features still relies primarily on external observation. This reliance leads to superficial explanations inferred from observed model behavior and computational inefficiency from collecting such behavioral evidence at scale.

arXiv Machine Learning
Oct 16

Diminishing Returns of Data in Sparse Autoencoders: Functional Behavior does not Guarantee Feature Reproducibility

Sparse autoencoders (SAEs) are usually trained on hundreds of millions to billions of activation tokens, but this study shows that the amount of data needed depends on the evaluation metric. Training four SAE architectures on three language models with token budgets from 5 M to 500 M revealed that reconstruction quality can reach near‑full‑data performance with far fewer tokens, while one‑to‑one feature reproducibility continues to improve as more data is used. The research also found that random sampling preserves activation distributions better for reconstruction, whereas diversity‑based sampling can enhance feature interpretability, and provided a theoretical framework distinguishing statistical consistency from the stricter conditions needed for exact feature recovery.

By Sambit Ghosh, Madelyn Mathai, R. Venkatesh Babu, Chirag Agarwal
arXiv AI
Oct 7

Inference and learning in sparse autoencoders as natural gradient flow

arXiv:2610.07389v1 Announce Type: cross Abstract: Sparse autoencoders are widely used to uncover interpretable features in neural networks, yet reliable recovery remains difficult when features overl...

By Hadi Vafaii, Tejas Rao, David Chanin, Thomas Fel, Jacob L. Yates, Bruno Olshausen, David Klindt, Dileep George, Miguel L\'azaro-Gredilla
Hugging Face Trending Papers
Jul 9

Cross-seed explainability using Procrustes-conditioned Joint End-to-end Top-K Sparse Autoencoders

We present a Procrustes-conditioned Joint End-to-end Top-K Sparse Autoencoder (SAE) for extracting cross-seed universal features from independently trained BERT models. Cross-seed feature universality is a fundamental challenge in mechanistic interpretability: because dictionary learning is non-convex, independently trained networks learn misaligned feature spaces, so apparently identical features may differ by random initialization.

arXiv AI
Jul 17

In-Place Tokenizer Expansion for Pre-trained LLMs

arXiv:2607. 15232v1 Announce Type: cross Abstract: A tokenizer fixed at the start of pre-training allocates vocabulary in proportion to the pre-training corpus, reflecting the deployment priorities at that time.

By Jimmy T. H. Smith, Tarek Dakhran, Alberto Cabrera, Simon S. Lee, Paul Pak, Aditya Tadimeti, Tim Seyde, Maxime Labonne, Alexander Amini, Mathias Lechner
arXiv Computation and Language
Aug 28

Double Trouble: Bilingual Pretraining Leaves Language-Conditioned Effects in Shared-Language Representations

The study compares an English-only and a bilingual decoder-only model, each 310 M parameters, trained on eight diverse languages while controlling for English exposure, compute, and document overlap. After aligning on shared English vocabulary, the authors find that token embeddings appear similar, but the deeper hidden states used for prediction diverge across models. This hidden‑state mismatch grows through middle transformer layers and persists despite controls, indicating that contextual processing differs between the models. "whyItMatters":"The findings show that embedding alignment can conceal significant internal representation differences, which is crucial for any downstream work that assumes aligned multilingual models are interchangeable."

By Anjishnu Mukherjee, Ziwei Zhu, Antonios Anastasopoulos
arXiv Machine Learning
Sep 3

Persistent Sparse Autoencoders: Learning Feature-Specific Timescales in Language Model Representations

Persistent Sparse Autoencoders (Persistent SAEs) extend standard sparse autoencoders by learning a persistence coefficient for each feature, enabling the model to capture feature‑specific timescales from reconstruction alone. The study shows that these persistent features maintain competitive reconstruction quality while distinguishing between short‑timescale, locally interpretable features and long‑timescale, context‑accumulating features. In a prompt‑injection monitoring case study, slow features were found to preserve injection‑related signals and remain causally effective over long contexts.

By Haoyan Luo, Mateo Espinosa Zarlenga, Mateja Jamnik