Graph spectral analysis (Fiedler value + Scheffer CSD indicators) predicts grokking 21k steps before loss function - five reproducible experiments [R]

· Source: Machine Learning · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Mathematics & Computational Sciences · Depth: Expert, quick

Summary

A new study applies graph spectral analysis, specifically the Fiedler value (second-smallest eigenvalue of the weight graph Laplacian) and Scheffer critical slowing down indicators, to monitor neural network topology during training. This method predicts "grokking" 21,000 steps before test accuracy improves, and distinguishes grokking from catastrophic forgetting by their distinct structural fingerprints (slopes of 0.00128 vs. 0.00471/step). The research demonstrates structurally-guided interventions that preserve 91.7% of knowledge compared to 2.6% in unsteered scenarios. It also shows 100%/100%/97.5% retention across three sequential tasks with 48x grokking acceleration, and a preemptive curriculum that preserves 100% knowledge by correctly ranking task disruption risk. Experiments were conducted on 2-layer MLPs (modular arithmetic) and 1-layer transformers (sequence prediction).

Key takeaway

For research scientists developing or training neural networks, understanding the structural dynamics of your models can provide significant predictive power. You should consider integrating graph spectral analysis, such as monitoring the Fiedler value, into your training diagnostics to preemptively detect phenomena like grokking and catastrophic forgetting, potentially enabling more effective interventions and curriculum design.

Key insights

Graph spectral analysis can predict and steer neural network training phenomena like grokking and catastrophic forgetting.

Principles

Method

Monitor neural network topology during training using the Fiedler value and Scheffer critical slowing down indicators, derived from the weight graph Laplacian, to detect and classify training phenomena.

In practice

Topics

Code references

Best for: Research Scientist, AI Scientist, Machine Learning Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Machine Learning.