GrokAlign, a Jacobian-norm regularizer, accelerates grokking by aligning Jacobians with training data, and centroid alignment tracks when generalization and robustness emerge.
Michaud, and Max Tegmark
1 Pith paper cite this work. Polarity classification is still indexing.
1
Pith paper citing it
fields
cs.LG 1years
2025 1verdicts
CONDITIONAL 1representative citing papers
citing papers explorer
-
GrokAlign: Geometric Characterisation and Acceleration of Grokking
GrokAlign, a Jacobian-norm regularizer, accelerates grokking by aligning Jacobians with training data, and centroid alignment tracks when generalization and robustness emerge.