acceptodds
Under review as a conference paper at ICLR 2027

Beyond Embedding Transfer: Component Roles in Grokking Transfer and Stability

Abstract

Warm-start transfer can make algorithmic tasks generalize rapidly, yet it is unclear which model components provide the gain and whether that gain remains stable under continued optimization. We study cross-operator transfer on modular arithmetic and separate *efficacy* (early velocity) from *stability* (post-reach drawdown). In a scale-matched 108-run battery across 12 seed blocks (a 96-run factorial plus a 12-run scale control), transferring internal attention/MLP weight matrices () in addition to token embeddings and readout () improves early mean accuracy by 5.46 percentage points (Holm ) and reduces confirmation latency by 558 steps (Holm ). While readout plus internal-block transfer satisfies the pre-specified -step mean-latency equivalence criterion in one-layer models (TOST , although Full is modestly faster in 11/12 paired seeds), a prospective two-layer replication confirms the internal-block acquisition advantage (12/12 seeds, integral units, ) while revealing an architecture-dependent boundary: omitting donor embeddings falls integral units below Full transfer, far outside the pre-specified -unit equivalence margin. Continued target training, however, frequently triggers severe post-grokking relapse. Freezing transferred representation carriers () nearly eliminates offline relapse (, Holm ). Online validation-triggered gating slashes True Max Drawdown from to on (), with prospective confirmations extending protection across affine, nonlinear quadratic, and two-layer targets (10.94–23.47 pp reductions), while distinguishing continual stabilization from static early stopping. In non-abelian , unshielded transfer surges transiently ( peak), but a prospective shielding cohort yields no confirmed benefit ( pp). These results establish a component-level dissociation between transfer acceleration and trajectory stability, and expose the empirical boundaries of parameter shielding.

open until 14 Dec 2026

est. 32% chance this paper gets accepted at ICLR 2027.

Reject 68%Accept 32%

What do you think this paper will get?

All positions stay anonymous.

Related papers

Loading the map…

Discussion (0)

Sign in to comment.