cs.LGApr 13, 2026

The Linear Centroids Hypothesis: Features as Directions Learned by Local Experts

Authors: Thomas Walker, Ahmed Imtiaz Humayun, Randall Balestriero, Richard Baraniuk

Organizations: Rice University · Google Research · Brown University

Abstract

The Linear Representation Hypothesis (LRH) identifies features of a trained deep network (DN) as linear directions in the activation spaces, i.e., output spaces of intermediate layers. This characterization decouples the input-output maps learned by a DN from the organization of feature directions in its activation spaces. We introduce the Linear Centroids Hypothesis (LCH), which instead identifies features with linear directions among a DN's centroid spaces -- where any vector denotes a centroid or summary of a local affine expert characterizing the learned input-output maps of the DN exactly (e.g., for piecewise-affine DNs) or approximately (e.g., for smooth DNs like transformers). We show that replacing intermediate activations with centroids yields a functional drop-in alternative for standard interpretability tools. Empirically, this change yields sparser, more downstream-useful feature dictionaries on DINO ViTs, suppresses spurious directions on a controlled task, recovers interpretable circuits in GPT2-Large, and produces faithful gradient-based saliency maps. LCH unifies dictionaries, probing, circuits, and saliency maps into a single geometric object grounded in the network's input-output map -- making interpretability mechanistic by construction rather than post hoc. Code to study the LCH https://github.com/ThomasWalker1/LinearCentroidsHypothesis .

Explore similar work

CardsList
  1. How are linear representations learned? Exact solutions to the dynamics of abstraction

    Jul 9, 2026William W. Yang, Andrew M. Saxe, Peter E. LathamLinear Representation HypothesisDeep Linear Networks

  2. Topographic Training Concentrates Causal Circuits Without Improving Neuron Monosemanticity

    Sep 21, 2026Gautam Ranka, Shubham Santosh Pandere, Aiden DsouzaFeature SuperpositionVision Transformer

  3. REALM: Regime-Switching, Explainable, and Activation-Induced Linear Models

    Sep 26, 2026Xiaoran Cheng, Sen Na, Jia LiInterpretable MLNeural Surrogate Modeling