English

Polar probe linearly decodes semantic structures from LLMs

Computation and Language 2026-05-19 v2

Abstract

How do artificial neural networks bind concepts to form complex semantic structures? Here, we propose a simple neural code, whereby the existence and the type of relations between entities are represented by the distance and the direction between their embeddings, respectively. We test this hypothesis in a variety of Large Language Models (LLMs), each input with natural-language descriptions of minimalist tasks from five different domains: arithmetic, visual scenes, family trees, metro maps and social interactions. Results show that the true semantic structures can be linearly recovered with a Polar Probe targeting a subspace of LLMs' layer activations. Second, this code emerges mostly in middle layers and improves with LLM performance. Third, these Polar Probes successfully generalize to new entities and relation types, but degrades with the size of the semantic structure. Finally, the quality of the polar representation correlates with the LLM's ability to answer questions about the semantic structure. Together, these findings suggest that LLMs learn to build complex semantic structures by binding representations with a simple geometrical principle.

Keywords

Cite

@article{arxiv.2605.14125,
  title  = {Polar probe linearly decodes semantic structures from LLMs},
  author = {Pablo J. Diego-Simón and Pierre Orhan and Emmanuel Chemla and Yair Lakretz and Jean-Rémi King},
  journal= {arXiv preprint arXiv:2605.14125},
  year   = {2026}
}