Polysemanticity or Polysemy? Lexical Identity Confounds Superposition Metrics
Organizations: Department of Computer Science George Washington University Washington, D.C, USA
Abstract
If the same neuron activates for both "lender" and "riverside," standard metrics attribute the overlap to superposition--the neuron must be compressing two unrelated concepts. This work explores how much of the overlap is due a lexical confound: neurons fire for a shared word form (such as "bank") rather than for two compressed concepts. A 2x2 factorial decomposition reveals that the lexical-only condition (same word, different meaning) consistently exceeds the semantic-only condition (different word, same meaning) across models spanning 110M-70B parameters. The confound carries into sparse autoencoders (18-36% of features blend senses), sits in <=1% of activation dimensions, and hurts downstream tasks: filtering it out improves word sense disambiguation and makes knowledge edits more selective (p = 0.002).
Figures & tables
| Model | Params | Layers | Type | Ref. | |
|---|---|---|---|---|---|
| GPT-2 | 117M | 12 | 3072 | Auto | Radford et al. 2019 |
| GPT-2-Med | 345M | 24 | 4096 | Auto | Radford et al. 2019 |
| BERT-base | 110M | 12 | 3072 | Bidir | Devlin et al. 2019 |
| ELECTRA-base | 110M | 12 | 3072 | Bidir | Clark et al. 2020 |
| Pythia-1B | 1.0B | 16 | 8192 | Auto | Biderman et al. 2023 |
| Pythia-6.9B | 6.9B | 32 | 16384 | Auto | Biderman et al. 2023 |
| Ablation | ppl A | ppl B | Specificity |
|---|---|---|---|
| Sense-A neurons | |||
| Sense-B neurons | |||
| Sense-blind | |||
| Random |
Appendix figures & tables15 assets
Supplementary material from the paper’s appendix.
Appendix
| Condition | Words | Pairs (total) | Pairs/word (mean) |
|---|---|---|---|
| Same-Lemma (SL) | 407 | 40,700 | 100 |
| Polysemantic (PS) | 407 | 40,700 | 100 |
| Synonym (SYN) | 251 | 12,500 | 50 |
| Cross-Lemma (CL) | 407 | 40,700 | 100 |
| Layer | GPT-2 | BERT | ELECTRA | GPT-2-Med |
|---|---|---|---|---|
| 0 | .735 | .662 | .686 | .751 |
| 1 | .629 | .566 | .528 | .700 |
| 2 | .560 | .552 | .482 | .690 |
| 3 | .494 | .471 | .428 | .612 |
| 4 | .446 | .398 | .367 | .549 |
| 5 | .441 | .409 | .307 | .524 |
| Layer | Pythia-12B | LLaMA-2-13B |
|---|---|---|
| 0 | .913 | .911 |
| 5 | .516 | .406 |
| 10 | .377 | .262 |
| 15 | .360 | .456 |
| 20 | .422 | .476 |
| 25 | .391 | .348 |
| Model | (L0) | Exceeds? | |
|---|---|---|---|
| GPT-2 | 0.713 | 0.735 | Yes |
| GPT-2-Med | 0.720 | 0.751 | Yes |
| BERT-base | 0.778 | 0.662 | No |
| ELECTRA-base | 0.727 | 0.686 | No |
| Pythia-1B | 0.821 | 0.836 | Yes |
| Pythia-6.9B | 0.882 | 0.910 | Yes |
| Layer | GPT-2 | BERT | ELECTRA | GPT-2-Med |
|---|---|---|---|---|
| 0 | 0.88 | 0.63 | 0.57 | 0.70 |
| 1 | 0.77 | 0.78 | 0.64 | 0.75 |
| 2 | 0.83 | 0.64 | 0.76 | 0.80 |
| 3 | 0.84 | 0.61 | 0.74 | 0.85 |
| 4 | 0.72 | 0.67 | 0.81 | 0.78 |
| 5 | 0.74 | 0.63 | 0.83 | 0.74 |
| 32k SAE | 128k SAE | |||||
|---|---|---|---|---|---|---|
| Layer | Active | Blind | Coll.% | Active | Blind | Coll.% |
| 0 | 33.6 | 10.4 | 31.7 | 31.5 | 9.8 | 32.6 |
| 2 | 27.8 | 7.4 | 29.1 | 26.4 | 7.6 | 31.3 |
| 5 | 21.3 | 4.3 | 24.1 | 20.0 | 4.4 | 25.7 |
| 8 | 18.6 | 2.7 | 17.7 | 15.2 | 2.6 | 22.4 |
| 11 | 18.4 | 3.9 | 28.9 | 15.2 | 2.9 | 25.9 |
| Model | L0 | L | L | L | L n-1 | min |
|---|---|---|---|---|---|---|
| GPT-2 | .43 | .34 | .45 | .56 | .37 | .28 |
| GPT-2-Med | .50 | .28 | .38 | .42 | .28 | .28 |
| BERT-base | .57 | .31 | .14 | .18 | .16 | .14 |
| ELECTRA | .48 | .33 | .50 | .53 | .26 | .26 |
| Pythia-1B | .62 | .20 | .33 | .30 | .16 | .16 |
| Pythia-6.9B | .80 | .43 | .26 | .13 | .25 | .13 |
| PS–SYN gap | PS | SYN | ||
| GPT-2 (d=3072, layer-averaged) | ||||
| Baseline | 0.119 | 0.65 | — | — |
| (0.3%) | 0.087 | 0.52 | 0.008 | 0.032 |
| (0.7%) | 0.047 | 0.41 | 0.019 | 0.054 |
| (1.6%) | 0.028 | 0 | 0.031 | 0.116 |
| Pythia-1B (d=8192, layer-averaged) | ||||
| Model | SL | PS | SYN | CL | PS SYN |
|---|---|---|---|---|---|
| GPT-2 | .90–.60 | .89–.47 | .31–.45 | .26–.45 | |
| BERT-base | .70–.58 | .54–.30 | .09–.32 | .05–.35 | |
| Pythia-1B | .99–.46 | .99–.27 | .09–.28 | .09–.26 |