cs.LGApr 15, 2026

Ordinary Least Squares is a Special Case of Transformer

Authors: Xiaojun Tan, Yuchen Zhao

Organizations: Center for Peak Performance, Department of Psychology and Behavioral Sciences, Zhejiang University · Hangzhou Higgs Asset Management Co., Ltd. · College of Control Science and Engineering, Zhejiang University

Abstract

The statistical essence of the Transformer architecture has long remained elusive: Is it a universal approximator, or a neural network version of known computational algorithms? Through rigorous algebraic proof, we show that the latter better describes Transformer's basic nature: Ordinary Least Squares (OLS) is a special case of the single-layer Linear Transformer. Using the spectral decomposition of the empirical covariance matrix, we construct a specific parameter setting where the attention mechanism's forward pass becomes mathematically equivalent to the OLS closed-form projection. This means attention can solve the problem in one forward pass, not by iterating. Building upon this prototypical case, we further uncover a decoupled slow and fast memory mechanism within Transformers. Finally, the evolution from our established linear prototype to standard Transformers is discussed. This progression facilitates the transition of the Hopfield energy function from linear to exponential memory capacity, thereby establishing a clear continuity between modern deep architectures and classical statistical inference.

Figures & tables

Explore similar work

CardsList
  1. Ordinary Least Squares as an Attention Mechanism

    Apr 13, 2025Philippe Goulet CoulombeTransformer AttentionLinear Regression

  2. Looped Transformers with Layer Normalization Provably Learn the Power Method

    May 30, 2026Lyumin Wu, Chenyang Zhang, Yuan CaoRecurrent TransformersLayer Normalization