MATH · IN · MODELS

ICL answer tokens are affine functions in a PCA subspace

measured in 1 paper

Lee & Vijayan collect residual-stream activations from six pretrained LLMs across six in-context relational tasks and reduce each layer to 30 principal components [lee-vijayan-2026-functional-subspace-vector-algebra] For each layer and component an answer token's projection is fit as an affine function of query and separator projections (a = alpha*q + beta*s + gamma) [lee-vijayan-2026-functional-subspace-vector-algebra] R^2 is high along a small, largely layer-consistent subset of components, rising in later layers, so ICL answer prediction reduces to an affine operation there [lee-vijayan-2026-functional-subspace-vector-algebra] Query, separator, and answer tokens form well-separated clusters in the top-3 high-R^2 component subspace across models and tasks [lee-vijayan-2026-functional-subspace-vector-algebra] Projections onto the highest-R^2 component differ significantly between correct and incorrect predictions (p<0.05) in mid/late layers, with no causal intervention performed [lee-vijayan-2026-functional-subspace-vector-algebra]

Context

PCA-reduced residual-stream subspace (30 components explain 40-90% of variance), affine relational decoding (a = alpha*q + beta*s + gamma, R^2 per layer/component), token-type clustering (query/separator/answer) in high-R^2 principal-component subspace, projection-correctness correlation (t-test, p<0.05 in mid/late layers)

Papers

Functional Subspace, where language models can use vector algebra to solve problems — Lee, Jung H., Vijayan, Sujith2026 · arXiv:2602.01687