Calc Consulting (@CalcCon) on X

X (formerly Twitter) ยท

2 min read Original article โ†—

@CalcCon

If you train a model for too long, it may overfit it's training data. Not surprising, this has been know for like forever. But did you know you can detect the signatures of overfitting in the layer weight matrices directly, without needing access to any data (train or test) ? In our recent paper (with hari kishan prakash ), ๐‹๐š๐ญ๐ž-๐’๐ญ๐š๐ ๐ž ๐†๐ž๐ง๐ž๐ซ๐š๐ฅ๐ข๐ณ๐š๐ญ๐ข๐จ๐ง ๐‚๐จ๐ฅ๐ฅ๐š๐ฉ๐ฌ๐ž ๐ข๐ง ๐†๐ซ๐จ๐ค๐ค๐ข๐ง๐ : ๐ƒ๐ž๐ญ๐ž๐œ๐ญ๐ข๐ง๐  ๐€๐ง๐ญ๐ข-๐†๐ซ๐จ๐ค๐ค๐ข๐ง๐  ๐ฐ๐ข๐ญ๐ก ๐–๐ž๐ข๐ ๐ก๐ญ๐–๐š๐ญ๐œ๐ก๐ž๐ซ, we show this explicitly in 2 different classic grokking experiments. And the overfitting we see is very different from what has been seen before! ๐Ÿ“„ paper: arxiv.org/abs/2602.02859 ๐Ÿ“ˆ ๐–๐ก๐š๐ญ ๐ญ๐ก๐ข๐ฌ ๐ฉ๐ฅ๐จ๐ญ ๐ฌ๐ก๐จ๐ฐ๐ฌ: Grokking โ†’ Stability โ†’ Anti-Grokking This figure below tracks training accuracy (red), test accuracy (purple), and WeightWatcher correlation traps (blue) while training for very long times. ๐๐ก๐š๐ฌ๐ž ๐Ÿ โ€” Memorization (pre-grokking) Training accuracy rises rapidly while test accuracy remains low. The model is fitting the training data without extracting the underlying structure. Correlation traps are minimal and largely uninformative. ๐๐ก๐š๐ฌ๐ž ๐Ÿ โ€” Grokking Test accuracy suddenly jumps to match training accuracy. The model transitions from memorization to true generalization. Correlation traps remain near zero, indicating stable, well-conditioned internal representations. ๐๐ก๐š๐ฌ๐ž ๐Ÿ‘ โ€” Late-stage instability (anti-grokking) Despite perfect training accuracy, test accuracy degrades over time. At the same time, correlation traps increase sharply and spread. Generalization collapses after it was achieved. The model is overfit ๐Š๐ž๐ฒ ๐ญ๐š๐ค๐ž๐š๐ฐ๐š๐ฒ So what ? It turns out, a lot of open-source LLMs, like OpenAI's GPT OSS 20B and 120B , show the exact same signatures! And a ton of them If you are training or fine-tuning your own models, watch out! You might be overfitting your data, even if you are following current NN best practices. Want to learn more? Check out the WeightWatcher project ๐ŸŒ weightwatcher.ai ๐–๐ž๐ข๐ ๐ก๐ญ๐–๐š๐ญ๐œ๐ก๐ž๐ซ ๐ข๐ฌ ๐š ๐จ๐ง๐ž-๐จ๐Ÿ-๐š-๐ค๐ข๐ง๐ ๐ฆ๐ฎ๐ฌ๐ญ-๐ก๐š๐ฏ๐ž ๐ญ๐จ๐จ๐ฅ ๐Ÿ๐จ๐ซ ๐š๐ง๐ฒ๐จ๐ง๐ž ๐ญ๐ซ๐š๐ข๐ง๐ข๐ง๐ , ๐๐ž๐ฉ๐ฅ๐จ๐ฒ๐ข๐ง๐ , ๐จ๐ซ ๐ฆ๐จ๐ง๐ข๐ญ๐จ๐ซ๐ข๐ง๐  ๐ƒ๐ž๐ž๐ฉ ๐๐ž๐ฎ๐ซ๐š๐ฅ ๐๐ž๐ญ๐ฐ๐จ๐ซ๐ค๐ฌ (๐ƒ๐๐๐ฌ). And you need help with AI, reach out. hashtag#TalkToChuck P.S. I'll be giving a talk at USF right here in SF (over by the GG Park) in 2 weeks on the weightwatcher project. Hope to see you there!