Is Grokking Worthwhile? Functional Analysis and Transferability of Generalization Circuits in Transformers

Kaiyu He, Mian Zhang, Peilin Wu, Xinya Du, Zhiyu Chen


Abstract
While Large Language Models (LLMs) excel at factual retrieval, they often struggle with the "curse of two-hop reasoning" in compositional tasks. Recent research suggests that parameter-sharing transformers can bridge this gap by forming a "Generalization Circuit" during a prolonged "grokking" phase. A fundamental question arises: Is a grokked model truly superior to its non-grokked counterparts? Furthermore, is the extensive computational cost of waiting for the grokking phase worthwhile? In this work, we conduct a mechanistic study to evaluate the Generalization Circuit’s role in knowledge assimilation and transfer. We demonstrate that: (i) The inference paths established by non-grokked and grokked models for in-distribution compositional queries are identical. This suggests that the "Generalization Circuit" does not represent the sudden acquisition of a new reasoning paradigm. Instead, we argue that grokking is the process of integrating memorized atomic facts into an easy-acquire, naturally established reasoning path. (ii) Achieving high accuracy on unseen cases after prolonged training and the formation of a certain reasoning path are not bound; they can occur independently under specific data regimes. (iii) Even a mature circuit exhibits limited transferability when integrating new knowledge, suggesting that "grokked" Transformers do not achieve a full mastery of compositional logic.
Anthology ID:
2026.findings-acl.1697
Volume:
Findings of the Association for Computational Linguistics: ACL 2026
Month:
July
Year:
2026
Address:
San Diego, California, United States
Editors:
Maria Liakata, Viviane P. Moreira, Jiajun Zhang, David Jurgens
Venue:
Findings
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
33993–34001
Language:
URL:
https://preview.aclanthology.org/ingest-acl/2026.findings-acl.1697/
DOI:
Bibkey:
Cite (ACL):
Kaiyu He, Mian Zhang, Peilin Wu, Xinya Du, and Zhiyu Chen. 2026. Is Grokking Worthwhile? Functional Analysis and Transferability of Generalization Circuits in Transformers. In Findings of the Association for Computational Linguistics: ACL 2026, pages 33993–34001, San Diego, California, United States. Association for Computational Linguistics.
Cite (Informal):
Is Grokking Worthwhile? Functional Analysis and Transferability of Generalization Circuits in Transformers (He et al., Findings 2026)
Copy Citation:
PDF:
https://preview.aclanthology.org/ingest-acl/2026.findings-acl.1697.pdf
Checklist:
 2026.findings-acl.1697.checklist.pdf