Fresh in memory: Training-order recency is linearly encoded in language model activations

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Krasheninnikov, Dmitrii, Turner, Richard E., Krueger, David
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866912598406987776
author Krasheninnikov, Dmitrii
Turner, Richard E.
Krueger, David
author_facet Krasheninnikov, Dmitrii
Turner, Richard E.
Krueger, David
contents We show that language models' activations linearly encode when information was learned during training. Our setup involves creating a model with a known training order by sequentially fine-tuning Llama-3.2-1B on six disjoint but otherwise similar datasets about named entities. We find that the average activations of test samples corresponding to the six training datasets encode the training order: when projected into a 2D subspace, these centroids are arranged exactly in the order of training and lie on a straight line. Further, we show that linear probes can accurately (~90%) distinguish "early" vs. "late" entities, generalizing to entities unseen during the probes' own training. The model can also be fine-tuned to explicitly report an unseen entity's training stage (~80% accuracy). Interestingly, the training-order encoding does not seem attributable to simple differences in activation magnitudes, losses, or model confidence. Our paper demonstrates that models are capable of differentiating information by its acquisition time, and carries significant implications for how they might manage conflicting data and respond to knowledge modifications.
format Preprint
id arxiv_https___arxiv_org_abs_2509_14223
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Fresh in memory: Training-order recency is linearly encoded in language model activations
Krasheninnikov, Dmitrii
Turner, Richard E.
Krueger, David
Machine Learning
Artificial Intelligence
Computation and Language
We show that language models' activations linearly encode when information was learned during training. Our setup involves creating a model with a known training order by sequentially fine-tuning Llama-3.2-1B on six disjoint but otherwise similar datasets about named entities. We find that the average activations of test samples corresponding to the six training datasets encode the training order: when projected into a 2D subspace, these centroids are arranged exactly in the order of training and lie on a straight line. Further, we show that linear probes can accurately (~90%) distinguish "early" vs. "late" entities, generalizing to entities unseen during the probes' own training. The model can also be fine-tuned to explicitly report an unseen entity's training stage (~80% accuracy). Interestingly, the training-order encoding does not seem attributable to simple differences in activation magnitudes, losses, or model confidence. Our paper demonstrates that models are capable of differentiating information by its acquisition time, and carries significant implications for how they might manage conflicting data and respond to knowledge modifications.
title Fresh in memory: Training-order recency is linearly encoded in language model activations
topic Machine Learning
Artificial Intelligence
Computation and Language
url https://arxiv.org/abs/2509.14223