associative-recurrent-memory-transformer-a8f37232·1 events·first seen Aliases: Associative Recurrent Memory Transformer
Researchers propose using the Associative Recurrent Memory Transformer (ARMT) as a practical method for extending LLM context length beyond original limits while achieving constant memory scaling instead of linear. The training recipe combines continued pre-training, synthetic long-context data, curriculum learning, and selective layer integration of associative memory. Experiments show ARMT-augmented models generalize to out-of-distribution context lengths and require 30% fewer FLOPs while preserving in-window baseline performance.