| E[L0] | Model | TopK | Vanilla | JumpReLU | BatchTopK |
|---|
| R2 | Norm. F1 | R2 | Norm. F1 | R2 | Norm. F1 | R2 | Norm. F1 |
|---|
| | Baseline | 61.1±0.2 | 58.0±1.4 | 40.8±0.8 | 51.5±1.0 | 38.4±0.7 | 52.9±0.7 | 57.9±0.9 | 54.2±0.5 |
|---|
| 8 | Spatial-sae | 45.1±0.1 | 67.0±1.2 | 27.7±0.5 | 55.7±2.0 | 30.4±0.3 | 57.2±1.2 | 43.9±0.3 | 60.2±1.6 |
|---|
| | Baseline | 70.4±0.2 | 57.5±1.1 | 53.7±0.7 | 50.5±1.0 | 52.2±0.8 | 52.1±0.5 | 69.2±0.9 | 55.9±0.7 |
|---|
| 16 | Spatial-sae | 52.4±0.3 | 65.4±1.9 | 39.1±0.7 | 55.5±2.0 | 41.4±0.3 | 56.7±1.6 | 52.4±0.4 | 60.3±2.0 |
|---|
| | Baseline | 78.1±0.2 | 56.1±0.7 | 66.1±0.6 | 49.1±0.7 | 67.5±0.5 | 49.8±0.3 | 78.0±0.2 | 54.9±0.7 |
|---|