Source
Park, Choe and Veitch (2024), LIT-606, read as NOTE-328.
The account
“Concepts are linear” is used in three senses. Subspace: the differences between word pairs that change only one concept (king/queen, man/woman) lie along one direction. Measurement: a linear probe reads the concept off the representation. Intervention: adding a vector changes the concept and nothing else.
In the unembedding space, the subspace direction is a probe: the log-odds of queen over king is linear in the context representation, with that direction as coefficient. In the embedding space, the subspace direction is a steering vector: adding it raises the target concept’s probability and leaves every concept that can vary independently of it unchanged.
What joins the two spaces is an inner product, and training does not supply one. Any invertible linear map on unembeddings, with its inverse transpose on embeddings, leaves every prediction unchanged. So “these two concepts have cosine 0.1” is a fact about a coordinate system, not about the model. The account picks the inner products under which causally separable concepts are orthogonal. Under any of them, a concept’s unembedding direction maps to its embedding direction, which makes the probe and the steering vector the same vector. If separable concepts are uncorrelated over words drawn uniformly from the vocabulary, the inverse covariance of the unembedding rows is one such product.
Each step is a theorem in LIT-606: the probe identity (Theorem 2.2), the
steering identity (Theorem 2.5), the map from one direction to the other under
a causal inner product (Theorem 3.2), and the covariance form (Theorem 3.4,
with D = I chosen). On LLaMA-2 7B, 26 of its 27 concepts show counterfactual
pair differences aligned with a shared direction well beyond random pairs.
What it explains
- Why probes and steering vectors for the same concept keep turning out related. They are one object seen from two spaces.
- Why cosine similarity between concept directions varies between models that behave alike. It is not identified, and on Gemma-2B (tied embeddings) it fails where the whitened product works.
- Why a fitted probe can be less clean than the pair direction. The probe absorbs correlated off-target concepts, and the pair direction does not.
- Why SOTA-378 tells you to whiten before comparing concept
directions. Raw cosine is not identified, and the practice’s product,
uᵀ Cov(γ)⁻¹ v, is the member of the causal family this account names, under which separable concepts are orthogonal. Its steering directionCov(γ)⁻¹γ̄is the map from unembedding to embedding direction above.
Where it is weak
- The premises are idealizations. Exact cones of differences, concepts read deterministically off one output token, and d − 1 separable concepts completing a basis. Real directions are approximate, and one of 27 tested concepts (thing⇒part) has none.
- Covariance is one member of a family. D = I is chosen, not derived.
- The evidence stops at the output. Intermediate-layer steering and probing, where the practice lives, is future work in the source. LIT-526 tested the whitened product for cross-lingual concept transport across 17 models and found no benefit over spectral regularization (p = 0.95).
- Orthogonality is read off heatmaps of 27 concepts on two models.
Relation to other accounts
THEORY-034 builds on this: it adds a threshold to each direction and reads concepts as intersections of half-spaces. THEORY-063 locates contextual concept directions in the low-variance tail of the same unembedding covariance this account whitens by. It does not say whether whitening is right, and its source’s null is against one use of it. THEORY-089 says why word vectors trained on co-occurrence have linear offsets at all. This account says what an offset means once it exists, and needs no mechanism for it.