In plain words: Memory Mosaics link many small associative memories—pattern-matching stores that recall similar past inputs—to predict together, and learn from examples shown in the prompt. They matched or beat transformers on medium-scale language tasks while making it clearer how each answer was formed.
Abstract
Memory Mosaics are networks of associative memories working in concert to achieve a prediction task of interest. Like transformers, memory mosaics possess compositional capabilities and in-context learning capabilities. Unlike transformers, memory mosaics achieve these capabilities in comparatively transparent way ("predictive disentanglement"). We illustrate these capabilities on a toy example and also show that memory mosaics perform as well or better than transformers on medium-scale language modeling tasks.
Jianyu Zhang, Niklas Nolte, Ranajoy Sadhukhan, Beidi Chen, Léon Bottou
arXiv:2405.06394 · cs.LG, cs.AI, cs.NE · submitted May 10, 2024 · updated Feb 27, 2025
abstract · pdf · html