In this paper
AbstractIntroductionRelated WorkProblem Setup and NotationMethod: A Four-Stage PipelineResultsDecodability Is Nearly Universal but Heavily SharedEvery Strongly Supported Comparison Selects a Ribbon or a TorusMost Shapes Are Inherited, Not OwnedThe Manifold Basis Retains 61% of the Causal Effect of the ProbeAnalysisDiscussion and ConclusionAI Use StatementEthics StatementReproducibility StatementNotation and TermsExtended Introduction MaterialModels and Activation CapturePrompts, Tokenisation, and DataConcept Definitions and Related ConceptsCell Counts and Exclusion RulesMethod DetailsChecks on the probe subspaceResults DetailsReferencesWorksComputer Science
Linear Probes to Bayesian Manifolds: Geometry of Arithmetic in Large Language Models
Abstract
Linear probes are a standard method to study geometric structures, such as helices and circles, from the activations of large language models (LLMs), and such structure is often taken as evidence that the model computes with it. Probe accuracy alone cannot establish this, because it conflates four separate claims. These are (1) that the concept is decodable from the activations, (2) that they form a specific shape, (3) that the shape belongs to the probed concept rather than to a related concept the model also represents, and (4) that the output of the model depends on the shape. We call these claims decodability, shape, ownership, and causal use, and we test them separately with a four-stage Bayesian pipeline. Stage 2 compares candidate shapes and selects the best one only when its evidence passes a predefined threshold. Stage 3 removes a set of related concepts, specified in advance, and tests whether the shape is owned or inherited. Stage 4 intervenes on the shape to test whether the output of the model depends on it. We apply the pipeline to two-digit addition and multiplication in GPT-J 6B, Llama 3.1 8B, and Pythia 6.9B. Nearly every concept is decodable, but the decoded directions overlap substantially across concepts. Every strongly supported comparison selects a ribbon, which is a helix whose radius changes with the value, or a torus. Of 1,330 cells with an ownership verdict, 19% of shapes are owned and 81% are inherited. Owned and inherited shapes move the output about equally often. Intervening on the shape retains approximately 61% of the effect of intervening on the full probe while using at most five dimensions, against a median of nine for the probe. Our results indicate that a probe verdict alone overstates what a model computes with, and we propose that probe results be reported as verdicts on all four claims rather than as a single accuracy.
Introduction
A central goal of mechanistic interpretability is to explain the computation of large language models (LLMs) in terms of concepts that humans understand [42]. Arithmetic is a useful setting for this goal because for addition and multiplication as we can name every intermediate concept, including operand digits, carries, partial products, and column sums. Recent work finds that such concepts and others are represented geometrically, on helices, circles, and curved manifolds [27, 35, 13, 21], as Section 2 surveys.
A standard method for locating such structure is the linear probe [1], and a high probe accuracy is often taken as evidence that the model computes with the structure. Accuracy alone cannot establish this [25, 3, 11], because it answers only the first of four separate questions. These are whether the concept can be decoded from the activations, what shape they form, whether that shape belongs to the probed concept or to a related concept the model also represents, and whether the output of the model depends on it. We call the corresponding claims decodability, shape, ownership, and causal use.
The four claims can receive different answers. Under superposition one direction can carry several features [12], so a probe identifies a direction or a subspace that the model may use, ignore, or share across concepts. Every intermediate concept is also a function of the operand digits, so a probe trained on one concept can succeed by decoding another. In , our running example, the units digit of the answer, 1, is fixed by the units partial product, , which also fixes the units carry, 2. A claim that a shape belongs to one concept therefore requires that its related concepts, those that could produce the same geometry, be removed first.
We test the four claims separately with a four-stage Bayesian pipeline (Figure 1, § 4, Appendix § 12). Stage 1 tests decodability with a probe, verifies the result against chance, and measures noise, overlap with other concepts, and distortion of distances. Stage 2 tests shape by fitting named low-dimensional manifolds, such as a circle, a torus, and a ribbon, which is a helix of varying radius. It selects the best one only when its evidence exceeds that of a generic baseline by a predefined threshold. Stage 3 tests ownership by removing the related concepts, listed before any experiment, from the activations and fitting the shapes again. A shape whose evidence over the baseline rises after the removal is owned by the concept, and otherwise it is inherited from a related concept. Stage 4 tests causal use by intervening on the probe subspace and on its manifold basis, the directions inside it that the shape occupies, and compares both with random subspaces of equal rank. Bayesian model comparison scores shapes of different complexity on one scale, the marginal likelihood, and shows when no candidate is supported [29]. Every verdict is issued for one cell, a fixed combination of model, task, layer, concept, and preprocessing setting (§ 3). The four verdicts therefore refer to the same data.
Figure 1. The four-stage pipeline, one stage per claim. All four criteria are passed in only 46 of 1,330 cells.
We limit our scope to two-digit addition and multiplication in GPT-J 6B, Llama 3.1 8B, and Pythia 6.9B [52, 7] (Graffatiori et al. 2024), and to a single token position. All activations are read, and all interventions applied, at the final = token of the prompt, which immediately precedes the answer and at which the answer is assembled from information moved there (Mandianna et al., 2025). Circuit-level analyses [53, 10, 30] study the computation across operand positions and attention routes [50]. Our results describe the geometry at the position where the answer is read out.
Decodability is nearly universal but heavily shared. Without preprocessing, 89% of the 1,620 cells are decodable, yet 78% to 91% of concept pairs align more closely than chance.
Every strongly supported comparison selects a ribbon or a torus (338 cells).
Most shapes are inherited, not owned (81% against 19% of the 1,330 Stage 3 verdicts).
The manifold basis retains 61% of the causal effect of the probe (from 0.32 to 0.70).
The four claims therefore separate in practice, and only 46 of the 1,330 cells pass all four criteria. We read this count as conservative, because a cell must meet the threshold of every stage, and as conditional on the listed related concepts. The closest work fits a generalised helix and confirms it by activation patching [27]. We compare candidate shapes by evidence, measure the change after removing related concepts listed before any experiment, and test interventions on the directions of the shape, all on the same cells. We propose that a probe result be reported as four verdicts and not as one accuracy.
Related Work
Probes and superposition. Linear probes [1] read concepts from directions of the residual stream [45, 46, 20]. Control tasks [25] separate what the activations encode from what a probe learns. Amnesic probing [11], the survey of Belinkov [3], and causal readings of probe results [26, 49] separate decoding from use. Superposition explains why one direction serves several concepts [12]. The four claims turn these concerns into four verdicts on one cell.
Arithmetic in language models. Small transformers solve modular addition on circles, by rotation [35, 43] or by other algorithms on the same circles [57]. In pretrained models, addition is localised to a few attention heads and feed-forward layers [50, 56], assembled at the last token from information moved there [38], and carried in part by Fourier features [58] or, in a competing account, by neuron-level heuristics [44]. Numbers are reported on a helix [27], on base 10 digit circles [33], on a line [59], and as periodic features shared across models [16]. The partial products of multiplication are reported as Minkowski sums of digit embeddings in a model trained with implicit chain of thought [2]. Wen et al. [54] describe the shape at the digit positions of a multi-digit answer, Feucht et al. [14] study the same position as we do and test whether the model computes along the circle, and neither tests whether the shape is inherited.
Manifolds and steering. Concepts represented on curves include the days of the week [13] and character counts [21]. Such curves are predicted by a continuous correspondence between a concept and its representation [42] and produced by symmetry in the statistics of language [28]. Gurnee et al. [21] validate one manifold with interventions, Li et al. (2025) cover many tasks but establish decodability alone, and steering along a fitted manifold [48, 41] (Wurgaft et al. 2026) establishes causal use without comparing named shapes. We compare named shapes by evidence, test whether the selected shape is owned, and intervene on its manifold basis, on the same cells.
Concept removal and interventions. Projecting a concept out of the activations [47, 5] and measuring what remains usable [11] or how completely and selectively the removal acts [8] are the ancestors of Stage 3, which refits the shape rather than a classifier, and of the ablation of Stage 4. Causal abstraction gives interventions their meaning [17], activation patching gives them their form [24], and the subspace illusion [37] sets the limit of what a control by random subspaces of matching rank can show.
Probabilistic geometry. The shape model is the linear-Gaussian limit of the Gaussian process latent variable model [31]. Isomap [51] and uniform manifold approximation and projection (UMAP, McInnes et al., 2018) do not supply the marginal likelihood on which the comparison rests, so they cannot say whether a shape is supported [cf. 23], which a verdict on the shape claim requires.
Problem Setup and Notation
Models and activations. The three models are base models without instruction tuning, share hidden size , and are those in which the number helix was reported [27]. For each model we read the residual stream at five layers of matched relative depth, which are layers 4, 8, 14, 20, and 24 of 28 in GPT-J and 4, 8, 16, 24, and 28 of 32 in Llama and Pythia, always at the final = token Appendix § 13.
Tasks and data. Both tasks use operands from 0 to 99 and one fixed prompt that ends in the problem, such as Appendix § 14, so every verdict is conditional on that prompt. We keep only problems whose correct answer is a single token in all three models, which holds for all 10,000 addition problems and for 3,023 multiplication problems. We analyse only the problems that a model answers correctly (Table 7 gives the shares), because our question concerns the geometry that accompanies a successful computation, and we make no claim about problems answered incorrectly.
| Model | Addition | Multiplication |
| GPT-J 6B | 8,415 (84.15%) | 2,751 (91.00%) |
| Llama 3.1 8B | 9,963 (99.63%) | 2,927 (96.82%) |
| Pythia 6.9B | 7,718 (77.18%) | 2,757 (91.20%) |
Table 7. Problems answered correctly by each model. Percentages are of 10,000 addition problems and 3,023 multiplication problems.
Concepts. The single concepts are the quantities that the standard written algorithms name, together with simple properties of the operands and of the answer such as parity, magnitude, and comparisons, which gives 41 for addition and 47 for multiplication. Ten joint concepts per task, listed before any experiment and each formed from two or three single concepts, bring the totals to 51 and 57. All 108 concepts enter the decodability and shape counts that we report. Six core concepts per task, all quantities of the standard written algorithms, are also reported separately, and we call the others extended concepts. For addition the core concepts are the operands and , the sum , the units and tens digits of the answer, and the units carry. For multiplication they are the units and tens digits of , the units digit of , the units partial product, the units carry, and the units digit of the answer, which take the values 3, 2, 7, 21, 2, and 1 in the running example Appendix § 15 gives the definitions and lists all concepts). Where the arithmetic ties a concept to others, we also list these related concepts § 1. The related concepts of the units digit of the answer in multiplication, for example, are the operands, their units digits, the units partial product and column sum, which are equal, and the units carry. Such a list exists for 46 of the 108 concepts, and only the cells of these concepts receive an ownership verdict.
Cells and notation. Each analysis runs under one of three preprocessing settings , named off, answer, and norm. The first leaves the activations unchanged. The second removes the unembedding direction of the answer token, because towards the output the residual stream aligns with that direction, and this readout effect could be mistaken for the geometry of a concept. The third rescales every activation to unit length, because the norm of the residual stream varies widely (Appendix C) and a shape should not depend on it. The two altered settings therefore repeat every verdict without readout alignment and without activation scale, and Section 5 pools the three settings from Stage 2 on. A cell is , with model , task , layer , concept , and setting . Three models, five layers, 108 concepts, and three settings give 4,860 cells, 1,620 candidate cells per setting. A value enters the analysis only if at least 30 correctly answered problems take it, a floor set because smaller groups give unstable covariance estimates, and a cell only if at least two of its values remain (Appendix F). For one cell, stacks the activations of the remaining problems and stacks their concept values. We write for the number of problems with value and for their mean activation, the centroid of (Appendix A lists all symbols, terms, and abbreviations).
Method: A Four-Stage Pipeline
The pipeline takes one cell, with its activations and concept values , and issues one verdict per claim. Every later stage fits or intervenes on directions of the subspace that Stage 1 returns, so the four verdicts describe the same directions. The same thresholds apply to every cell (Appendix G, Table 15), and a stage that cannot meet its threshold declines.
Stage 1, decodability. A concept is linearly decodable when the mean activation changes with its value, so we build the probe from the centroids. Let be the mean activation of the cell. We stack the centred centroids as rows, each weighted by to limit the influence of values with few problems, and take the singular value decomposition (SVD). We keep the leading right singular vectors whose singular values exceed the 99th percentile of their counterparts under 1,000 random permutations of . The kept vectors are orthonormal and form , a basis of the probe subspace. A cell is decodable when , a verdict on whether the centroids vary more than chance and not on held-out accuracy, and its probe subspace is then verified. A linear discriminant analysis (LDA), Fisher, 1936) inside then checks each direction with a second permutation test and with held-out nearest-neighbour accuracy, as diagnostics that leave and the verdict unchanged. In the running example the values are the digits 0 to 9, the activation for a correctly answered belongs to the group of value 1, and spans the directions that separate the ten centroids.
Three further checks remove no cell but qualify the verdict. The first compares the largest eigenvalue of the covariance of the activations outside with the noise level under the Marčenko–Pastur law [39]. The second marks two concepts as superposed when the smallest principal angle between their subspaces lies more than below the fifth percentile of 1,000 pairs of random subspaces of the same ranks, with the false discovery rate (FDR) held at 0.05 [6], and the share of marked pairs among the concepts of the same model, layer, and setting is the superposition rate. The third is the Spearman correlation between the pairwise distances of the problems before and after projection onto , which shows whether keeps the order of those distances.
Stage 2, shape. Stage 2 asks which named shape, if any, the activations form inside the probe subspace, using the coordinates , where is row of and its concept value. A candidate shape is a feature map that places the value on a curve, and the torus and the ribbon name the surfaces on which their curves lie. The circle has the features and with , so for the units digit of the answer the period of places the digits evenly around one turn, with 9 next to 0. The ellipse has a linear feature to the circle, the torus is the product of two circles with two periods, the ribbon is a helix whose radius changes linearly with the value, and a line and the concentric shape complete the six named shapes (Table 14). We model the coordinates as a linear image of the shape plus noise,
where orients, scales, and can flatten the shape inside the subspace, is our prior variance of its entries, and gives each value its own spread. Step 2a proposes the period from a Fourier analysis of the centroids, and Step 2b supplies from the covariance of the activations of each value. In Step 2c, is integrated out in closed form because the model is linear and Gaussian, which gives the marginal likelihood of every shape, its evidence. It rewards fit and penalises unused flexibility [36], so richer shapes need no separate penalty.
The baseline is a cubic polynomial in the value, with a smooth trend and no periodic feature. Let be the named shape with the highest evidence and its evidence gap, in nats. The shape is selected when , a Bayes factor of about 150 [29], and the bar rises to 10 when six or fewer values remain. The stage declines when no shape is selected or when the alignment of the selected shape, its correlation with the concept values, is below 0.5. The strictest evidence grade, Tier A, adds two conditions and a permutation control (Appendix G). The held-out error of the shape may exceed that of the best alternative by at most 10%, and, for stability, the shape must keep the highest evidence under all nine combinations of three random seeds and three values of . Tier A names the exact shape when the best shape leads the runner-up by the bar after a correction for the uncertainty of the period, which we call a strongly supported comparison, and otherwise it names only the family of the shape, one of three groups (Appendix § 17), when the family evidence leads the other families by the bar. A selected and aligned shape that fails either condition receives Tier B (Appendix § 17 gives the lower grades). Stage 2 validates named shapes within one candidate set, one noise model, and one prior, and does not discover new ones.
Stage 3, ownership. Related concepts share directions under superposition, so a shape found for one concept may be produced by another. Stage 3 removes the related concepts by projection, as in linear concept removal (Ravfogel et al., [47]; Belrose et al., [5]), and measures what remains. For a concept , we orthonormalise the union of the Stage 1 bases of its related concepts, taken at the same model, layer, and setting, into . We project this span out of every activation and recompute the coordinates, . All shapes are fitted again with , the periods, and the noise held at their Stage 2 values and only the weights refitted. With the evidence gap after the removal, the verdict rests on
The shape is owned when and inherited when . The removal is linear and deletes only the span of , so Appendix § 17 (Equation (9)) decomposes into the advantage of the selected shape over the baseline on the removed component of the coordinates and the interaction between that component and the remaining one. The interaction term vanishes when , that is, when the removal lies entirely within the probe subspace. The rule is strict at the boundary, so a cell whose coordinates the removal leaves unchanged () counts as inherited, as does a representation that is the sum of the representations of its related concepts. In the running example, the units digit of the answer is a function of the units partial product alone, so each of its centroids is a weighted mean of the centroids of that related concept, its probe subspace lies within their span, and an inherited verdict is the expected outcome. The verdict is operational and relative to the listed related concepts, and it does not show that the shape is unique to among all concepts (Appendix § 17 gives its two limits).
Stage 4, causal use. The first three stages describe the activations. Stage 4 changes them and observes the output, as a causal claim requires (Geiger et al., [17]). The interventions act on and on the manifold basis , an orthonormal basis of the column space of , where is the posterior mean of for the selected shape without its constant row. The manifold basis spans the directions of that the shape occupies, and its rank is at most five, the number of non-constant features of the ribbon. Neither intervention moves an activation along the shape, as manifold steering does (Wurgraft et al., 2026; Sarfati et al., [48]; Modell, [41]). Let be the activation at the final = token of layer and either basis. Ablation replaces the part of inside the subspace by its mean, and patching replaces it by that of a donor problem with a different concept value (Heimersheim & Nanda, [24]),
where is the mean of over the correctly answered problems and is the activation of the donor. In the running example, ablation removes what separates the digit 1 from the other digits, and patching inserts the digit of a donor whose answer ends differently.
The accuracy change is the difference, in percentage points, between the shares of problems answered correctly with and without ablation. The causal excess is the change in the logit of a target token minus the mean change under five random subspaces of the same rank. The target token is the correct answer for ablation and the donor answer for patching, so the expected excess is negative under ablation and positive under patching. The random subspaces, drawn in the full activation space, control for disturbing any subspace of that rank, and not for a pathway that is dormant on ordinary inputs (Makelov et al., [37]). A cell is causally active when the causal excess of is at least 0.5 logits in absolute value. A change in either direction shows that the output depends on the manifold basis, so the count uses the absolute value, and § 5 reports the signs and the stricter level of 1.0. The retention is the mean accuracy change under ablation of divided by that under ablation of , over the same cells. Stages 3 and 4 issue verdicts only for cells whose shape reaches Tier A or Tier B. A cell satisfies all four claims when it is decodable, its shape reaches Tier A and is owned, and it is causally active.
Fixed thresholds. The prior variance of the two cross features of the ribbon is scaled by 0.1, which lowers the prior weight of the cross features. Only the floor, the prior variance , the causal level, and the FDR of the permutation control of Tier A were examined at other values (§ 5 and Appendices E and G).
Results
We ran the pipeline on 4,860 cells, 1,620 in each of three preprocessing settings. Without preprocessing 89% of the cells are decodable, every strongly supported shape comparison ends in a ribbon or a torus, 19% of the 1,330 ownership verdicts are owned and 81% inherited, the manifold basis keeps 61% of the accuracy effect of the probe subspace, and 46 cells pass all four criteria. Table 1 gives every stage by model and task. The cells behind it are not independent, because one concept appears at five layers and in three settings and some concepts are functions of others, so we report counts and shares and attach no significance test to them.
| Model | Task | Superposition | Ribbon | Torus | Other | Owned | Inherited | Retention | ||
| GPT-J | addition | 0.808 | 161 | 42 | 0 | 24 | 147 | −21.5 | −15.0 | 0.70 |
| GPT-J | multiplication | 0.897 | 238 | 43 | 7 | 50 | 199 | −11.2 | −7.6 | 0.68 |
| Llama | addition | 0.782 | 173 | 37 | 1 | 42 | 151 | −6.3 | −2.0 | 0.32 |
| Llama | multiplication | 0.907 | 256 | 19 | 5 | 46 | 226 | −4.0 | −2.2 | 0.55 |
| Pythia | addition | 0.815 | 141 | 61 | 3 | 42 | 149 | −15.5 | −6.7 | 0.43 |
| Pythia | multiplication | 0.909 | 239 | 28 | 3 | 49 | 205 | −20.0 | −13.8 | 0.69 |
| All | 1,208 | 230 | 19 | 253 | 1,077 | −12.6 | −7.7 | 0.61 |
Table 1. Results by model and task. Superposition is the mean rate over the cells of the row. Ribbon, torus, and other count the cells with a shape fit, 1,457 in all, pooled over the three settings, by the shape with the highest evidence, and other covers the baseline, the circle, and the open helix. Owned and inherited count the cells with an ownership verdict, 1,330 in all. Ablation gives the mean accuracy change in percentage points under ablation of and of over the same 1,330 cells, and retention is their ratio.
Decodability Is Nearly Universal but Heavily Shared
Without preprocessing, 1,445 of the 1,620 candidate cells (89%) are decodable (Table 13 traces every population). The verified probe subspaces have a median rank of nine, a mean of 18.3, and a maximum of 199. The units digit of the answer is decodable in all three models.
| Population | Setting | Condition | Cells |
| All cells | all three | every model, task, layer, concept, and setting | 4,860 |
| Candidate cells | one | before any exclusion rule | 1,620 |
| Removed by the floor | off | fewer than two values with 30 problems | 155 |
| Removed by the permutation test | off | no direction kept, | 20 |
| Decodable cells | off | 1,445 | |
| Cells with a shape fit | pooled | evidence of every candidate shape computed | 1,457 |
| Strongly supported comparisons | pooled | Tier A naming the exact shape | 338 |
| Cells with an ownership verdict | pooled | Tier A or Tier B, a list of related concepts, and of positive rank | 1,330 |
| Owned cells | pooled | 253 | |
| Cells with all four claims | pooled | decodable, Tier A, owned, and causally active | 46 |
Table 13. Populations of cells, with the setting in which each is counted and the condition that defines it.
The superposition rate averages 0.80 on addition and 0.90 on multiplication (Table 1), so the subspaces of most pairs of concepts align more closely than chance, in part because the values of related concepts are functions of the same digits, and probe accuracy alone is not evidence of structure specific to one concept. In the addition cells, where the Marčenko–Pastur edge is a usable reference (Appendices § 17 and § 18), the largest residual eigenvalue outside the probe subspace lies at 6 to 8 times the noise level on average and at up to 27 times, so the residual stream carries structure outside the probe subspace. Inside the subspace, the Spearman correlation of the pairwise distances before and after projection deviates from one by a median of , so Stage 2 fits its shapes to coordinates that keep the distance ordering.
Every Strongly Supported Comparison Selects a Ribbon or a Torus
Among the 1,457 cells with a shape fit, pooled over the three settings, the ribbon has the highest evidence in 1,208 and the torus in 230 (Table 1). The ribbon or the torus is the selected shape in each of the 338 strongly supported comparisons, 111 of them among the core concepts. The other 19 cells fall to the baseline (13), the circle (4), and the open helix (2). For concepts that span the two-digit range, such as the operands and the sum, the ribbon mostly has the highest evidence, since its linear feature carries the trend. For residues modulo 10 the torus has it in many cells, with the exceptions listed below, and concepts of mixed range are split between the two. Among the core concepts without preprocessing, the highest evidence goes to the ribbon for some residues as well, for the units digit of in multiplication in every model, for every concept with a verdict in Llama, and for the units digits of and of the answer in GPT-J multiplication. For the operand in Pythia addition the baseline has the highest evidence, so the comparison can prefer the baseline to every named shape (Appendix § 18). In multiplication the units digit of the answer, our running example, gives the ribbon the highest evidence in GPT-J and Llama and the torus in Pythia.
With the weights of its two cross features, the products of its linear and circular features, at zero the ribbon is the open helix, and with the weight of its linear feature at zero as well it is the circle (Table 14), so the evidence selects it only when these features are supported. The torus has the highest evidence in the 230 cells where a product of two circles fits better. The open helix among the candidates is the single-period form of the helix of Kantamneni & Tegmark [27], which the ribbon contains and the evidence prefers. Among the core concepts, the median evidence gap, which depends on the number of problems and of coordinates of a cell (Appendix § 17), is 10,866 nats for strongly supported comparisons, 224 nats for Tier A verdicts that name a family, and 509 nats for Tier B. The bar of 5 nats is therefore a necessary condition and not the deciding one.
| Shape | Features | Description |
| generic baseline | , , | smooth trend with no periodic feature |
| line | values ordered along one direction | |
| circle | , | values repeat with period |
| open helix | , , | circle that advances along a line |
| torus | , , , | product of two circles |
| concentric | , , | circle with a fixed radial drift |
| ribbon | , , , , | helix whose radius changes linearly |
Table 14. Candidate shapes of Stage 2. Every feature map also contains the constant 1. The value rescaled linearly to the unit interval is , the angular frequency is for a period , the torus has two periods with angular frequencies and , and the radius of the concentric shape is .
Most Shapes Are Inherited, Not Owned
Of the 1,330 cells with an ownership verdict, 253 (19%) are owned under the rule of Stage 3 and 1,077 (81%) are inherited, and the owned share lies between 14% and 22% in every model and task (Table 1) and is 18%, 20%, and 19% under the settings off, answer, and norm (Table 17). These cells have a shape of Tier A or Tier B, a list of related concepts, and a manifold basis of positive rank, and they are also the cells eligible for intervention. Among the core concepts the inherited share is higher still, with 16 of 337 verdicts owned (5%), and the running example gives the outcome that § 4 expects, since the units digit of the answer is inherited in all 30 of its cells with a verdict, and the tens column sum of GPT-J addition gives the other outcome at layer 20 and passes all four criteria (Appendix § 18, Figure 4).
| off | answer | norm | all | |
| Tier A, exact shape | 100 | 118 | 120 | 338 |
| Tier A, family | 31 | 14 | 36 | 81 |
| Tier B | 348 | 245 | 318 | 911 |
| Cells with an ownership verdict | 479 | 377 | 474 | 1,330 |
| Ribbon has the highest evidence | 399 | 314 | 394 | 1,107 |
| Torus has the highest evidence | 78 | 61 | 78 | 217 |
| Owned | 88 | 77 | 88 | 253 |
| Owned share | 18% | 20% | 19% | 19% |
| Retention | 0.64 | 0.61 | 0.57 | 0.61 |
| Cells that pass all four criteria | 16 | 14 | 16 | 46 |
Table 17. Results by preprocessing setting over the 1,330 cells with an ownership verdict. Retention is the ratio of the mean accuracy changes under ablation of and of .
Figure 4. The tens column sum of addition in GPT-J at layer 20 without preprocessing, the cell that passes all four criteria. Left and middle, the centroids of its 19 values (dots, every third labelled), a sample of 600 problems (light), and the fitted ribbon (curve), in the three leading principal coordinates of the ribbon inside the manifold basis, which hold 0.72 of the variance of the centroids. Right, the centroids after the subspaces of the two tens digits are removed, in their own two leading coordinates and joined in the order of the values, at 0.49 of the spread before removal. The values wind around twice with a growing radius, the ribbon that Stage 3 refits, which is why the evidence gap grows.
The verdicts lie far from the boundary. Over the 1,330 cells the smallest absolute value of is 1.51 nats and the median about 3,024 nats, four verdicts lie within 5 nats of zero and none at it, so a margin of 5 nats would change four verdicts of 1,330 and none among the core concepts, whose smallest is 10.58 nats. Among the core concepts the removal takes on average 0.85 of the variance of the coordinates on owned cells and 0.89 on inherited cells, and leaves the alignment on average at 0.97 and 1.03 of its value, so owned cells lose about as much variance as inherited cells and their verdict rests on the evidence gap and not on the size of the projection.
The Manifold Basis Retains 61% of the Causal Effect of the Probe
Averaged over the 1,330 cells, ablating at the final token lowers the accuracy by 12.6 points and ablating by 7.7 points, a retention of 61%, between 0.54 and 0.67 when the concepts are resampled (Appendix § 18). Over the cells of each model, retention is 0.69 in GPT-J, 0.60 in Pythia, and 0.42 in Llama, and by setting it is 0.64, 0.61, and 0.57 in the order of Table 17, where ablating itself lowers the accuracy by 5.0 points against 15.4 in GPT-J and 18.1 in Pythia, over the same cells. The retention is a ratio of effects and not of ranks, and the causal excess controls for rank.
The causal excess has the expected sign. On the core concepts, the mean under ablation is logits in , with a standard deviation (SD) of 1.03, and at (SD 0.59), the mean under patching is +1.70 (SD 1.77) and +1.02 (SD 0.99), and every concept-level mean under ablation of is negative. Of the 337 cells of the core concepts, 154 (46%) are causally active at 0.5 logits and 66 (20%) at 1.0, all of them with a negative excess, and under patching keeps 60% of the effect of . For the running example, the mean causal excess under ablation of is logits in GPT-J, in Llama, and in Pythia.
The largest effect of ablating lies at the middle layer, 14 of 28 or 16 of 32, in five of the six model and task pairs and at layer 24 in Llama multiplication, and at layer 4 every change lies within two points of zero (Figure 2). A geometry that served only the readout, where the residual stream aligns with the unembedding direction ([4]), would matter most at the deepest layers, so the profile does not match that of a readout direction alone.
Figure 2. Mean accuracy change under ablation of the probe subspace (solid) and of the manifold basis (dashed) by layer, over the core concepts and the three settings of each model and task.
All four claims. All four criteria are passed in 46 of the 1,330 cells, 38 with a strongly supported comparison and 8 with a Tier A verdict that names only the family. They lie in every model and task, 37 with a ribbon and 9 with a torus, and 26 combinations of model, task, layer, and concept account for them, 15 of which pass in at least two of the three settings and 5 in all three. Causal activity is about as common among owned shapes as among inherited ones, 103 of the 253 owned cells (41%) against 42% of the inherited cells, so an ownership verdict does not stand in for a causal one. Which concepts own their shapes follows the direction of the arithmetic, since the answer digits and the carries own none of their 217 cells, while the operands, their digits and functions, the column and running sums, the partial products, and the sum own 253 of 1,113, and all 46 cells are of these kinds (Table 18). The units digit of the answer shows the separation. It is decodable, forms a ribbon or a torus, inherits that shape as its dependence on the units partial product predicts, and passes the causal level on average in two of the three models, so probe accuracy alone would have credited it with a shape of its own.
| Kind of concept | Verdicts | Owned | Pass all four |
| Functions of the operands | 252 | 108 | 10 |
| Running sums | 180 | 49 | 7 |
| Operand digits | 298 | 33 | 2 |
| Column sums | 151 | 30 | 7 |
| Operands | 124 | 25 | 17 |
| Partial products | 98 | 5 | 3 |
| Sum of addition | 10 | 3 | 0 |
| Answer digits | 129 | 0 | 0 |
| Carries | 88 | 0 | 0 |
| All | 1,330 | 253 | 46 |
Table 18. Ownership and the cells that pass all four criteria by kind of concept, over the 1,330 cells with a verdict. Functions of the operands are the maximum, the minimum, the difference, and the absolute difference.
Analysis
Ribbons and tori against the circles and helices of prior work. Two differences from the experiments of prior work [35, 43, 57, 27] bear on the shapes that Stage 2 selects. The arithmetic on the operands 0 to 99 is not modular, so a two-digit quantity advances along a line while its units digit repeats, which the linear and periodic features of the ribbon carry. A residue modulo 10 advances along no line, and for these concepts the torus, a product of two periods, has the highest evidence more often than the circle. By the final = token the residual stream has begun to turn towards the unembedding direction of the answer, a component that the linear and cross features of the ribbon carry and the circle cannot, which is a candidate explanation that a separate experiment would test.
Depth. The largest effects of ablating a manifold basis lie at the middle layers, and owned shapes appear at the two earliest layers in every model and task, 88 of the 253 in all, so neither the effect of a manifold basis nor the ownership of its shape is confined to the layers at which the residual stream turns towards the answer.
Structure outside the probe subspace. The structure outside the probe subspace and the overlap of the probe subspaces describe one state of the residual stream at the final token, in which many concepts share directions, so the shape found in one probe subspace can belong to another concept.
What the four verdicts add to a probe. Most shapes that a probe recovers are inherited under the rule of Stage 3, the quantities that enter a column of the written algorithm own shapes and the digits and carries that leave one never do, and owned shapes are causally active no more often than inherited ones. A probe accuracy therefore credits a concept with a shape it may inherit, an ownership verdict credits it with a shape it may not use, and an intervention on the probe subspace credits it with an effect that the manifold basis carries only in part.
Robustness and scope of the interventions. The cells of the core concepts that lose values stay verified at floors of 20 and 50, the ribbon and the torus remain the dominant shapes of Tier A at FDR levels of 0.05 and 0.20 of its permutation control, and every Tier A shape is stable over three values of the prior variance by construction (Appendices § 15 and § 17). Each intervention on the manifold basis is compared with random subspaces of the same rank, so the causal excess reports what the directions of the shape add beyond a perturbation of that rank, while the distance of the changed activation from the native distribution [24, 22] lies outside the scope of that control (Appendix § 17). Stages 1, 2, and 4 apply to any task with a labelled concept and a candidate set of shapes, and Stage 3 to any task that lists related concepts.
Discussion and Conclusion
The cells that pass all four criteria are the yield of a standard that asks four questions of one cell. Every cell with a verdict that falls short does so for a stated reason, no candidate shape at Tier A, a shape that the listed related concepts account for, or a shape whose intervention effect stays below the causal level. The causal measures concern the logit and the accuracy of the answer token, so a shape may serve other outputs without appearing in these measures. Where all four criteria are passed, the interventions show that changing the directions of the shape moves the output, and necessity would need interventions that remove every substrate of the concept at once. The three model families cover two block layouts and three tokenisers at 6 to 9 billion parameters, with the owned share from 14% to 22% and the retention between 0.32 and 0.70 across the six pairs of model and task.
A geometric claim about an arithmetic representation then has four parts, a subspace of the residual stream, a curve inside it, the concept that owns the curve, and an effect on the output, and a probe accuracy settles only the first. The manifold basis may serve as a handle for model editing, and arithmetic beyond single-token answers is the next task.
AI Use Statement
In this work, we used generative AI tools to aid or polish the writing. We have reviewed all AI-assisted text. We take responsibility for the final content of this work, including text, claims or artifacts produced with the aid of generative AI.
Ethics Statement
This work analyses three publicly released base models on synthetic arithmetic problems. It involves no human subjects, uses no data of human origin, releases no model, and changes no model weights. We identified no risk beyond those documented for the models themselves.
Reproducibility Statement
Appendix C describes the three models and the capture convention, Appendix D the prompt and the data, Appendix E every concept with its definition and the lists of related concepts, Appendix F the cell counts and the exclusion rules, Appendix G the method with every threshold (Table 15), and Appendix H the values behind Figure 2. The problems of both tasks, the definitions of the 108 concepts, the code of every stage, and the per-cell verdicts will be released with the final version of this paper.
Notation and Terms
We follow the notation of Goodfellow et al. [18], in which scalars are italic, vectors are bold lower case, and matrices are bold upper case. Table 2 and Table 3 give the symbols of the main text with their values in the running example where one exists, Table 4 gives the symbols introduced in Appendix § 17, Table 5 gives every abbreviation, and Table 6 gives every term that we introduce.
| Symbol | Meaning | Example |
| The two operands, integers from 0 to 99 | 23, 17 | |
| The sum and the product, which are the answers in addition and in multiplication | product 391 | |
| Units and tens digits of , equal to and . The same subscripts apply to , and the tens digit of an answer is | 3, 2 | |
| Indicator, equal to 1 when the condition holds and 0 otherwise | ||
| The real numbers | ||
| A model, one of GPT-J 6B, Llama 3.1 8B, and Pythia 6.9B | ||
| A task, addition or multiplication | multiplication | |
| A layer of the model | ||
| A concept | units digit of the answer | |
| A preprocessing setting, one of off, answer, and norm | ||
| A cell, the unit of analysis | ||
| Hidden size of the model, 4,096 | ||
| Number of problems in a cell, those answered correctly whose value passes the floor | ||
| Activations of a cell, one row per problem | ||
| Concept values of a cell, one entry per problem | entry 1 | |
| One value of a concept, and a second value | 1 | |
| Number of problems in a cell with value | ||
| Centroid of value , the mean activation of its problems |
Table 2. Symbols of Sections 1 and 3 and Appendix E.
| Symbol | Meaning | Example |
| Mean activation of the problems of a cell | ||
| The identity matrix, the vectors of zeros and of ones, and the transpose | ||
| Gaussian distribution with mean and covariance . Its density at is | ||
| Activation and concept value of problem | ||
| Rank of the probe subspace, the number of singular vectors kept in Stage 1 | ||
| Orthonormal basis of the probe subspace found in Stage 1 | ||
| Coordinates of problem inside , before and after the related concepts are removed | ||
| Index of a candidate shape, from 0 to 6, and the shape itself | ||
| Generic baseline shape | ||
| Named shapes, in order line, circle, open helix, torus, concentric, and ribbon | ||
| Feature map of the shape , and the number of its features with the constant | ||
| A value rescaled linearly to the unit interval | ||
| Period of a periodic shape and its angular frequency | ||
| Weights that place a shape inside , one of their entries, and their posterior mean | ||
| Noise added to the coordinates of problem | ||
| Prior variance of the entries of | ||
| Spread of value , the standard deviation of its noise, supplied by Step 2b | ||
| Marginal likelihood of the shape , also called its evidence, a scalar | ||
| nat | Unit of log evidence, from the natural logarithm | |
| Named shape with the highest evidence | ||
| Evidence gap over , before and after the removal | ||
| Orthonormal basis of the span of the subspaces of the related concepts | ||
| The difference , negative when the shape is owned | ||
| Manifold basis, an orthonormal basis of the directions of that the selected shape occupies, or rank at most five | ||
| Basis on which an intervention acts, either or | ||
| Activation at the final = token of a layer, its mean, and that of a donor |
Table 3. Symbols of Section 4 and Table 14.
| Symbol | Meaning | Example |
| Number of values of a concept that remain in a cell | 10 | |
| Weighted and centred centroids, one row per value | ||
| , | Squared singular value of rank of , and the percentile of the same quantity under permutation | |
| , | Ratio of the residual dimension to , and trace of the residual covariance over the residual dimension, in the Marčenko–Pastur check | |
| , | Parts of the activations outside , and the largest eigenvalue of their covariance | |
| Marčenko–Pastur edge, the largest eigenvalue that noise alone produces outside | ||
| , , | Probe bases of two concepts and , and the smallest principal angle between them | |
| , | Euclidean norm of a vector, and largest singular value of a matrix | |
| Frequency of the periodogram of Step 2a, in cycles per ordered values | ||
| Period accepted by Step 2a, which seeds the candidate periods of Step 2c | ||
| Estimated covariance of the activations of value inside | ||
| Radius of the concentric shape at the value | ||
| Angular frequencies of the two circles of the torus | ||
| , | Coordinates of all problems of a cell, one row per problem, before and after the removal | |
| Features of all problems of a cell under the shape | ||
| Diagonal matrix of the noise variances , one per problem | ||
| The matrix from which the evidence and follow | ||
| , , | Linear and circular correlations between a fitted shape and the concept values, and the larger of their absolute values, the alignment | |
| Row of that belongs to the linear feature | ||
| Covariance of one column of coordinates under the shape | ||
| , | The part of the coordinates that the removal deletes, , and the matrix of Equation 9 | |
| , , | Log evidence at the candidate period , the half distance between its two neighbouring periods, and the second difference of there |
Table 4. Symbols introduced in Appendix G.
| Abbreviation | Full term |
| BPE | Byte pair encoding, the tokenisation scheme of the three models |
| CPU | Central processing unit |
| FDR | False discovery rate |
| LDA | Linear discriminant analysis |
| LLM | Large language model |
| OAS | Oracle approximating shrinkage, a covariance estimator |
| SD | Standard deviation |
| SVD | Singular value decomposition |
| UMAP | Uniform manifold approximation and projection, an embedding method |
Table 5. Abbreviations.
| Term | Meaning |
| Cell | One combination of model, task, layer, concept, and preprocessing setting |
| Candidate cell | A cell of one setting before any exclusion rule applies |
| Cell with a shape fit | A cell in which Step 2c computes the evidence of every candidate shape |
| Core concepts | The six concepts per task that are also reported separately |
| Extended concepts | All other listed concepts |
| Joint concept | A concept formed from two or three single concepts |
| Related concepts | The concepts that the arithmetic ties to a concept, listed in advance |
| Floor | The minimum of 30 correctly answered problems that a value needs |
| Flagged cell | A cell with more than three times as many problems at its most frequent value as at its least frequent, kept in the analysis |
| Decodability, shape, ownership, causal use | The four claims, tested by Stages 1 to 4 in that order |
| Decodable | A cell in which Stage 1 keeps at least one direction, |
| Probe subspace | The span of the singular vectors that Stage 1 keeps, with orthonormal basis |
| Verified probe subspace | The probe subspace of a decodable cell. The three further checks of Stage 1 qualify it and remove no cell |
| Superposed | Two concepts whose subspaces align more closely than random subspaces of the same ranks |
| Superposition rate | Share of concept pairs whose subspaces align more closely than chance |
| Candidate shape | One of the seven shapes to that Stage 2 compares |
| Named shape | One of the six candidate shapes to |
| Generic baseline | The shape , a cubic polynomial in the value with no periodic feature |
| Evidence | The marginal likelihood of a shape, with the weights integrated out |
| Evidence gap | Log evidence of the best named shape minus that of the generic baseline, in nats |
| Selected | A shape whose evidence gap reaches the predefined bar of 5 nats, or 10 nats for six or fewer values |
| Alignment | The larger of the absolute linear and circular correlations between a fitted shape and the concept values |
| Decline | A stage issues no verdict for a cell because a predefined threshold is not met |
| Tier A | The strictest evidence grade for a selected shape. It names the exact shape or only its family |
| Held-out condition | The condition of Tier A that the held-out error of the shape exceeds that of the best alternative by at most 10% |
| Stability condition | The condition of Tier A that the shape keeps the highest evidence under all nine combinations of three seeds and three values of |
| Tiers B to D | Lower evidence grades. Tier B meets the evidence bar and alignment only, Tier C supports only a dimension, and Tier D is a declined verdict |
| Strongly supported comparison | A shape comparison that reaches Tier A and names the exact shape |
| Family | One of three groups of candidate shapes, the circle and the open helix, the torus, the concentric shape, and the ribbon, and the line with the baseline |
| Owned | A shape with , whose evidence gap grows when its related concepts are removed |
| Inherited | A shape with , whose evidence gap does not grow when its related concepts are removed |
| Manifold basis | An orthonormal basis of the directions of that the selected shape occupies |
| Cross features | The two features of the ribbon that multiply the linear feature by the cosine and the sine |
| Ablation | Replacing the part of an activation inside a subspace by its mean |
| Patching | Replacing the part of an activation inside a subspace by that of a donor |
| Donor | A problem with another concept value whose activation is patched in |
| Accuracy change | Change in the share of correctly answered problems under ablation, in percentage points |
| Causal excess | Logit change under an intervention minus the mean change under random subspaces of the same rank |
| Causally active | A cell whose causal excess under ablation of is at least 0.5 logits in absolute value |
| Retention | Mean accuracy change under ablation of divided by that under ablation of |
| Preprocessing setting | One of the three treatments of the activations, off, answer, and norm |
Table 6. Terms.
Extended Introduction Material
Geometric structure in prior work. In three mid-sized LLMs, integer tokens lie on a generalised helix with periods 2, 5, 10, and 100, fitted at the operand tokens and, for the sum, at the last token, and activation patching confirms that the helix is causally implicated in addition and subtraction [27]. In small transformers trained on modular addition, small changes to hyperparameters and initialisation lead to qualitatively different solutions [57]. Character counts in Claude 3.5 Haiku lie on low-dimensional curved manifolds that the model twists to estimate the distance to a line boundary [21]. On the theoretical side, Modell et al. [42] model a concept as a metric space. They show that, if the correspondence between concept and representation is continuous and one-to-one, the representation forms a manifold with the same topology as the concept. If cosine similarity also decreases locally with distance in the concept, the lengths of paths along the manifold recover the distances of the concept. The linear representation hypothesis holds that concepts occupy linear directions [45]. Recent work still labels distance statistics over nearest neighbours a manifold without recovering geodesics, curvature, or topology (Li et al., 2025).
Dependencies among concepts. Dependencies like the one in the running example of § 1 hold by construction throughout the standard written algorithms, and some are exact identities (Appendix § 15).
Design of the pipeline. A subspace shows where a concept is encoded, and the shape inside it is the candidate for the structure that the model computes with, which is why Stage 2 follows Stage 1. The checks of Stage 1 follow the control-task rationale of Hewitt & Liang [25].
Scope of the single-position design. Every intervention overwrites the final = token alone, before the residual stream is passed to the next block. The pipeline does not identify which neurons, Fourier components, attention edges, or sparse-autoencoder features realise a shape, and the Fourier-feature and neuron-heuristic accounts of § 2 are candidate substrates for the same geometry.
Models and Activation Capture
The three tokenisers are variants of byte pair encoding (BPE). GPT-J uses GPT-2 BPE, Llama uses TikToken BPE, and Pythia uses GPT-NeoX BPE. GPT-J and Pythia run attention and the feed-forward layer in parallel, and Llama runs them in sequence. We exclude instruction-tuned variants because instruction tuning changes how a model follows a prompt and would confound the analysis. We stop before the final layers, where the activation at the last token is expected to align mostly with the unembedding direction of the answer token, which reflects readout rather than computation. At each layer we read the residual stream after the transformer block, with prompts padded on the left. This vector is the full state passed to the next block, not the contribution of one block alone, and the convention is the same for both block layouts. The mean norm of the residual stream grows over the five layers by a factor of 3.1 in GPT-J, 8.8 in Llama, and 3.7 in Pythia, and at each layer it is 4.8 to 20.1 times larger in GPT-J and Pythia than in Llama.
Prompts, Tokenisation, and Data
Every problem uses the prompt Output ONLY a number. {a}+{b}= for addition and the same with * for multiplication. Ending the prompt with = fixes the next-token position to where the answer is expected and gives every problem the same final position. We add no trailing newline, because a newline would become the final token in place of =. Without the instruction, GPT-J and Pythia often continue the prompt with further text, and the most probable next token can be punctuation and not the answer. Decoding is greedy, which makes the selection of correctly answered problems deterministic. A problem is usable for a model when its correct answer is a single token under the tokeniser of that model. Every integer up to 520 is a single token for GPT-J, up to 999 for Llama, and up to 530 for Pythia, whether the number follows an empty prefix, a plus sign, a multiplication sign, or an equals sign. Every addition answer lies within all three limits. Some larger products are also single tokens, and the multiplication set consists of the 3,023 pairs whose product is a single token for all three models. A problem is answered correctly when the most probable next token equals the answer token (Table 7). Because every answer is a single token, this equals correctness of the full answer.
Concept Definitions and Related Concepts
Definitions. For an operand we write and , and likewise for . An answer can exceed 99, so its tens digit is . The units partial product is , and the units column sum is the sum of the partial products in the units column, which in two-digit multiplication is the same quantity. The units carry is in addition and in multiplication, so the two tasks share the name but not the quantity. The column sum of a column is the sum of the digits of the operands in that column in addition, and the sum of the partial products that land in that column in multiplication, without the carry that arrives from the column to its right. The running sum of a column is its column sum plus that incoming carry, and the carry out of a column is the running sum divided by 10 and rounded down, which for addition is 1 when the running sum reaches 10. The number of digits of a quantity is its number of decimal digits, one for zero. The magnitude tier of a quantity is zero, one digit, two digits, or three digits. The core concepts of multiplication include the tens digit of and not that of , which has the same distribution by construction and is an extended concept. The ten joint concepts per task consist of four pairs of operand digits, two that bind a carry to another quantity, and four that join two operand digits with a third quantity, one of which is determined by construction. The values of a joint concept are coded in mixed radix with the first concept as the most significant part, so that takes the value and a concept of three parts nests in the same way. The core concepts are single concepts, so the results reported for them do not depend on this order.
Values removed by the floor. Among the core concepts, the sum has 199 values and loses 58 to 78 of them, depending on the model. The units partial product has 37 values and loses 3 or 4. The value 8 of the units carry in multiplication is removed for all three models, with 8, 14, and 11 problems for GPT-J, Llama, and Pythia. The other core concepts keep every value. With a floor of 20 these losses shrink to 39 to 54, 1 or 2, and 1, and with a floor of 50 they grow to 98 to 125, 14 or 15, and 2, while Pythia also loses 1 value of and 5 of . The probe subspaces of the cells of the core concepts that lose values remain verified under all three floors, and we report the later stages at a floor of 30 only.
Related concepts. The lists cover 20 concepts in addition and 26 in multiplication, and Table 8, Table 9, and Table 10 give all of them. Each list holds the concepts from which the standard written algorithm computes its target, and for an operand its digits and for a digit its operand, so that the geometry of an operand is tested against its own digits and that of a digit against its operand. The 62 concepts without a list are the properties of the operands and of the answer, such as parity and magnitude, and the joint concepts.
| Task | Concept | Related concepts |
| Addition | sum | , |
| Addition | units digit of the answer | , , , , units column sum, units carry |
| Addition | tens digit of the answer | , , , , tens column sum, units carry, tens carry |
| Multiplication | units carry | units column sum, units partial product |
| Multiplication | units digit of the answer | , , , , units partial product, units column sum, units carry |
Table 8. Related concepts of five core concepts, listed before any experiment.
| Concept | Related concepts |
a | a_units, a_tens |
b | b_units, b_tens |
a_units, a_tens | a |
b_units, b_tens | b |
answer | a, b |
ans_units | a, b, a_units, b_units, column_sum_units, carry_units |
ans_tens | a, b, a_tens, b_tens, column_sum_tens, carry_units, carry_tens |
ans_hundreds | a, b, carry_tens |
column_sum_units | a_units, b_units |
column_sum_tens | a_tens, b_tens |
carry_units | a_units, b_units, column_sum_units |
carry_tens | a_tens, b_tens, column_sum_tens, carry_units |
running_sum_units | a_units, b_units |
running_sum_tens | a_tens, b_tens, running_sum_units, carry_units |
max_operand, min_operand, operand_diff, operand_abs_diff | a, b |
Table 9. The 20 lists of related concepts of addition, in the names of Table 11.
| Concept | Related concepts |
a | a_units, a_tens |
b | b_units, b_tens |
a_units, a_tens | a |
b_units, b_tens | b |
answer | a, b |
ans_units | a, b, a_units, b_units, partial_product_units, column_sum_units, carry_units |
ans_tens | a, b, a_tens, b_tens, partial_product_a_units_b_tens, partial_product_a_tens_b_units, column_sum_tens, carry_units, carry_tens |
ans_hundreds | a, b, partial_product_a_tens_b_tens, column_sum_hundreds, carry_tens |
partial_product_units | a_units, b_units |
partial_product_a_units_b_tens | a_units, b_tens |
partial_product_a_tens_b_units | a_tens, b_units |
partial_product_a_tens_b_tens | a_tens, b_tens |
column_sum_units | partial_product_units |
column_sum_tens | partial_product_a_units_b_tens, partial_product_a_tens_b_units |
column_sum_hundreds | partial_product_a_tens_b_tens |
carry_units | column_sum_units, partial_product_units |
carry_tens | column_sum_tens, carry_units |
running_sum_units | partial_product_units |
running_sum_tens | partial_product_a_units_b_tens, partial_product_a_tens_b_units, running_sum_units, carry_units |
running_sum_hundreds | partial_product_a_tens_b_tens, running_sum_tens, carry_tens |
max_operand, min_operand, operand_diff, operand_abs_diff | a, b |
Table 10. The 26 lists of related concepts of multiplication, in the names of Table 12.
All concepts. Table 11 and Table 12 list all 108 concepts by the names used in our data, in which ans stands for the answer and a double underscore joins the parts of a joint concept.
| Group | Concepts |
| Core (6) | a, b, answer, ans_units, ans_tens, carry_units |
| Extended single (35) | a_eq_b, a_is_zero, a_magnitude_tier, a_num_digits, a_parity, a_tens, a_units, ans_ends_in_zero, ans_hundreds, ans_is_zero, ans_magnitude_tier, ans_num_digits, ans_parity, b_is_zero, b_magnitude_tier, b_num_digits, b_parity, b_tens, b_units, both_one, both_zero, carry_tens, column_sum_tens, column_sum_units, either_one, either_zero, larger_operand, max_operand, min_operand, operand_abs_diff, operand_diff, parity_match, parity_xor, running_sum_tens, running_sum_units |
| Joint (10) | a_tens_b_tens, a_tens_b_units, a_units_b_tens, a_units_b_units, carry_units_ans_units, carry_units_column_sum_units, a_tens_b_tens_ans_tens, a_tens_b_tens_carry_units, a_units_b_units_ans_tens, a_units_b_units_ans_units |
Table 11. The 51 concepts of addition.
| Group | Concepts |
| Core (6) | a_units, a_tens, b_units, partial_product_units, carry_units, ans_units |
| Extended single (41) | a, a_eq_b, a_is_zero, a_magnitude_tier, a_num_digits, a_parity, ans_ends_in_zero, ans_hundreds, ans_is_zero, ans_magnitude_tier, ans_num_digits, ans_parity, ans_tens, answer, b, b_is_zero, b_magnitude_tier, b_num_digits, b_parity, b_tens, both_one, both_zero, carry_tens, column_sum_hundreds, column_sum_tens, column_sum_units, either_one, either_zero, larger_operand, max_operand, min_operand, operand_abs_diff, operand_diff, parity_match, parity_xor, partial_product_a_tens_b_tens, partial_product_a_tens_b_units, partial_product_a_units_b_tens, running_sum_hundreds, running_sum_tens, running_sum_units |
| Joint (10) | a_tens_b_tens, a_tens_b_units, a_units_b_tens, a_units_b_units, carry_units_ans_units, carry_units_column_sum_units, a_tens_b_tens_ans_tens, a_tens_b_tens_carry_units, a_units_b_units_ans_tens, a_units_b_units_partial_product_units |
Table 12. The 57 concepts of multiplication.
Cell Counts and Exclusion Rules
The cells divide into 2,295 for the 51 addition concepts and 2,565 for the 57 multiplication concepts. Three rules apply in order. First, the floor drops values, and a cell with fewer than two remaining values is removed (Appendix § 15 gives the values removed). Second, a cell is removed when the permutation test on the singular values keeps no direction, so that . Third, a cell is flagged when its most frequent value has more than three times as many problems as its least frequent value, and flagged cells remain in the analysis. Stage 1 counts are given for the setting without preprocessing, and the later stages pool the three settings (Table 13).
Method Details
Probe subspace. Let be the number of values of the concept that remain in the cell, with and taken over their problems. The matrix decomposed in Stage 1 is , with rows . With this weighting, equals the between-class scatter , so the right singular vectors of are the directions along which the centroids vary most. Centring forces the last singular value to zero, which leaves at most directions.
Each permutation shuffles while keeping every , recomputes the centroids, and records the squared singular values. With the observed squared singular value of rank and the corresponding percentile of its permuted counterpart, the kept rank is
so that when the first singular value fails. This step does not whiten the activations. The LDA that follows works on the coordinates inside . It regularises the total scatter with Ledoit–Wolf shrinkage [32] when , and otherwise adds times the mean eigenvalue to the diagonal. A direction is marked as confirmed, a mark that does not change , only if it passes both the rule of Equation (4), applied to the discriminant eigenvalues under 1,000 permutations of , and the one-standard-error rule on the accuracy of a nearest-neighbour classifier under five-fold cross-validation. As a consistency check we repeat the LDA in the full -dimensional space with oracle approximating shrinkage (OAS, Chen et al., 2010) and compare the leading directions of the two analyses. An absolute cosine of at least 0.9 counts as agreement, and one from 0.7 to 0.9 as partial agreement. The eigenvalues of the full-space analysis are not interpreted, because with between 0.7 and 2.4 they are inflated by sampling noise. The LDA and this check remove no cell (Appendix F).
Checks on the probe subspace
The three checks measure what the probe subspace leaves behind, what it shares with other concepts, and what it keeps of the geometry. For the first, let hold the parts of the activations outside , and let be the largest eigenvalue of their covariance, computed from the leading 500 components of a randomised SVD. If the residual were noise of the same variance in every direction, sampling alone would produce eigenvalues up to the Marčenko–Pastur edge [39]
where is the ratio of the dimension of the residual space to and is the trace of the residual covariance divided by that dimension. The check reports the ratio . A ratio above one shows a direction outside whose variance exceeds what noise produces, that is, structure in the residual stream that the probe of this concept does not use. The law describes the limit of large and large dimension at a fixed , and its edge is a usable reference when . Multiplication cells lie above this range, because their correctly answered problems number fewer than the residual dimension, so for them we report the ratio and draw no conclusion from it. For the second check, let and be the probe bases of two concepts at the same model, layer, and setting. The smallest principal angle between two subspaces is the angle between their closest pair of unit vectors, and it is
where denotes the largest singular value, so that when the subspaces share a direction and when every direction of one is orthogonal to the other. In dimensions, two random subspaces of small rank are nearly orthogonal, so under the null hypothesis lies close to at the median rank of nine and lower at the largest ranks, which the pairs drawn at the same ranks reflect. We draw 1,000 pairs of random subspaces of the same ranks, each the orthonormal factor of a Gaussian matrix, and take the fifth percentile of their smallest principal angles. The pair of concepts is superposed when the observed lies more than below this percentile. The margin of keeps a pair that is only slightly less orthogonal than chance from being marked, and the FDR is controlled over all pairs of concepts. The superposition rate of a model and task is the share of marked pairs, averaged over its cells. For the third check, the squared distance between two activations splits exactly into its parts inside and outside ,
because has orthonormal columns, and we verify the identity to a relative tolerance of over all pairs. The check then takes the Spearman correlation between the distances and the distances of the projections over all pairs of problems and reports its deviation from one. A correlation of one means that the projection keeps the order of every pairwise distance.
Step 2a, period. The rows of , ordered by value, expressed in the coordinates of , and centred, form one sequence per coordinate. We compute the periodogram of every sequence, in which frequency corresponds to a period of steps of the ordered values. A period is accepted when the strongest frequencies of the two leading coordinates are within one frequency bin or, failing that, when the power of the strongest frequency, summed over the coordinates, is at least twice that of the second. Otherwise the step declines and reports an inconsistent period. A circle is reported when the accepted frequency in two coordinates holds at least 0.3 of the power of all frequencies and coordinates, when both coordinates pass a permutation test at the 0.01 level, and when the second squared singular value of the matrix of these sequences reaches 0.3 of the first. The last condition rejects a line, which leaks power into the lowest frequency. A helix is reported instead when a linear trend outside the plane of the circle is also significant. Each permutation test compares the observed power with the maximum over all frequencies under shuffles of the order of the values, which accounts for the search over periods, and the FDR over cells is held at 0.05. Cells with fewer than four values are not analysed. The accepted period seeds the candidates , , , and of Step 2c, and no other period enters Step 2c.
Step 2b, spread. The covariance of a value is estimated in the coordinates of as the sample covariance when , with the Ledoit–Wolf estimator when , and with OAS when . When the values of a cell call for different estimators, all values use the one with the most shrinkage, so that the covariances are comparable. The spread of (1) is the square root of , the variance per coordinate averaged over the coordinates. The step also tests whether the arrangement of the centroids survives once spread is taken into account. It computes the Mahalanobis distance between every pair of centroids under the pooled covariance of the pair, plus a ridge of times its mean eigenvalue, and takes the Spearman correlation between these distances and the Euclidean ones. The arrangement is confirmed when this correlation is at least 0.85 and the lower end of its 95% bootstrap interval, from 1,000 resamples, is at least 0.70. A permutation test, held at an FDR of 0.05, guards the result. The outcome of this test is a diagnostic and does not enter the evidence grade.
Step 2c, evidence. We stack the coordinates as , the features (Table 14) as with rows , where counts the features with the constant, and the noise variances as the diagonal matrix with entries . The weights are . Integrating out of Equation 1, conditional on the basis of Stage 1, the noise variances of Step 2b, and the candidate periods of Step 2a, leaves the columns independent and Gaussian, so that
The covariance is never formed. By the Woodbury identity and the matrix determinant lemma, every term follows from the matrix , which also gives the posterior mean used in Stage 4, and each fit takes seconds on a CPU. For the ribbon, the two cross features are multiplied by . This gives their weights the scaled prior variance of Section 4, and Equation 8 then holds as written with the scaled features. The prior variance is chosen on a logarithmic grid of 13 points from to by maximising the evidence of the best shape. Because this choice is made on the best shape, the stability condition of Tier A repeats the comparison at three fixed values of . The evidence of a periodic shape is the average over the candidate periods of Step 2a, for candidates, and the torus is averaged in the same way over the six unordered pairs of distinct candidates. For the circle and the open helix, whose period is their only free parameter, a Laplace term is added to each candidate when at least three are evaluated. With the candidates ordered by period, , and , the second difference estimates the curvature of the log evidence in the period at an interior candidate, and each candidate receives the term before the average, with floored at , which also applies to the two end candidates, where no second difference exists. The term is the Laplace estimate of the width of the evidence peak, which the average over a few candidates does not carry. Because Step 2a can decline, and because the baseline and the line have no period, a candidate set does not favour periodic shapes. The line follows the linear account of number representations [59], the circle follows the modular addition solutions of small transformers [35, 43], and the open helix is the single-period form of the generalised helix of Kantamneni & Tegmark [27]. The torus is motivated by residues modulo 10 and the ribbon by the two-digit range, which carries a linear trend together with a periodic one. The concentric shape is a radial variant of the circle, included for completeness. Figure 3 draws the circle, the open helix, the ribbon, and the torus as linear images of their feature maps. Because Equation (8) sums over the coordinates of a cell and each term is the log density of the values of one coordinate, a gap depends on and on , and the pipeline compares gaps only within a cell.
Figure 3. The circle, the open helix, the ribbon, and the torus of Table 14 as linear images of their feature maps for the values 0 to 29, with period 10, and periods 10 and 5 for the torus, drawn with one fixed choice of and viewed in three of the coordinates. Dots mark the values, and values ten apart coincide on the closed shapes. The pipeline fits , so an observed shape is a rotated, scaled, and possibly flattened image of these curves.
Evidence grades. The alignment of a shape is , where is the Pearson correlation between the concept values and the position of each problem along the fitted linear feature, for the row of that belongs to the feature , and is the circular correlation between and the angle of each problem in the plane of the fitted cosine and sine features. The held-out error is the mean squared error of the coordinates under five-fold cross-validation, with the weights refitted on the training folds and the period and the noise held, and the error of the best shape may exceed the smallest error of the other shapes by a factor of at most 1.10. The basis of Stage 1, the candidate periods of Step 2a, the noise variances of Step 2b, and are estimated on all problems, so the check validates the fitted weights conditional on the estimated representation and hyperparameters and does not test the pipeline as a whole on unseen problems. The best shape is separated from the runner-up when its log evidence, less 1.96 times a standard error from 100 bootstrap resamples of the evidence under a 5% jitter of the period, leads that of the runner-up by the bar of selection, 5 nats, or 10 nats for six or fewer values, and Tier A then names the exact shape. The stability condition uses the seeds 42, 43, and 44 with equal to 0.1, 1, and 10. The three families are the shapes with one periodic feature, the circle and the open helix, the shapes with two periodic or radial features, the torus, the concentric shape, and the ribbon, and the line with the baseline. The evidence of a family is the average of the evidence of its members, logsumexp over the members less the log of their number, and Tier A names only the family when the shape is not separated from its runner-up but the family evidence leads the other families by the bar. Below Tiers A and B, Tier C supports only the dimension of the structure and no shape, and Tier D is the declined verdict. As a guard against multiple testing, every Tier A verdict is also compared with 10,000 permutations of the values, on the statistic , with the FDR controlled at 0.10. Repeating this control at levels of 0.05 and 0.20 leaves the ribbon and the torus as the dominant shapes of Tier A.
Removal of related concepts. The bases of the related concepts are stacked and decomposed, and singular vectors whose singular values fall below times the largest are dropped, which removes redundant directions. In the running example spans the subspaces of the operands, of their units digits, of the units partial product, and of the units carry, the units column sum being the same quantity as the partial product (Appendix § 15). Because only coordinates inside enter the fit, the removal is computed as , where is the vector of ones, which needs the small matrix and never a projection. Row of is the vector of Section 4. The gap is that of the shape selected in Stage 2, refitted. Let be the removed part and , with . Because the determinant terms of Equation (8) do not depend on the coordinates, and with , the period, and held at their Stage 2 values,
The first term is the advantage of the selected shape over the baseline on the removed part, and the second is the interaction between the removed part and what remains. When the span of lies inside the probe subspace, the removal zeroes whole coordinates, the second term is zero, and is the first term alone. In general both terms enter, and the verdict is the sign of their sum, with when the removal takes nothing from . Three diagnostics are recorded and do not enter the verdict. They are the share of variance that the removal takes away, whether the family of the selected shape changes, and the ratio of the alignment after and before the removal. A shape whose evidence gap falls is inherited even if it would still be selected, and so is a shape whose coordinates the removal leaves unchanged, with , so the rule is strict at the boundary. The verdict has two limits. The removal can take away directions that belong to the concept itself, which would produce a false inherited verdict, and a related concept missing from the list would produce a false owned verdict.
Interventions. The manifold basis is computed by a QR decomposition. It spans the row space of expressed in the full activation space, and not the latent parametrisation of the shape. The constant feature is absorbed into the mean, so the bound on the rank is the number of non-constant features of the selected shape, five for the ribbon and four for the torus. Each random control comes from the QR decomposition of a Gaussian matrix with rows and as many columns as , drawn anew for each cell and each type of intervention, with separate seeds for ablation and patching. A random subspace of this rank holds a very small share of the variance of an activation, so the control matches the rank of an intervention and not the variance that its subspace holds. The pairs of donor and recipient are drawn with a fixed seed, and all interventions run in bfloat16 precision in batches of 16. Stages 1 to 3 use all problems of a cell, and the interventions evaluate at most 256 of them, with the shares of problems answered correctly with and without ablation both read in this pass. The batched bfloat16 pass does not reproduce every answer of the selection pass, so the share without ablation lies below one hundred percent in a few cells, and a problem answered correctly at selection and incorrectly at this baseline can be answered correctly again under ablation, which is the source of the two positive values of Table 20. A cell receives the verdicts of Stages 3 and 4 only if its also has positive rank.
| Model | Task | 4 | 8 | 14 or 16 | 20 or 24 | 24 or 28 |
| GPT-J | addition | |||||
| GPT-J | multiplication | |||||
| Llama | addition | |||||
| Llama | multiplication | |||||
| Pythia | addition | |||||
| Pythia | multiplication |
Table 20. Accuracy change under ablation of and of , in percentage points and written as , by model, task, and layer, over the cells of Figure 2.
| Quantity | Value |
| Stage I | |
| Minimum number of problems per value | 30 |
| Flag for imbalance, most against least frequent value | more than 3 times |
| Permutations for the rank of the probe subspace, and level | 1,000, 99th percentile |
| Permutations and folds for the LDA directions | 1,000, 5 folds |
| Ledoit–Wolf shrinkage of the total scatter | when |
| Diagonal load otherwise | times the mean eigenvalue |
| Agreement of the two LDAs | absolute cosine , partial from 0.7 |
| Components of the residual decomposition | 500 |
| Reliable range of the Marčenko–Pastur comparison | |
| Random pairs, percentile, margin, and FDR for superposition | 1,000, fifth, , 0.05 |
| Tolerance of the distance identity | |
| Stage 2 | |
| Step 2a shuffles, level, and FDR | 1,000, 0.01, 0.05 |
| Step 2a share of power and ratio of squared singular values | each |
| Step 2a agreement of periods, and margin of power | one bin, factor 2.0 |
| Step 2a minimum number of values | 4 |
| Step 2b estimator by | sample from 10, Ledoit–Wolf from 5, OAS below |
| 5 | |
| Step 2b ridge on the pooled covariance | times the mean eigenvalue |
| Step 2b permutations, bootstrap resamples, and FDR | 1,000, 1,000, 0.05 |
| Step 2b confirmed | correlation , lower end |
| Evidence bar for selection | 5 nats, 10 nats for six or fewer values |
| Alignment, and held-out error factor | , |
| Seeds and prior variances of the stability condition | 42, 43, 44 and 0.1, 1, 10 |
| Grid for | 13 points from to |
| Prior variance scale of the cross features of the ribbon | 0.1 |
| Permutations, weight of the alignment, and FDR for Tier A | 10,000, , 0.10 |
| Stages 3 and 4 | |
| Tolerance when orthonormalising the related concepts | |
| Ownership rule | owned when |
| Problems per cell for ablation, donor pairs, and random subspaces | at most 256, 100, 5 |
| Seeds of the random subspaces, ablation and patching | 42 and 49 |
| Causally active, and stricter level | 0.5 logits, 1.0 logits |
Table 15. Fixed thresholds and settings of the pipeline, identical for all cells.
Results Details
Stage 1 and Stage 4 by model and task. The superposition rate varies little within a model and task, with an SD of at most 0.046 over the cells of a row (Table 16). On the core concepts without preprocessing, 179 of the 180 cells are decodable, and the one cell that is not decodable lies in Pythia addition. Under patching, the logit of the donor answer rises in every model and task, and keeps from 0.43 to 0.77 of the effect of .
| Model | Task | SD | Mean | Maximum | Baseline | Circle | Open helix | ||
| GPT-J | addition | 0.046 | 6.0 | 11.9 | 0 | 0 | 0 | +2.65 | +1.14 |
| GPT-J | multiplication | 0.014 | 10.0 | 17.3 | 4 | 2 | 1 | +1.57 | +1.19 |
| Llama | addition | 0.027 | 7.8 | 27.4 | 1 | 0 | 0 | +1.68 | +0.77 |
| Llama | multiplication | 0.020 | 8.9 | 13.0 | 5 | 0 | 0 | +0.74 | +0.52 |
| Pythia | addition | 0.046 | 6.9 | 14.1 | 3 | 0 | 0 | +3.09 | +1.62 |
| Pythia | multiplication | 0.013 | 10.8 | 19.1 | 0 | 2 | 1 | +1.37 | +1.06 |
Table 16. Detail by model and task. SD is the standard deviation of the superposition rate over the cells of the row. The ratio is given as a mean and a maximum over the cells, and the multiplication rows lie outside the range in which the edge is a usable reference (Appendix § 17). Other shapes counts the cells in which the baseline, the circle, or the open helix has the highest evidence. Patching gives the mean causal excess under patching, in logits, on the core concepts over the layers and the three settings.
Results by setting. Table 17 gives the shape grades, the ownership verdicts, the retention, and the cells that pass all four criteria under each preprocessing setting. The owned share and the retention change little across the settings, and the cells that pass all four criteria split 16, 14, and 16. Resampling the 122 concepts of the six pairs of model and task 2,000 times, with all layers and settings of a concept drawn together, gives a 95% interval of 0.54 to 0.67 for the retention, 13% to 25% for the owned share, and 37% to 47% for the share of causally active cells among the 1,330. Within the extended concepts the shares of causally active cells are 41% for owned and 40% for inherited shapes, at layers 14 to 20 they are 69% and 66%, and at layers 24 and 28 they are 41% and 57%, so the parity of Section 5 holds where the effects are largest and inherited shapes lead only at the deepest layers.
A cell that passes all four criteria. The tens column sum of addition, , in GPT-J at layer 20 without preprocessing has 19 values over 8,415 problems, and its related concepts are the two tens digits (Table 9). Its shape is the ribbon in a strongly sloped comparison, with an evidence gap of 29,580 nats over the baseline and an alignment of 0.65. Removing the subspaces of the two digits takes 0.70 of the variance of its coordinates, and the gap rises to 55,758 nats, so nats and the cell is owned. Ablating its manifold basis lowers the accuracy by 35.5 points against 56.6 under ablation of , a retention of 0.63 in this cell, the causal excess is logits against , and under patching it is against , with the share answered correctly without ablation in the intervention pass as 99.6%. Figure 4 shows its centroids on the fitted ribbon and after the removal. We read the cell as a code for the column sum that the two digits it is computed from do not account for, on which the output depends. The running example of Section 5 gives the inherited outcome that the arithmetic predicts, and this cell gives the owned outcome that it does not.
Shapes of the core concepts. Without preprocessing the ribbon has the highest evidence for 24 of the 33 pairs of core concept and model, the torus for 8, and the baseline for 1 (Table 19). The units carry of addition receives no shape verdict in this setting.
| Addition | ||||||
| Model | Answer units | Answer tens | Sum | |||
| GPT-J | ribbon | ribbon | torus | torus | ribbon | |
| Llama | ribbon | ribbon | ribbon | ribbon | ribbon | |
| Pythia | ribbon | baseline | torus | torus | ribbon | |
| Multiplication | ||||||
| Model | Units of | Tens of | Units of | Partial product | Carry | Answer units |
| GPT-J | ribbon | ribbon | ribbon | torus | ribbon | ribbon |
| Llama | ribbon | ribbon | ribbon | ribbon | ribbon | ribbon |
| Pythia | ribbon | ribbon | torus | torus | ribbon | torus |
Table 19. Shape with the highest evidence for each core concept and model without preprocessing, as one shape per concept over its five layers.
Causal effect by layer. In the values plotted in Figure 2 (Table 20), the change under ablation of exceeds that under ablation of in magnitude by at most 0.3 points, and the only positive values are +0.1 and +0.2 points, for GPT-J multiplication at layer 4, from problems that the baseline of the intervention pass answers incorrectly (Appendix § 17).
References
- [1]Guillaume Alain and Yoshua Bengio. Understanding intermediate layers using linear classifier probes. arXiv preprint arXiv:1610.01644, 2016. URL https://arxiv.org/abs/1610.01644.
- [2]Xiaoyan Bai, Itamar Pres, Yuntian Deng, Chenhao Tan, Stuart Shieber, Fernanda Viégas, Martin Wattenberg, and Andrew Lee. Why can’t transformers learn multiplication? Reverse-engineering reveals long-range dependency pitfalls. arXiv preprint arXiv:2510.00184, 2025. URL https://arxiv.org/abs/2510.00184.
- [3]Yonatan Belinkov. Probing classifiers: Promises, shortcomings, and advances. Computational Linguistics, 48(1):207–219, 2022. URL https://arxiv.org/abs/2102.12452.
- [4]Nora Belrose, Igor Ostrovsky, Lev McKinney, Zach Furman, Logan Smith, Danny Halawi, Stella Biderman, and Jacob Steinhardt. Eliciting latent predictions from transformers with the tuned lens. arXiv preprint arXiv:2303.08112, 2023a. URL https://arxiv.org/abs/2303.08112.
- [5]Nora Belrose, David Schneider-Joseph, Shauli Ravfogel, Ryan Cotterell, Edward Raff, and Stella Biderman. LEACE: Perfect linear concept erasure in closed form. In Advances in Neural Information Processing Systems (NeurIPS), 2023b. URL https://arxiv.org/abs/2306.03819.
- [6]Yoav Benjamini and Yosef Hochberg. Controlling the false discovery rate: A practical and powerful approach to multiple testing. Journal of the Royal Statistical Society: Series B (Methodological), 57(1):289–300, 1995.DOI
- [7]Stella Biderman, Hailey Schoelkopf, Quentin Anthony, Herbie Bradley, Kyle O’Brien, Eric Hallahan, Mohammad Aflaq Khan, Shivanshu Purohit, USVSN Sai Prashanth, Edward Raff, Aviya Skowron, Lintang Sutawika, and Oskar van der Wal. Pythia: A suite for analyzing large language models across training and scaling. In International Conference on Machine Learning (ICML), pp. 2397–2430. PMLR, 2023. URL https://arxiv.org/abs/2304.01373.
- [8]Marc Canby, Adam Davies, Chirag Rastogi, and Julia Hockenmaier. How reliable are causal probing interventions? In Proceedings of the International Joint Conference on Natural Language Processing and the Asia-Pacific Chapter of the Association for Computational Linguistics (IJCNLP-AACL), 2025. URL https://arxiv.org/abs/2408.15510.
- [10]Arthur Conmy, Augustine N. Mavor-Parker, Aengus Lynch, Stefan Heimersheim, and Adrià Garriga-Alonso. Towards automated circuit discovery for mechanistic interpretability. In Advances in Neural Information Processing Systems (NeurIPS), 2023. URL https://arxiv.org/abs/2304.14997.
- [11]Yanai Elazar, Shauli Ravfogel, Alon Jacovi, and Yoav Goldberg. Amnesic probing: Behavioral explanation with amnesic counterfactuals. Transactions of the Association for Computational Linguistics, 9:160–175, 2021. URL https://arxiv.org/abs/2006.00995.
- [12]Nelson Elhage, Tristan Hume, Catherine Olsson, Nicholas Schiefer, Tom Henighan, Shauna Kravec, Zac Hatfield-Dodds, Robert Lasenby, Dawn Drain, Carol Chen, Roger Grosse, Sam McCandlish, Jared Kaplan, Dario Amodei, Martin Wattenberg, and Christopher Olah. Toy models of superposition. Transformer Circuits Thread, 2022. URL https://arxiv.org/abs/2209.10652.
- [13]Joshua Engels, Eric J. Michaud, Isaac Liao, Wes Gurnee, and Max Tegmark. Not all language model features are one-dimensionally linear. In International Conference on Learning Representations (ICLR), 2025. URL https://arxiv.org/abs/2405.14860.
- [14]Sheridan Feucht, Tal Haklay, Usha Bhalla, Daniel Wurgaf, Can Rager, Raphaël Sarfati, Jack Merullo, Thomas McGrath, Owen Lewis, Ekdeep Singh Lubana, Thomas Fel, and Atticus Geiger. Arithmetic in the wild: Llama uses base-10 addition to reason about cyclic concepts. arXiv preprint arXiv:2605.01148, 2026. URL https://arxiv.org/abs/2605.01148.
- [16]Dejing Fu, Tianyi Zhou, Mikhail Belkin, Vatsal Sharan, and Robin Jia. Convergent evolution: How different language models learn similar number representations. In Conference on Language Modeling (COLM), 2026. URL https://arxiv.org/abs/2604.20817.
- [17]Atticus Geiger, Duilgur Ibeling, Amir Zur, Maheep Chaudhary, Sonakshi Chauhan, Jing Huang, Aryaman Arora, Zhengxuan Wu, Noah Goodman, Christopher Potts, and Thomas Icard. Causal abstraction: A theoretical foundation for mechanistic interpretability. Journal of Machine Learning Research, 26(83):1–64, 2025. URL https://arxiv.org/abs/2301.04709.
- [18]Ian Goodfellow, Yoshua Bengio, and Aaron Courville. Deep Learning. MIT Press, 2016.
- [20]Wes Gurnee and Max Tegmark. Language models represent space and time. In International Conference on Learning Representations (ICLR), 2024. URL https://arxiv.org/abs/2310.02207.
- [21]Wes Gurnee, Emmanuel Ameisen, Isaac Kauvar, Julius Tarng, Adam Pearce, Chris Olah, and Joshua Batson. When models manipulate manifolds: The geometry of a counting task. Transformer Circuits Thread, 2025. URL https://arxiv.org/abs/2601.04480.
- [22]Michael Hanna, Sandro Pezzelle, and Yonatan Belinkov. Have faith in faithfulness: Going beyond circuit overlap when finding model mechanisms. In Conference on Language Modeling (COLM), 2024. URL https://arxiv.org/abs/2403.17806.
- [23]Søren Hauberg. Only Bayes should learn a manifold (on the estimation of differential geometric structure from data). arXiv preprint arXiv:1806.04994, 2018. URL https://arxiv.org/abs/1806.04994.
- [24]Stefan Heimersheim and Neel Nanda. How to use and interpret activation patching. arXiv preprint arXiv:2404.15255, 2024. URL https://arxiv.org/abs/2404.15255.
- [25]John Hewitt and Percy Liang. Designing and interpreting probes with control tasks. In Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing and the 9th International Joint Conference on Natural Language Processing (EMNLP-IJCNLP), pp. 2733–2743, 2019. URL https://arxiv.org/abs/1909.03368.
- [26]Shruti Joshi, Aaron Mueller, David Klindt, Wieland Brendel, Patrik Reizinger, and Dhanya Sridhar. Causality is key for interpretability claims to generalise. arXiv preprint arXiv:2602.16698, 2026. URL https://arxiv.org/abs/2602.16698.
- [27]Subhash Kantamneni and Max Tegmark. Language models use trigonometry to do addition. arXiv preprint arXiv:2502.00873, 2025. URL https://arxiv.org/abs/2502.00873.
- [28]Dhruva Karkada, Daniel J. Korchinski, Andres Nava, Matthieu W’yart, and Yasaman Bahri. Symmetry in language statistics shapes the geometry of model representations. In International Conference on Machine Learning (ICML), 2026. URL https://arxiv.org/abs/2602.15029.
- [29]Robert E. Kass and Adrian E. Raftery. Bayes factors. Journal of the American Statistical Association, 90(430):773–795, 1995. Preprint. Under review.
- [30]János Kramár, Tom Lieberum, Rohin Shah, and Neel Nanda. AtP*: An efficient and scalable method for localizing LLM behaviour to components. arXiv preprint arXiv:2403.00745, 2024. URL https://arxiv.org/abs/2403.00745.
- [31]Neil Lawrence. Probabilistic non-linear principal component analysis with Gaussian process latent variable models. Journal of Machine Learning Research, 6(60):1783–1816, 2005.DOI
- [32]Olivier Ledoit and Michael Wolf. A well-conditioned estimator for large-dimensional covariance matrices. Journal of Multivariate Analysis, 88(2):365–411, 2004.
- [33]Amit Arnold Levy and Mor Geva. Language models encode numbers using digit representations in base 10. In Proceedings of the 2025 Conference of the Nations of the Americas Chapter of the Association for Computational Linguistics (NAACL), Volume 2: Short Papers, pp. 385–395, 2025. URL https://arxiv.org/abs/2410.11781.
- [35]Ziming Liu, Ouail Kitouni, Niklas Nolte, Eric J. Michaud, Max Tegmark, and Mike Williams. Towards understanding grokking: An effective theory of representation learning. In Advances in Neural Information Processing Systems (NeurIPS), 2022. URL https://arxiv.org/abs/2205.10343.
- [36]David J. C. MacKay. Bayesian interpolation. Neural Computation, 4(3):415–447, 1992.
- [37]Aleksandar Makelov, Georg Lange, Atticus Geiger, and Neel Nanda. Is this the subspace you are looking for? An interpretability illusion for subspace activation patching. In International Conference on Learning Representations (ICLR), 2024. URL https://arxiv.org/abs/2311.17030.
- [38]Siddarth Mamidanna, Daking Rai, Ziyu Yao, and Yilun Zhou. All for one: LLMs solve mental math at the last token with information transferred from other tokens. In Proceedings of the 2025 Conference on Empirical Methods in Natural Language Processing (EMNLP), 2025. URL https://arxiv.org/abs/2509.09650.
- [39]V. A. Marčenko and L. A. Pastur. Distribution of eigenvalues for some sets of random matrices. Mathematics of the USSR-Sbornik, 1(4):457–483, 1967.DOI
- [41]Alexander Modell. Probing for representation manifolds in superposition. arXiv preprint arXiv:2605.18537, 2026. URL https://arxiv.org/abs/2605.18537.
- [42]Alexander Modell, Patrick Rubin-Delanchy, and Nick Whiteley. The origins of representation manifolds in large language models. arXiv preprint arXiv:2505.18235, 2025. URL https://arxiv.org/abs/2505.18235.
- [43]Neel Nanda, Lawrence Chan, Tom Lieberum, Jess Smith, and Jacob Steinhardt. Progress measures for grokking via mechanistic interpretability. In International Conference on Learning Representations (ICLR), 2023. URL https://arxiv.org/abs/2301.05217.
- [44]Yaniv Nikankin, Anja Reusch, Aaron Mueller, and Yonatan Belinkov. Arithmetic without algorithms: Language models solve math with a bag of heuristics. In International Conference on Learning Representations (ICLR), 2025. URL https://arxiv.org/abs/2410.21272.
- [45]Kiho Park, Yo Joong Choe, and Victor Veitch. The linear representation hypothesis and the geometry of large language models. In Proceedings of the 41st International Conference on Machine Learning (ICML), volume 235, pp. 39643–39666. PMLR, 2024. URL https://arxiv.org/abs/2311.03658.
- [46]Kiho Park, Yo Joong Choe, Yibo Jiang, and Victor Veitch. The geometry of categorical and hierarchical concepts in large language models. In International Conference on Learning Representations (ICLR), 2025. URL https://arxiv.org/abs/2406.01506.
- [47]Shauli Ravfogel, Yanai Elazar, Hila Gonen, Michael Twiton, and Yoav Goldberg. Null it out: Guarding protected attributes by iterative nullspace projection. In Proceedings of the 58th Annual Meeting of the Association for Computational Linguistics (ACL), pp. 7237–7256, 2020. URL https://arxiv.org/abs/2004.07667.
- [48]Raphaël Sarfati, Eric Bigelow, Daniel Wurgaf, Siddharth Boppana, Jack Merullo, Atticus Geiger, Owen Lewis, Tom McGrath, and Ekdeep Singh Lubana. The shape of beliefs: Geometry, dynamics, and interventions along representation manifolds of language models’ posteriors. arXiv preprint arXiv:2602.02315, 2026. URL https://arxiv.org/abs/2602.02315.
- [49]Aryan Sharma, Cutter Dawes, and Shivam Raval. Dissociating decodability and causal use in bracket-sequence transformers. arXiv preprint arXiv:2604.22128, 2026. URL https://arxiv.org/abs/2604.22128.
- [50]Alessandro Stolfo, Yonatan Belinkov, and Mrinmaya Sachan. A mechanistic interpretation of arithmetic reasoning in language models using causal mediation analysis. In Proceedings of the 2023 Conference on Empirical Methods in Natural Language Processing (EMNLP), 2023. URL https://arxiv.org/abs/2305.15054.
- [51]Joshua B. Tenenbaum, Vin de Silva, and John C. Langford. A global geometric framework for nonlinear dimensionality reduction. Science, 290(5500):2319–2323, 2000.pubmed.ncbi.nlm.nih.gov/11125149
- [52]Ben Wang and Aran Komatsuzaki. GPT-J-6B: A 6 billion parameter autoregressive language model. https://github.com/kingoflolz/mesh-transformer-jax, 2021.
- [53]Kevin Wang, Alexandre Variengien, Arthur Conmy, Buck Shlegers, and Jacob Steinhardt. Interpretability in the wild: a circuit for indirect object identification in GPT-2 small. In International Conference on Learning Representations (ICLR), 2023. URL https://arxiv.org/abs/2211.00593.
- [54]Liuyuan Wen, Xun Zhu, Lihao Huang, Wenbin Li, and Yang Gao. The shape of addition: Geometric structures of arithmetic in large language models. In International Conference on Machine Learning (ICML), 2026. URL https://arxiv.org/abs/2606.03645.
- [56]Wei Zhang, Chaoqun Wan, Yonggang Zhang, Yiu-ming Cheung, Xinmei Tian, Xu Shen, and Jieping Ye. Interpreting and improving large language models in arithmetic calculation. In International Conference on Machine Learning (ICML), 2024. URL https://arxiv.org/abs/2409.01659.
- [57]Ziqian Zhong, Ziming Liu, Max Tegmark, and Jacob Andreas. The clock and the pizza: Two stories in mechanistic explanation of neural networks. In Advances in Neural Information Processing Systems (NeurIPS), 2023. URL https://arxiv.org/abs/2306.17844.
- [58]Tianyi Zhou, Deqing Fu, Vatsal Sharan, and Robin Jia. Pre-trained large language models use Fourier features to compute addition. In Advances in Neural Information Processing Systems (NeurIPS), 2024. URL https://arxiv.org/abs/2406.03445.
- [59]Fangwei Zhu, Damai Dai, and Zhifang Sui. Language models encode the value of numbers linearly. In Proceedings of the 31st International Conference on Computational Linguistics (COLING), pp. 693–709, 2025. URL https://arxiv.org/abs/2401.03735.



