SharedSAE: One Feature Dictionary Across Language Models
Organizations: MBZUAI · Sorbonne Université · Tohoku University · RIKEN AIP
Abstract
Sparse autoencoders (SAEs) are widely used to interpret language model activations, but SAE training and latent labelling are typically repeated for every model. Here, we show that a single shared SAE can replace a collection of dedicated per-model SAEs. Our method, SharedSAE, combines a shared dictionary with model-specific encoder-decoder pairs. Unlike the closest prior method, which discards activation magnitudes and requires all models at inference, SharedSAE instead normalizes only selection scores, preserving magnitudes, and uses model dropout for single-model inference. We train SharedSAE on four 1B-scale base language models spanning distinct families and tokenizers. Despite sharing its latents across models, SharedSAE retains 96.6% of dedicated SAEs' mean explained variance; its latent activations exhibit cross-model correlations 1.8 times as high as separate SAEs aligned post-hoc, and its latent descriptions transfer across models. After the dictionary is frozen, new models can be efficiently adapted to it, achieving near-dedicated-SAE reconstruction quality while reusing the shared latent descriptions.
Figures & tables
| Qwen3 | Llama-3.2 | OLMo-2 | Gemma-3 | Mean | Cross-FVE | |
| Reconstruction (FVE) | ||||||
| Dedicated (ceiling) | 0.838 | 0.807 | 0.812 | 0.868 | 0.831 | - |
| SharedSAE , joint TopK | 0.798 | 0.785 | 0.787 | 0.825 | 0.799 | 0.749 |
| SharedSAE , solo TopK | 0.804 | 0.794 | 0.795 | 0.821 | 0.803 | - |
| USAE | 0.760 | 0.743 | 0.750 | 0.787 | 0.760 | 0.640 |
| SPARC, joint TopK | 0.769 | 0.765 | 0.763 | 0.631 | 0.732 | 0.663 |
| decoded into model | ||||
|---|---|---|---|---|
| code from | Qwen3 | Llama-3.2 | OLMo-2 | Gemma-3 |
| Qwen3 | 0.798 | 0.738 | 0.738 | 0.782 |
| Llama-3.2 | 0.746 | 0.785 | 0.738 | 0.786 |
| OLMo-2 | 0.741 | 0.733 | 0.787 | 0.780 |
| Gemma-3 | 0.739 | 0.733 | 0.736 | 0.825 |
| Dedicated | 0.815 | 0.783 | 0.795 | 0.825 |
| Clarity | Resp. | Purity | Faith. | Retained | |
| Within-stream clear labels (clarity ) | |||||
| SharedSAE shared vote | 0.979 | 0.642 | 0.463 | 0.281 | 10.6% |
| SharedSAE per-stream | 0.976 | 0.629 | 0.438 | 0.217 | 21.7% |
| SAE Qwen3 | 0.975 | 0.584 | 0.416 | 0.333 | 7.4% |
| SAE Llama-3.2 | 0.962 | 0.558 | 0.431 | 0.403 | 9.4% |
| SAE OLMo-2 | 0.954 | 0.583 | 0.373 | 0.320 | 4.1% |
| Seed TopK | Solo | ||||
| Held-out model | FVE | /co-tr. | /ded. | FVE | /ded. |
| Leave-one-out (three-model seed) | |||||
| Qwen3 | 0.769 | 0.997 | 0.944 | 0.793 | 0.972 |
| Llama-3.2 | 0.753 | 0.990 | 0.962 | 0.769 | 0.982 |
| OLMo-2 | 0.751 | 0.984 | 0.945 | 0.776 | 0.977 |
| Gemma-3 | 0.787 | 0.979 | 0.954 | 0.812 | 0.984 |
| corr. | top-50 | choice | FVE/ded. | |
| Co-trained SharedSAE (4 models) | 0.785 | 31.5 | 59.0 | 0.999 |
| Adapted, leave-one-out mean | 0.785 | 31.5 | 66.0 | 0.979 |
| Held-out Gemma fold, attach routes | ||||
| adapted encoder-decoder (ours) | 0.785 | 32.0 | 60.3 | 0.984 |
| linear alignment (Atlas-style) | 0.722 | 29.5 | 57.5 | 0.692 |
| decoder-only attach | - | - | - | 0.795 |
| Vote input (10 epochs, no dropout) | |||
|---|---|---|---|
| Share | Vote shares (Q/L/O/G) | ||
| raw votes | 0.919 | 0.07/0.00/0.00/0.93 | |
| RMS-cal. inputs | 0.925 | 0.24/0.40/0.30/0.07 | |
| vote norm (raw) | 0.930 | 0.25/0.25/0.25/0.25 | |
| Vote-source dropout dose (vote norm) | |||
| Share | Cross-FVE | Gemma solo | |
| CE-recovered | ||||||||
|---|---|---|---|---|---|---|---|---|
| Qwen3 | Llama | OLMo-2 | Gemma | Qwen3 | Llama | OLMo-2 | Gemma | |
| Dedicated (8-epoch) | 0.966 | 0.922 | 0.965 | 0.943 | 0.243 | 0.484 | 0.309 | 0.423 |
| SharedSAE (settled) | 0.958 | 0.933 | 0.970 | 0.968 | 0.309 | 0.438 | 0.279 | 0.293 |
| Qwen3 | Llama-3.2 | OLMo-2 | Gemma-3 | Mean | |
| SharedSAE (mean consensus size of 128) | |||||
| all own 128 latents | 0.804 | 0.794 | 0.795 | 0.821 | 0.803 |
| consensus latents only | 0.736 | 0.721 | 0.727 | 0.716 | 0.725 |
| own strongest latents | 0.758 | 0.740 | 0.746 | 0.739 | 0.746 |
| latents outside the consensus | -1.139 | -0.010 | 0.032 | -8.247 | -2.341 |
| USAE, 32 epochs (mean consensus size of 128) | |||||
| 100k window, latents alive 10 times | ||
|---|---|---|
| Pairwise (of 50) | All-four | |
| System | median (span) | (of 50) |
| SharedSAE | 37 (37-38) | 26 |
| Matched dedicated | 15 (11-17) | - |
| USAE (8 epochs) | 0 (0-0) | - |
| 300k window, latents with 50 firings ( ) | ||
| Vocabulary classes ( usage, 100k-token window) | ||
| Class | Latents | Med. corr. |
| Consensus (all four) | 10,776 | 0.91 |
| Partial (two or three) | 1,650 | 0.85 |
| Specific (one model) | 1,144 | 0.84 |
| Below everywhere | 19,190 | 0.82 |
| Stability of the 20k-window classes at 100k | ||
Appendix figures & tables20 assets
Supplementary material from the paper’s appendix.
Appendix
| seed | corr. | top-50 | pairwise | -way | FVE/conv. |
|---|---|---|---|---|---|
| without Qwen3-1.7B | 0.778 | 32.0 | 83.3 | 71.8 (3-way) | 0.945 |
| without Llama-3.2-1B | 0.784 | 32.0 | 85.9 | 75.0 (3-way) | 0.942 |
| without OLMo-2-1B | 0.791 | 32.0 | 85.8 | 74.8 (3-way) | 0.935 |
| without Gemma-3-1B | 0.792 | 32.0 | 80.9 | 68.5 (3-way) | 0.956 |
| full SharedSAE | 0.797 | 31.5 | 78.0 | 59.0 (4-way) | 0.967 |
| under the solo TopK | Qwen3 | Llama | OLMo | Gemma |
|---|---|---|---|---|
| SPARC, vote share | 0.245 | 0.276 | 0.274 | 0.205 |
| SPARC, solo joint | 0.711 | 0.625 | 0.592 | 0.635 |
| SPARC, latents used | 22,933 | 30,635 | 31,097 | 7,719 |
| SharedSAE , latents used | 24,328 | 24,776 | 24,532 | 24,964 |
| Four-way intersection of the solo selections, of 128 | ||||
| SPARC 49.2 SharedSAE 58.9 | ||||
| model | ded. | joint | share | solo | solo/ded | sink % |
|---|---|---|---|---|---|---|
| Qwen3 | 0.807 | 0.768 | 0.952 | 0.767 | 0.951 | 0.015 |
| Llama-3.2 | 0.782 | 0.738 | 0.944 | 0.753 | 0.963 | 0.000 |
| OLMo-2 | 0.712 | 0.653 | 0.917 | 0.681 | 0.957 | 0.000 |
| Gemma-3 | 0.819 | 0.782 | 0.954 | 0.751 | 0.917 | 0.000 |
| class | @20k | @100k | corr |
|---|---|---|---|
| consensus (all four) | 9,987 | 10,776 | 0.91 |
| partial (2-3 models) | 1,877 | 1,650 | 0.85 |
| specific (one model) | 1,397 | 1,144 | 0.84 |
| joint-only | 17 | 8 | 0.72 |
| below everywhere | 19,490 | 19,190 | 0.82 |
| latent | fires/100k | Qwen3-1.7B | Llama-3.2-1B | OLMo-2-1B | Gemma-3-1B |
|---|---|---|---|---|---|
| 8319 | 50 | free | free | free | free |
| 29780 | 52 | i | i | i | i |
| 11723 | 75 | ,’’, ," | ,", ,’’ | ,’’, ," | ,", ,’’ |
| 16628 | 22 | United | United | United | United |
| 30588 | 20 | United | United | United | United |
| 10904 | 56 | bar, Bar | bar, Bar, -bar | bar, Bar | bar, Bar |
| Token level: ten strongest positions decoded in | |||
|---|---|---|---|
| each model’s own tokenizer (medians over 6 pairs) | |||
| SharedSAE | Matched ded. | ||
| Modal decoded-token agreement | 0.63 | 0.45 | |
| Decoded token-set Jaccard | 0.50 | 0.19 | |
| Concept level: AP retention of each model’s | |||
| five most selective detectors | |||
| Latent | Per-stream Qwen3 | Per-stream OLMo-2 | Per-stream Gemma-3 | Per-stream Llama-3.2 | Shared Vote |
|---|---|---|---|---|---|
| 16291 | word ‘in’ or ‘of’ | word ‘of’ or ‘to’ or punctuation marks | word ‘of’ or ‘in’ | word ‘of’ or ‘is’ or ‘in’ | prepositions and articles |
| 2317 | four-year term or similar duration | four-year term or similar duration | year term election context | four-year term or similar duration | four-year term or similar duration |
| 6418 | using or a before handheld gps | handheld gps usage | word-initial ‘a’ or ‘using’ | word-initial ‘a’ or ‘the’ | word-initial ‘a’ or ‘using’ |
| 2727 | word-initial uppercase fragment | word-initial uppercase fragment | word-initial uppercase fragment | word-initial uppercase fragment | word-initial uppercase fragment |
| 14390 | possessive form | possessive form | word-initial uppercase fragment | possessive ‘s or whose | possessive form |
| 22504 | should | should | should fragment | should | should |
| Dedicated baseline | ||
|---|---|---|
| SAE Gemma-3 | -1.154 | 0.259 |
| SAE Llama-3.2 | -1.002 | 0.327 |
| SAE OLMo-2 | -0.210 | 0.840 |
| SAE Qwen3 | -0.420 | 0.679 |
| #1537 - positive appraisal (English and French probes) |
|---|
| Qwen3 |
| (r1, in 4/4 top-10) …ists. |The spine and ribs ofgnathodon saturator| It’s a very nice display. And while the amount … † |
| (r5, in 4/4 top-10) …hem in a history that is both analytical and narrative…A superb book that will be a standard m … † |
| (r10, in 4/4 top-10) …inifa should get an Israel Business award! What a Kiddushem! Nice to hear about the positive asp … † |
| Llama-3.2 |
| (r1, in 4/4 top-10) …ists. |The spine and ribs ofgnathodon saturator| It’s a very nice display. And while the amount … † |
| #7393 - general negative affect (moral-wrongdoing and distress probes) |
|---|
| Qwen3 |
| (r1, in 4/4 top-10) But the power it would need to guarantee that no one is poor would be so great it could crush the natural rights and liberty of individuals. |
| (r5, in 3/4 top-10) Smederevo citizens, frightened and hurt by a harsh occupation, and on April 6, by the bombing of the city, too, withdrew to their homes and only came out on the streets when needed: to finish their jobs and to procure food. |
| (r10, in 3/4 top-10) Those disciples of Christ who have experienced their status to any degree have become keenly aware of the intimate relationship they have with Marya relationship that at the moment is perhaps closer than that with Christ. For it is Mary who brings us to Christ. |
| Llama-3.2 |
| (r1, in 4/4 top-10) He concludes-echoing historian Rick Atkinson’s excellent recent account of the campaign, The Day of Battle-that despite its terrible cost, the fight in Italy played a decisive role in defeating Germany. |
| #27599 - epistemic hedging: conjecture and possibility |
|---|
| Qwen3 |
| (r1, in 4/4 top-10) All three were calibrated using the same unit of measurement, the cubit; the Egyptians broke the cubit into smaller units, which allowed them to keep remarkably accurate records, perhaps more accurate than would have been warranted for the purposes of merely agriculture and taxation. |
| (r5, in 4/4 top-10) But, Namibian elephants are of great interest to tourists, and this may be the key to their salvation in this country that has been described as the land that God created in anger. |
| (r10, in 2/4 top-10) But, Namibian elephants are of great interest to tourists, and this may be the key to their salvation in this country that has been described as the land that God created in anger. |
| Llama-3.2 |
| (r1, in 4/4 top-10) The American Blacks played a crucial role in the very beginning of the Sydney steel plant, but never stayed around to reap any benefits. |
| #23065 - conditional mood in hypothetical clauses |
|---|
| Qwen3 |
| (r1, in 2/4 top-10) Simulations are supporting the education system to better equip students with practical knowledge, so when it comes to tackling real-life situations in their future careers they can perform better. Taking into consideration the benefits offered by simulations in the education process, it would be a shame to miss out on them. |
| (r5, in 3/4 top-10) The feature was named by Blanchard and Davis,66 who called it the Bone Springs arch. It would seem from their paper that they considered the feature to be anticlinal, and to have a similar, opposing flank to the north. |
| (r10, in 3/4 top-10) Thoth - Link With Sirius Sirius Mythology - Zep Tepi Star - Star Tetrahedron Thoth The Scribe used sacred geometry - geometry of the stars to write our program placing coded messages in the heavens for you to find. Creators - Guardians - ’God/Dog’ - Dog Star - would live beyond the stars in a higher realm of consciousness. |
| Llama-3.2 |
| (r1, in 3/4 top-10) Didn’t scientific research recently concluded that when in the dark (can’t be seen!), people are more likely to steal? It would seem that we all agree on the moral high grounds but consistently fail when faced such tests. |
| #20341 - programming operations, in code and in prose about code |
|---|
| Qwen3 |
| (r1, in 4/4 top-10) …ned a new file via my IDE and turned it into a list via read lines . Why? Because I want to † |
| (r5, in 4/4 top-10) …e opened a new file via my IDE and turned it into a list via read lines. Why? Because I want † |
| (r10, in 3/4 top-10) When I use [’Some name__________1.5 6.5 6.5\n’, ’Another name________6.3 1.2 1.5\n’] 1.5 3.5 4.5 2.5 3.5 4.5 5.5 3.5 4.5 fileFolder = open(’TEXTFILE’, ’r’) readFile = fileFolder.readlines () for line in readFile: line = line.split("_") grades = line[-1] grades = map(float,line[-1]) could not convert string to float: . |
| Llama-3.2 |
| (r1, in 4/4 top-10) map is operating on each character of the string, because when you iterate over a string you get individual characters. |
| #30007 - sentence-initial imperatives: calls to action |
|---|
| Qwen3 |
| (r1, in 4/4 top-10) Glyptodonts became extinct at the end of the last ice age along with a large number of other megafaunal species, including pampatheres, the giant ground sloths, and the Macrauchenia. Finding an ancient shell like this is surely not an ordinary discovery. |
| (r5, in 1/4 top-10) The second class is known as “blockchain technology”, and consists of both cryptocurrencies like Bitcoin and Ethereum, and alternative distributed ledger designs like Hyperledger and ErisDB. |
| (r10, in 3/4 top-10) If this ratio is used to design the board the board size would be increased in area by greater than 9 times. |
| Llama-3.2 |
| (r1, in 2/4 top-10) …d her bed, pulling down the covers, and patting the pillow). Give your child some advance notice … † |
LLM Interpretability
Large language model (LLM) interpretability studies how internal representations and computations produce outputs, and how they can be explained or controlled.
See all 776 papers