mlsanityData to 14 Oct 2026

Grokking in the Wild: Data Augmentation for Real-World Multi-Hop Reasoning with Transformers

arXiv 2504.20752 ↗First posted 29 Apr 2025cs.CL cs.AI cs.LGcited by 12

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

  1. 2018HotpotQA: A Dataset for Diverse, Explainable Multi-hop Question Answeringcited by 3,754
  2. 2017Decoupled Weight Decay Regularizationcited by 1,831
  3. 2020Constructing A Multi-hop QA Dataset for Comprehensive Evaluation of Reasoning Stepscited by 1,282
  4. 2018Reconciling modern machine learning practice and the bias-variance trade-offcited by 927
  5. 2021MuSiQue: Multihop Questions via Single-hop Question Compositioncited by 916
  6. 2023Progress measures for grokking via mechanistic interpretabilitycited by 839
  7. 2019Deep Double Descent: Where Bigger Models and More Data Hurtcited by 672
  8. 2022Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasetscited by 629
  9. 2024Multi-Step Reasoning with Large Language Models, a Surveycited by 143
  10. 2022The Slingshot Mechanism: An Empirical Study of Adaptive Optimizers and the Grokking Phenomenoncited by 94
  11. 2024Grokked Transformers are Implicit Reasoners: A Mechanistic Journey to the Edge of Generalizationcited by 79
  12. 2023Unifying Grokking and Double Descentcited by 50
3 more
  1. 2024Grokfast: Accelerated Grokking by Amplifying Slow Gradientscited by 41
  2. 2021Multi-scale Feature Learning Dynamics: Insights for Double Descentcited by 24
  3. 2023Training Dynamics of Deep Network Linear Regionscited by 3

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2025: 5; 2026: 7.

12 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026Learning from Synthetic Data Improves Multi-hop Reasoning
  2. 2026Loop, Think, & Generalize: Implicit Reasoning in Recurrent-Depth Transformers
  3. 2025Language models can learn implicit multi-hop reasoning, but only if they have lots of training data
  4. 2025Finding the Minimal Parameter Budget for Implicit Reasoning: A Data Complexity Driven Scaling Law for Language Models
  5. 2024Large Language Models Still Face Challenges in Multi-Hop Reasoning with External Knowledge
  6. 2026Addressing the Reasoning Gap: Mechanistic Circuit-Based Knowledge Editing in Large Language Models
  7. 2023Memory Injections: Correcting Multi-Hop Reasoning Failures during Inference in Transformer-Based Language Models
  8. 2026Scaling Reasoning Hop Exposes Weaknesses: Demystifying and Improving Hop Generalization in Large Language Models