Neel Nanda's Classic Paper: Mechanistically Explaining Grokking in Three Phases

burny_tech · x · 2026-11-10

Reshared for Neel Nanda's birthday: his 2023 paper with Lawrence Chan, Tom Lieberum, Jess Smith and Jacob Steinhardt, Progress measures for grokking via mechanistic interpretability. The work fully reverse-engineers the algorithm learned by small transformers on modular addition—discrete Fourier transforms and trig identities convert addition into rotation on a circle—confirmed via activation/weight analysis and ablations in Fourier space. From this understanding the authors define progress measures splitting training into three continuous phases: memorization, circuit formation, and cleanup, showing grokking is not a sudden shift but gradual amplification of structured mechanisms followed by removal of memorizing components. The poster calls it a pure golden nerd-snipe that got them into neural geometry manifolds.

Related event: Neel Nanda's Classic Paper Decomposes Grokking into Three Stages(2 posts)→

Original post →

More from Research

Research channel →