arXiv:2501.07382cs.LGcs.AI2025-01被引 4

提出双记忆系统,让模型持续学习不遗忘。

Information-Theoretic Dual Memory System for Continual Learning

  • 用快慢双缓冲区分别存新样本和关键旧样本
  • 通过信息论方法选出多样且重要数据,提升记忆效率
  • 自动删冗余数据,适合长期增量学习场景

持续从动态环境中获取新知识是动物生存的核心能力,对应于机器学习中的持续学习问题,即在不损害已有知识的前提下学习一系列任务。现有基于记忆的方法多采用单一固定大小的缓冲区存储历史数据,难以兼顾新旧样本的管理。受互补学习系统(CLS)理论启发,本文提出信息论驱动的双记忆系统(ITDMS),包含一个用于暂存新样本的快速记忆缓冲区和一个保存关键信息的慢速记忆缓冲区。快速缓冲区采用高效水库采样优化;慢速缓冲区则引入新颖的信息论优化策略,选择并保留具有多样性和信息量的数据。此外,设计了平衡采样机制,自动识别并移除冗余记忆,释放空间以适应不断增长的任务需求。在多个持续学习实验中验证了该方法的有效性。

原文摘要 · Abstract (English)

Continuously acquiring new knowledge from a dynamic environment is a fundamental capability for animals, facilitating their survival and ability to address various challenges. This capability is referred to as continual learning, which focuses on the ability to learn a sequence of tasks without the detriment of previous knowledge. A prevalent strategy to tackle continual learning involves selecting and storing numerous essential data samples from prior tasks within a fixed-size memory buffer. However, the majority of current memory-based techniques typically utilize a single memory buffer, which poses challenges in concurrently managing newly acquired and previously learned samples. Drawing inspiration from the Complementary Learning Systems (CLS) theory, which defines rapid and gradual learning mechanisms for processing information, we propose an innovative dual memory system called the Information-Theoretic Dual Memory System (ITDMS). This system comprises a fast memory buffer designed to retain temporary and novel samples, alongside a slow memory buffer dedicated to preserving critical and informative samples. The fast memory buffer is optimized employing an efficient reservoir sampling process. Furthermore, we introduce a novel information-theoretic memory optimization strategy that selectively identifies and retains diverse and informative data samples for the slow memory buffer. Additionally, we propose a novel balanced sample selection procedure that automatically identifies and eliminates redundant memorized samples, thus freeing up memory capacity for new data acquisitions, which can deal with a growing array of tasks. Our methodology is rigorously assessed through a series of continual learning experiments, with empirical results underscoring the effectiveness of the proposed system.

持续学习双记忆信息论增量学习

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