cs.CLDate pending

A retrieval conditioned rebinding circuit for dynamic entity tracking in large language models

Authors: Soyoung OhVera Demberg

Organizations: 1Language Science and Technology, Saarland University · 2Max Planck Institute for Informatics

Abstract

To interpret context correctly and retrieve relevant information, large language models must bind entities to their attributes and update these bindings as state changes. We analyze how LLMs implement this binding process in a dynamic state tracking. Using causal interventions, we identify a retrieval conditioned rebinding mechanism, a compact attention head circuit that propagated binding information and when the entity is queried, uses the updated binding to retrieve the corresponding attribute. Across Gemma and Llama models, this circuit supports rebinding behavior, but the representational signature of the mechanism differs across model families. In Gemma models, the binding signature is clearly expressed in the query/key subspaces of the relevant attention heads, whereas in Llama models, the binding information is carried primarily in key vectors. Overall, our results reveal an interpretable mechanism for context dependent state tracking in LLMs.

Explore similar work

CardsList
  1. Cell-Based Representation of Relational Binding in Language Models

    Apr 21, 2026Qin Dai, Benjamin Heinzerling, Kentaro InuiRelationalDiscourse

  2. Do Language Models Track Entities Across State Changes?

    May 28, 2026Zilu Tang, Qiao Zhao, Gabriel Franco +4Entities