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

Soyoung Oh, Vera Demberg


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.
Anthology ID:
2026.emnlp-main.1238
Volume:
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
EMNLP
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
28229–28243
Language:
URL:
https://aclanthology.org/2026.emnlp-main.1238/
DOI:
Bibkey:
Cite (ACL):
Soyoung Oh and Vera Demberg. 2026. A retrieval conditioned rebinding circuit for dynamic entity tracking in large language models. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, pages 28229–28243, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
A retrieval conditioned rebinding circuit for dynamic entity tracking in large language models (Oh & Demberg, EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-main.1238.pdf