MATH · IN · MODELS

PLS-found numeric-comparison subspaces causally control LLM answers

measured in 1 paper

El-Shangiti et al. show LLMs answering numeric-comparison questions encode each entity's numeric attribute in a low-dimensional linear subspace, found via a 5-component PLS regression on the last-token state [el-shangiti-etal-2024-geometry-of-numerical-reasoning] The fit reaches R^2 > 0.8 for birth year, death year and latitude across Llama-3-8B-Instruct, Mistral-7B-Instruct and Qwen2.5-7B-Instruct [el-shangiti-etal-2024-geometry-of-numerical-reasoning] Intervening along the first PLS component flips the model's Yes/No comparison answer far more than an equal-magnitude random control, concentrated in roughly the first half of layers [el-shangiti-etal-2024-geometry-of-numerical-reasoning]

Context

PLS-identified numeric subspace, causal comparison-answer flip, cross-model replication

Papers

The Geometry of Numerical Reasoning: Language Models Compare Numeric Properties in Linear Subspaces — El-Shangiti, Ahmed Oumar, Hiraoka, Tatsuya, AlQuabeh, Hilal, Heinzerling, Benjamin, Inui, Kentaro2024 · arXiv:2410.13194