Papers › ∞-former: Infinite Memory Transformer

∞-former: Infinite Memory Transformer

1 Sep 2021arXiv:2109.00301archive 2025-07-28

Pedro Henrique Martins, Zita Marinho, André F. T. Martins

Transformers are unable to model long-term memories effectively, since the amount of computation they need to perform grows with the context length. While variations of efficient transformers have been proposed, they all have a finite memory capacity and are forced to drop old information. In this paper, we propose the ∞-former, which extends the vanilla transformer with an unbounded long-term memory. By making use of a continuous-space attention mechanism to attend over the long-term memory, the ∞-former's attention complexity becomes independent of the context length, trading off memory length with precision. In order to control where precision is more important, ∞-former maintains "sticky memories" being able to model arbitrarily long contexts while keeping the computation budget fixed. Experiments on a synthetic sorting task, language modeling, and document grounded dialogue generation demonstrate the ∞-former's ability to retain information from long sequences.

PaperPDFCode

In Syntology Open this paper in Syntology's Atlas, the map of the papers in Syntology's graph and their citations.

Code

deep-spin/infinite-former officialmentioned in paperjax report

Repository list and official/mentioned flags are the archive's, frozen 2025-07-28. Reachability, where shown, is from one Syntology probe window (2026-09-16 to 2026-09-18); repositories not probed show nothing. GitHub stars are not tracked.

Code Syntology ran Syntology

Not run by Syntology. Nothing on this page verifies that the listed code works.

Tasks

Dialogue GenerationLanguage ModelingLanguage Modelling

Results from the paper archive 2025-07-28

TaskDatasetModelMetricValueRank at snapshotLeaderboardReport
Dialogue Generation CMU-DoG ∞-former (Sticky memories) F1 9.01 #1 of 1 Archive leaderboard report
Dialogue Generation CMU-DoG ∞-former (Sticky memories) Meteor 7.55 #1 of 1 Archive leaderboard report
Dialogue Generation CMU-DoG ∞-former (Sticky memories) ROUGE-1 15.37 #1 of 1 Archive leaderboard report
Dialogue Generation CMU-DoG ∞-former (Sticky memories) Rouge-L 12.56 #1 of 1 Archive leaderboard report
Dialogue Generation PG-19 ∞-former (Sticky memories + initialized GPT-2 Small) Perplexity 32.48 #1 of 1 Archive leaderboard report
Language Modelling WikiText-103 [?]-former (SM) Test perplexity 16.61 #14 of 89 Archive leaderboard report
Language Modelling WikiText-103 -former (SM) Test perplexity 16.61 #15 of 89 Archive leaderboard report
Language Modelling WikiText-103 ∞-former (Sticky memories + initialized GPT-2 Small) Test perplexity 16.61 #16 of 89 Archive leaderboard report
Language Modelling WikiText-103 ∞-former (initialized GPT-2 Small) Test perplexity 16.64 #17 of 89 Archive leaderboard report
Language Modelling WikiText-103 [?]-former (Sticky memories) Test perplexity 24.22 #56 of 89 Archive leaderboard report
Language Modelling WikiText-103 \infty-former (Sticky memories) Test perplexity 24.22 #57 of 89 Archive leaderboard report
Language Modelling WikiText-103 ∞-former (Sticky memories) Test perplexity 24.22 #58 of 89 Archive leaderboard report

Ranks are positions in the archive's leaderboards as they stood at the 2025-07-28 snapshot. Results published since then are not among these rows, so a rank here is not a current standing.

Methods

Absolute Position EncodingsAdamAttentionBPEDense ConnectionsDropoutLabel SmoothingLayer NormalizationLinear LayerMulti-Head AttentionPosition-Wise Feed-Forward LayerResidual ConnectionSoftmaxTransformer

Report a problem or propose a change · a person checks every report against the paper or source before anything changes; decisions are listed on /corrections