Cross-model Transfer of Matryoshka Sparse Autoencoders
Replicating and extending Matryoshka SAEs across model families
Image will load when scrolled into view
Mechanistic InterpretabilitySparse AutoencodersPyTorchReplication
About the Project
A replication and extension of Matryoshka Sparse Autoencoders. I reproduced the paper's core result on the synthetic hierarchy and at scale on Gemma-2-2B, confirming that Matryoshka SAEs avoid the feature-absorption failure mode that vanilla SAEs exhibit.
I then transferred the setup to other model families — Llama-3.1-8B and Gemma-3-12B — to test how well the result generalizes beyond the original setting, and documented the full replication in a public writeup on LessWrong.
This project is part of my broader research direction on the generalizability of mechanistic interpretability methods: understanding when results established on one model transfer to others.
Project Details
StatusPublished Writeup (LessWrong)
Role
Independent Researcher
Stack
PyTorch
Sparse Autoencoders
Gemma-2-2B / Llama-3.1-8B / Gemma-3-12B