用细粒度记忆单元实现持续学习,避免遗忘与干扰。
Little by Little: Continual Learning via Incremental Mixture of Rank-1 Associative Memory Experts
- 将低秩适配器视为可复用的记忆原子,逐次增量扩展。
- 在CLIP和大模型上显著降低遗忘,优于当前最优方法。
- 无需路由模块,通过内在键值自激活,适合长期学习场景。
基于大预训练模型的持续学习旨在不发生灾难性遗忘的前提下逐步获取知识。现有基于LoRA的专家混合(MoE)方法通过添加独立新专家来扩展容量,同时冻结旧专家,但仍存在冗余、干扰、路由模糊等问题。本文研究发现,粗粒度专家粒度(如高秩LoRA)编码信息特异性低,导致专家重复、干扰及路由退化。为此,提出MoRAM(Rank-1关联记忆专家混合)。基于权重矩阵作为线性关联记忆的观点,MoRAM以可重用的原子级秩-1专家形式实现增量扩展。每个秩-1适配器作为细粒度专家或记忆单元,视作键值对。通过内在键实现自激活,消除显式路由,推理过程变为对累积学习快照的按内容检索与召回。在CLIP和大语言模型上的大量实验表明,MoRAM显著优于当前最优方法,实现了更优的可塑性-稳定性权衡、更强泛化能力与更低遗忘率。
原文摘要 · Abstract (English)
Continual learning (CL) with large pre-trained models aims to incrementally acquire knowledge without catastrophic forgetting. Existing LoRA-based Mixture-of-Experts (MoE) methods expand capacity by adding isolated new experts while freezing old ones, but still suffer from redundancy, interference, routing ambiguity, and consequent forgetting. We investigate the issues stemming from coarse-grained expert granularity. Coarse-grained experts (e.g., high-rank LoRA) encode low-specialty information, leading to expert duplication/interference and routing degradation/confusion as experts accumulate. In this work, we propose MoRAM (Mixture of Rank-1 Associative Memory). Grounded in the view that weight matrices act as linear associative memories, MoRAM achieves CL as incremental expansion of reusable atomic rank-1 experts as memory. Each rank-1 adapter acts as a fine-grained MoE expert or an associative memory unit. By viewing rank-1 experts as key-value memory pairs, we eliminate explicit MoE-LoRA routers with self-activation, where each memory atom evaluates its relevance via its intrinsic key. The inference process thus becomes a content-addressable retrieval and recall over the incrementally accumulated memory of learning snapshots. Extensive experiments on CLIP and LLMs show that MoRAM significantly outperforms state-of-the-art methods, achieving a better plasticity-stability trade-off, stronger generalization, and reduced forgetting. Project Page: https://artificer-ai-lab.github.io/MoRAM/.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。