Grokking in the Wild: Data Augmentation for Real-World Multi-Hop Reasoning with Transformers
On the map: LLM reasoning → Test-time compute and efficient thinking
Abstract (arXiv)
Transformers have achieved great success in numerous NLP tasks but continue to exhibit notable gaps in multi-step factual reasoning, especially when real-world knowledge is sparse. Recent advances in grokking have demonstrated that neural networks can transition from memorizing to perfectly generalizing once they detect underlying logical patterns - yet these studies have primarily used small, synthetic tasks. In this paper, for the first time, we extend grokking to real-world factual data and address the challenge of dataset sparsity by augmenting existing knowledge graphs with carefully designed synthetic data to raise the ratio φr of inferred facts to atomic facts above the threshold required for grokking. Surprisingly, we find that even factually incorrect synthetic data can strengthen emergent reasoning circuits rather than degrade accuracy, as it forces the model to rely on relational structure rather than memorization. When evaluated on multi-hop reasoning benchmarks, our approach achieves up to 95-100% accuracy on 2WikiMultiHopQA - substantially improving over strong baselines and matching or exceeding current state-of-the-art results. We further provide an in-depth analysis of how increasing φr drives the formation of generalizing circuits inside Transformers. Our findings suggest that grokking-based data augmentation can unlock implicit multi-hop reasoning capabilities, opening the door to more robust and interpretable factual reasoning in large-scale language models.
Key facts. Cited by 12 papers in the map (data through 14 Oct 2026). Placed in Test-time compute and efficient thinking, one of 17 directions in LLM reasoning. Builds on 15 of its 18 references that are in the map. First posted on arXiv 29 Apr 2025.
Builds on
15 of 18 references.
References · 15
- 2018HotpotQA: A Dataset for Diverse, Explainable Multi-hop Question Answeringcited by 3,754
- 2017Decoupled Weight Decay Regularizationcited by 1,831
- 2020Constructing A Multi-hop QA Dataset for Comprehensive Evaluation of Reasoning Stepscited by 1,282
- 2018Reconciling modern machine learning practice and the bias-variance trade-offcited by 927
- 2021MuSiQue: Multihop Questions via Single-hop Question Compositioncited by 916
- 2023Progress measures for grokking via mechanistic interpretabilitycited by 839
- 2019Deep Double Descent: Where Bigger Models and More Data Hurtcited by 672
- 2022Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasetscited by 629
- 2024Multi-Step Reasoning with Large Language Models, a Surveycited by 143
- 2022The Slingshot Mechanism: An Empirical Study of Adaptive Optimizers and the Grokking Phenomenoncited by 94
- 2024Grokked Transformers are Implicit Reasoners: A Mechanistic Journey to the Edge of Generalizationcited by 79
- 2023Unifying Grokking and Double Descentcited by 50
3 more
- 2024Grokfast: Accelerated Grokking by Amplifying Slow Gradientscited by 41
- 2021Multi-scale Feature Learning Dynamics: Insights for Double Descentcited by 24
- 2023Training Dynamics of Deep Network Linear Regionscited by 3
Built on by
Citing papers per year: 2025: 5; 2026: 7.
12 papers.
- 2025RL Grokking Recipe: How Does RL Unlock and Transfer New Algorithms in LLMs?cited by 23
- 2025Provable Scaling Laws of Feature Emergence from Learning Dynamics of Grokkingcited by 15
- 2025Can LLMs Reconcile Knowledge Conflicts in Counterfactual Reasoningcited by 7
- 2026The Geometric Inductive Bias of Grokking: Bypassing Phase Transitions via Architectural Topologycited by 5
- 2026Is Grokking Worthwhile? Functional Analysis and Transferability of Generalization Circuits in Transformerscited by 1
- 2026The Scaling Properties of Implicit Deductive Reasoning in Transformerscited by 1
- 2025Input-Time Scaling: Adding Noise and Irrelevance into Less-Is-More Drastically Improves Reasoning Performance and Efficiencycited by 0
- 2025The Kinetics of Reasoning: How Chain-of-Thought Shapes Learning in Transformers?cited by 0
- 2026Fast Generalization after Interpolation via Critically Damped Momentum Optimizationcited by 0
- 2026The Periodic Table of LLM Reasoning: A Structured Survey of Reasoning Paradigms, Methods, and Failure Modescited by 0
- 2026Why Knowing Both Hops Is Not Enough: Understanding Two-Hop Generalization in Language Modelscited by 0
- 2026Misalignment of Low-Loss Regions Causes Grokkingcited by 0
- 2026Misalignment of Low-Loss Regions Causes Grokking30 Sep 2026
- 2026Why Knowing Both Hops Is Not Enough: Understanding Two-Hop Generalization in Language Models7 Aug 2026
- 2026The Periodic Table of LLM Reasoning: A Structured Survey of Reasoning Paradigms, Methods, and Failure Modes9 Jun 2026
- 2026Fast Generalization after Interpolation via Critically Damped Momentum Optimization1 Jun 2026
- 2026The Scaling Properties of Implicit Deductive Reasoning in Transformers5 May 2026
- 2026The Geometric Inductive Bias of Grokking: Bypassing Phase Transitions via Architectural Topology5 Mar 2026
- 2026Is Grokking Worthwhile? Functional Analysis and Transferability of Generalization Circuits in Transformers14 Jan 2026
- 2025The Kinetics of Reasoning: How Chain-of-Thought Shapes Learning in Transformers?28 Oct 2025
- 2025Provable Scaling Laws of Feature Emergence from Learning Dynamics of Grokking25 Sep 2025
- 2025RL Grokking Recipe: How Does RL Unlock and Transfer New Algorithms in LLMs?25 Sep 2025
- 2025Input-Time Scaling: Adding Noise and Irrelevance into Less-Is-More Drastically Improves Reasoning Performance and Efficiency19 Aug 2025
- 2025Can LLMs Reconcile Knowledge Conflicts in Counterfactual Reasoning15 Jun 2025
Similar, not cited
Close in wording, no citation either way
- 2026Learning from Synthetic Data Improves Multi-hop Reasoning
- 2026Loop, Think, & Generalize: Implicit Reasoning in Recurrent-Depth Transformers
- 2025Language models can learn implicit multi-hop reasoning, but only if they have lots of training data
- 2025Finding the Minimal Parameter Budget for Implicit Reasoning: A Data Complexity Driven Scaling Law for Language Models
- 2024Large Language Models Still Face Challenges in Multi-Hop Reasoning with External Knowledge
- 2026Addressing the Reasoning Gap: Mechanistic Circuit-Based Knowledge Editing in Large Language Models
- 2023Memory Injections: Correcting Multi-Hop Reasoning Failures during Inference in Transformer-Based Language Models
- 2026Scaling Reasoning Hop Exposes Weaknesses: Demystifying and Improving Hop Generalization in Large Language Models