Sources

542 sources

Source

Orhan (2023)

Recognition, recall, and retention of few-shot memories in large language models

arXiv

Find It

DOI

Abstract

The training of modern large language models (LLMs) takes place in a regime where most training examples are seen only a few times by the model during the course of training. What does a model remember about such examples seen only a few times during training and how long does that memory persist in the face of continuous training with new examples? Here, we investigate these questions through simple recognition, recall, and retention experiments with LLMs. In recognition experiments, we ask if the model can distinguish the seen example from a novel example; in recall experiments, we ask if the model can correctly recall the seen example when cued by a part of it; and in retention experiments, we periodically probe the model's memory for the original examples as the model is trained continuously with new examples. We find that a single exposure is generally sufficient for a model to achieve near perfect accuracy even in very challenging recognition experiments. We estimate that the recognition performance of even small language models easily exceeds human recognition performance reported in similar experiments with humans (Shepard, 1967). Achieving near perfect recall takes more exposures, but most models can do it in just 3 exposures. The flip side of this remarkable capacity for fast learning is that precise memories are quickly overwritten: recall performance for the original examples drops steeply over the first 10 training updates with new examples, followed by a more gradual decline. Even after 100K updates, however, some of the original examples are still recalled near perfectly. A qualitatively similar retention pattern has been observed in human long-term memory retention studies before (Bahrick, 1984). Finally, recognition is much more robust to interference than recall and memory for natural language sentences is generally superior to memory for stimuli without structure.

Reference

Orhan, A. E. (2023). Recognition, recall, and retention of few-shot memories in large language models. arXiv preprint arXiv:2303.17557. https://doi.org/10.48550/arxiv.2303.17557

What Rests on This

1 mechanism cites this source. Each finding below is quoted as it appears on that mechanism's own page.

Recognition Memory Processes

  • Measured in AI

    A 2023 preprint gave 19 language models (GPT-2, 2019; GPT-Neo and GPT-J, 2021; OPT and BLOOM, 2022; 125 million to 13 billion parameters) a sentence test modeled on Shepard's: each model was trained once on 600 new sentences, then shown each studied sentence with a new one, but it made no old/new choice; the author counted a trial correct when the model's loss (how surprising it found the text) was lower for the studied sentence, explicitly as "a proxy for familiarity"

  • Measured in AI

    After one exposure, every model's loss comparison picked the studied sentence on roughly 94% (smallest GPT-2) to nearly 100% (best BLOOM, OPT and GPT-J models) of pairs, plotted against a human band of about 88–91%; the human value is not stated in the text, no human sample or spread is given, and no statistical test against people was run

  • Measured in AI

    The design matches Shepard's in its essentials (one exposure, randomly chosen new foils, a two-choice test straight after study), but the scoring does not: people made a recognition decision, while the models' familiarity-like strength signal was read off by the experimenter, learning settings were searched to maximize memory (about 20,000 runs across the study), models were scored on 600 pairs rather than 68, and the sentences differed; a result favouring the side these differences helped cannot be counted