arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2511.09487cs.LG
收藏

PDAC: 基于概率密度感知的持续学习高效核心集选择

PDAC: Efficient Coreset Selection for Continual Learning via Probability Density Awareness

School of Mathematics, Harbin Institute of Technology(哈尔滨工业大学数学学院);Shanghai Artificial Intelligence Laboratory(上海人工智能实验室)

Junqi Gao, Zhichang Guo, Dazhi Zhang, Yao Li, Yi Ran, Biqing Qi

AI总结:

针对基于重放的持续学习中主流核心集选择方法计算成本高的问题,本文通过分析样本概率密度对误差抑制的作用,提出PDAC方法及流式场景的SPDAC变体,在提升效率的同时优于基线方法。

摘要:

基于重放的持续学习(CL)通过有限的记忆缓冲存储重放样本以保留知识,因此这类方法高度依赖存储样本的质量。当前基于重放的CL方法通常通过选择代表性子集(称为核心集)构建记忆缓冲,旨在以最小存储开销逼近全数据集的训练效果。然而,主流核心集选择(CS)方法通常将CS问题表述为双层优化问题,依赖大量内外迭代求解,导致计算成本高昂,限制了其实用效率。本文旨在为核心集构建提供更高效的选择逻辑与方案。为此,我们首先通过局部误差分解视角分析缓冲训练模型与贝叶斯最优模型之间的均方误差(MSE),以探究不同区域样本对MSE抑制的贡献。进一步的理论与实验分析表明,高概率密度样本在误差抑制中起主导作用。受此启发,我们提出概率密度感知核心集(PDAC)方法。PDAC利用投影高斯混合(PGM)模型估计每个样本的联合密度,实现高效的密度优先缓冲选择。最后,我们引入流式期望最大化(EM)算法增强PGM参数对流式数据的适应性,得到面向流式场景的流式PDAC(SPDAC)。大量对比实验表明,我们的方法在各种CL设置下优于其他基线,同时保证良好的效率。

英文摘要:

Rehearsal-based Continual Learning (CL) maintains a limited memory buffer to store replay samples for knowledge retention, making these approaches heavily reliant on the quality of the stored samples. Current Rehearsal-based CL methods typically construct the memory buffer by selecting a representative subset (referred to as coresets), aiming to approximate the training efficacy of the full dataset with minimal storage overhead. However, mainstream Coreset Selection (CS) methods generally formulate the CS problem as a bi-level optimization problem that relies on numerous inner and outer iterations to solve, leading to substantial computational cost thus limiting their practical efficiency. In this paper, we aim to provide a more efficient selection logic and scheme for coreset construction. To this end, we first analyze the Mean Squared Error (MSE) between the buffer-trained model and the Bayes-optimal model through the perspective of localized error decomposition to investigate the contribution of samples from different regions to MSE suppression. Further theoretical and experimental analyses demonstrate that samples with high probability density play a dominant role in error suppression. Inspired by this, we propose the Probability Density-Aware Coreset (PDAC) method. PDAC leverages the Projected Gaussian Mixture (PGM) model to estimate each sample's joint density, enabling efficient density-prioritized buffer selection. Finally, we introduce the streaming Expectation Maximization (EM) algorithm to enhance the adaptability of PGM parameters to streaming data, yielding Streaming PDAC (SPDAC) for streaming scenarios. Extensive comparative experiments show that our methods outperforms other baselines across various CL settings while ensuring favorable efficiency.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑