用信息论设计有益噪声,提升预训练模型持续学习能力
Mixture of Noise for Pre-Trained Model-Based Class-Incremental Learning
- 基于高维特征学习任务专属噪声,动态混合以优化
- 在6个数据集上50步增量设置下达到顶尖性能
- 适合研究持续学习与预训练模型优化的学者
类别增量学习(CIL)旨在持续学习新类别同时保留旧知识。预训练模型(PTMs)在CIL中表现优异,但现有轻量微调方法仍导致参数漂移,损害模型泛化能力。参数漂移可视为遮蔽先前任务关键模式的噪声。然而,近期研究表明噪声并非总是有害:预训练中大量视觉模式易被单一任务滥用,引入适度噪声可抑制低相关特征,为未来任务留出空间。为此,我们提出基于信息论的有益噪声学习方法——混合噪声(Min),旨在缓解适应新任务时骨干网络泛化能力下降问题。具体地,从新任务的高维特征中学习任务专属噪声,动态调整权重以最优混合不同任务噪声,并将有益噪声嵌入中间特征,掩蔽低效模式响应。在六个基准数据集上的大量实验表明,Min在多数增量设置下达到当前最优性能,尤其在50步增量设置中表现突出,展现了有益噪声在持续学习中的巨大潜力。代码已开源。
原文摘要 · Abstract (English)
Class Incremental Learning (CIL) aims to continuously learn new categories while retaining the knowledge of old ones. Pre-trained models (PTMs) show promising capabilities in CIL. However, existing approaches that apply lightweight fine-tuning to backbones still induce parameter drift, thereby compromising the generalization capability of pre-trained models. Parameter drift can be conceptualized as a form of noise that obscures critical patterns learned for previous tasks. However, recent researches have shown that noise is not always harmful. For example, the large number of visual patterns learned from pre-training can be easily abused by a single task, and introducing appropriate noise can suppress some low-correlation features, thus leaving a margin for future tasks. To this end, we propose learning beneficial noise for CIL guided by information theory and propose Mixture of Noise (Min), aiming to mitigate the degradation of backbone generalization from adapting new tasks. Specifically, task-specific noise is learned from high-dimension features of new tasks. Then, a set of weights is adjusted dynamically for optimal mixture of different task noise. Finally, Min embeds the beneficial noise into the intermediate features to mask the response of inefficient patterns. Extensive experiments on six benchmark datasets demonstrate that Min achieves state-of-the-art performance in most incremental settings, with particularly outstanding results in 50-steps incremental settings. This shows the significant potential for beneficial noise in continual learning. Code is available at https://github.com/ASCIIJK/MiN-NeurIPS2025.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。