arXiv:2601.00756cs.LGcs.CL2026-01中稿 · the 41st ACM/SIGAP…被引 3

压缩记忆库,让大模型持续学习不遗忘

Memory Bank Compression for Continual Adaptation of Large Language Models

  • 用码本优化策略动态压缩记忆库,防止膨胀
  • 内存缩小至基线的0.3%,仍保持高准确率
  • 适合需要长期更新的大模型应用开发者

大型语言模型在日常应用中日益普及,但数据演变导致其知识快速过时。持续学习旨在不丢失旧知识的前提下更新模型。虽然全微调可引入新数据,但计算成本高且易产生灾难性遗忘。记忆增强方法通过外置记忆库存储信息来缓解此问题,但在真实场景下,随着大规模数据流持续涌入,记忆库会不断膨胀。本文提出MBC模型,在在线适应学习过程中通过码本优化策略压缩记忆库,并引入在线重置机制防止码本崩溃。同时,在注意力层采用键值低秩适配,高效利用压缩后的记忆表示。在基准问答数据集上的实验表明,与最先进基线相比,MBC将记忆库规模缩减至0.3%,同时在在线适应学习中保持高知识保留准确率。代码已公开于https://github.com/Thomkat/MBC。

原文摘要 · Abstract (English)

Large Language Models (LLMs) have become a mainstay for many everyday applications. However, as data evolve their knowledge quickly becomes outdated. Continual learning aims to update LLMs with new information without erasing previously acquired knowledge. Although methods such as full fine-tuning can incorporate new data, they are computationally expensive and prone to catastrophic forgetting, where prior knowledge is overwritten. Memory-augmented approaches address this by equipping LLMs with a memory bank, that is an external memory module which stores information for future use. However, these methods face a critical limitation, in particular, the memory bank constantly grows in the real-world scenario when large-scale data streams arrive. In this paper, we propose MBC, a model that compresses the memory bank through a codebook optimization strategy during online adaptation learning. To ensure stable learning, we also introduce an online resetting mechanism that prevents codebook collapse. In addition, we employ Key-Value Low-Rank Adaptation in the attention layers of the LLM, enabling efficient utilization of the compressed memory representations. Experiments with benchmark question-answering datasets demonstrate that MBC reduces the memory bank size to 0.3% when compared against the most competitive baseline, while maintaining high retention accuracy during online adaptation learning. Our code is publicly available at https://github.com/Thomkat/MBC.

大模型持续学习记忆压缩

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。