SourceUncovering Latent Memories in Large Language Models
For sequences that were un-memorized (KL-LD > 50) at checkpoint 10k but memorized (KL-LD < 10) at a later checkpoint, the authors add Gaussian noise (magnitude 2×10⁻³) to each weight parameter and repeat 200 times, selecting the perturbation yielding the lowest KL-LD. This recovers the latent memories, and the resulting distribution matches that of the later checkpoint. In contrast, sequences never memorized and sequences not yet seen by the model cannot be recovered by perturbation. As a control, sampling at four different temperatures (200 samples each) fails to recover the latent memories. Additionally, latent memorized sequences exhibit significantly lower cross-entropy loss than un-memorized or unseen sequences, providing a diagnostic.