Locas: Your Models are Principled Initializers of Locally-Supported Parametric Memories
Not on the map. Why?
Abstract (arXiv)
In this paper, we aim to bridge test-time-training with a new type of parametric memory that can be flexibly offloaded from or merged into model parameters. We present Locas, a Locally-Supported parametric memory that shares the design of FFN blocks in modern transformers, allowing it to be flexibly permanentized into the model parameters while supporting efficient continual learning. We discuss two major variants of Locas: one with a conventional two-layer MLP design that has a clearer theoretical guarantee; the other one shares the same GLU-FFN structure with SOTA LLMs, and can be easily attached to existing models for both parameter-efficient and computation-efficient continual learning. Crucially, we show that proper initialization of such low-rank sideway-FFN-style memories -- performed in a principled way by reusing model parameters, activations and/or gradients -- is essential for fast convergence, improved generalization, and catastrophic forgetting prevention. We validate the proposed memory mechanism on the PG-19 whole-book language modeling and LoCoMo long-context dialogue question answering tasks. With only 0.02% additional parameters in the lowest case, Locas-GLU is capable of storing the information from past context while maintaining a much smaller context window. In addition, we also test the model's general capability loss after memorizing the whole book with Locas, through comparative MMLU evaluation. Results show the promising ability of Locas to permanentize past context into parametric knowledge with minimized catastrophic forgetting of the model's existing internal knowledge.
Key facts. Cited by 6 papers in the map (data through 7 Oct 2026). Builds on 35 of its 39 references that are in the map. First posted on arXiv 4 Feb 2026.
Builds on
35 of 39 references.
References · 35
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,160
- 2017Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networkscited by 4,861
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,616
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,581
- 2023Toolformer: Language Models Can Teach Themselves to Use Toolscited by 4,072
- 2023Lost in the Middle: How Language Models Use Long Contextscited by 3,383
- 2020Longformer: The Long-Document Transformercited by 3,362
- 2023QLoRA: Efficient Finetuning of Quantized LLMscited by 3,355
- 2023Universal and Transferable Adversarial Attacks on Aligned Language Modelscited by 2,914
- 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 1,979
- 2020GLU Variants Improve Transformercited by 1,724
23 more
- 2019Generating Long Sequences with Sparse Transformerscited by 1,523
- 2021Calibrate Before Use: Improving Few-Shot Performance of Language Modelscited by 1,371
- 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,307
- 2021Train Short, Test Long: Attention with Linear Biases Enables Input Length Extrapolationcited by 1,085
- 2023Large Language Models Can Be Easily Distracted by Irrelevant Contextcited by 820
- 2024Evaluating Very Long-Term Conversational Memory of LLM Agentscited by 800
- 2024DoRA: Weight-Decomposed Low-Rank Adaptationcited by 740
- 2025Mem0: Building Production-Ready AI Agents with Scalable Long-Term Memorycited by 722
- 2019Compressive Transformers for Long-Range Sequence Modellingcited by 699
- 2014Neural Turing Machinescited by 630
- 2023Prompt Injection attack against LLM-integrated Applicationscited by 568
- 2016Weight Normalization: A Simple Reparameterization to Accelerate Training of Deep Neural Networkscited by 541
- 2014Memory Networkscited by 354
- 2024Learning to (Learn at Test Time): RNNs with Expressive Hidden Statescited by 329
- 2024WISE: Rethinking the Knowledge Memory for Lifelong Model Editing of Large Language Modelscited by 90
- 2025Test-Time Learning for Large Language Modelscited by 46
- 2024With Greater Text Comes Greater Necessity: Inference-Time Training Helps Long Text Generationcited by 36
- 2025Continuous Self-Improvement of Large Language Models by Test-time Training with Verifier-Driven Sample Selectioncited by 13
- 2025SPINE: Token-Selective Test-Time Reinforcement Learning with Entropy-Band Regularizationcited by 11
- 2025GraLoRA: Granular Low-Rank Adaptation for Parameter-Efficient Fine-Tuningcited by 8
- 2025CTR-LoRA: Curvature-Aware and Trust-Region Guided Low-Rank Adaptation for Large Language Modelscited by 4
- 2025LoKI: Low-damage Knowledge Implanting of Large Language Modelscited by 2
- 2025DiffoRA: Enabling Parameter-Efficient Fine-Tuning via Differential Module Selectioncited by 1
Built on by
Citing papers per year: 2026: 6.
6 papers.
- 2026MementoGUI: Learning Agentic Multimodal Memory Control for Long-Horizon GUI Agentscited by 9
- 2026SmartSearch: How Ranking Beats Structure for Conversational Memory Retrievalcited by 5
- 2026Metis: Memory Foundation Modelcited by 5
- 2026Training LLM Agents for Spontaneous, Reward-Free Self-Evolution via World Knowledge Explorationcited by 3
- 2026MemOps: Benchmarking Lifecycle Memory Operations in Long-Horizon Conversationscited by 3
- 2026Beyond Perplexity: A Behavioral Evaluation Framework for Deployment-Memory Claims in LLM Test-Time Trainingcited by 0
- 2026Metis: Memory Foundation Model29 Jul 2026
- 2026MemOps: Benchmarking Lifecycle Memory Operations in Long-Horizon Conversations14 Jul 2026
- 2026Beyond Perplexity: A Behavioral Evaluation Framework for Deployment-Memory Claims in LLM Test-Time Training1 Jul 2026
- 2026MementoGUI: Learning Agentic Multimodal Memory Control for Long-Horizon GUI Agents18 May 2026
- 2026Training LLM Agents for Spontaneous, Reward-Free Self-Evolution via World Knowledge Exploration20 Apr 2026
- 2026SmartSearch: How Ranking Beats Structure for Conversational Memory Retrieval16 Mar 2026