Emergence of Linear Truth Encodings in Language Models

Alberto Bietti (Flatiron Institute, Simons Foundation) · Shauli Ravfogel (Bar-Ilan University) · Gilad Yehudai (Weizmann Institute of Technology) · Tal Linzen (NYU and Google) · Joan Bruna (NYU)
data distributionempirical motivationfactual associationsfactual statementslanguage model losslinear subspaceslinear truth representationsmechanistic demonstrationone-layer transformerpretrained language modelsprobing studiestruth encodingtruth subspacestwo-phase learning dynamic

Recent probing studies reveal that large language models exhibit linear subspaces that separate true from false statements, yet the mechanism behind their emergence is unclear. We introduce a transparent, one-layer transformer toy model that reproduces such truth subspaces end-to-end and exposes one concrete route by which they can arise. We study one simple setting in which truth encoding can emerge: a data distribution where factual statements co-occur with other factual statements (and vice-versa), encouraging the model to learn this distinction in order to lower the LM loss on future tokens. We corroborate this pattern with experiments in pretrained language models. Finally, in the toy setting we observe a two-phase learning dynamic: networks first memorize individual factual associations in a few steps, then