解决持续学习中分类器与主干网络不匹配问题,提升模型鲁棒性。
LCA: Local Classifier Alignment for Continual Learning
- 提出局部分类器对齐损失,动态调整分类器以适配更新的主干网络。
- 在多个基准测试中表现领先,部分场景超越现有最佳方法。
- 适合需要长期学习且避免灾难性遗忘的智能系统应用。
智能系统需在动态环境中持续学习,但传统方法常面临灾难性遗忘问题。利用预训练模型作为特征提取器可加速并增强适应能力。早期方法仅微调首个任务,随任务增多和数据分布差异加剧而性能下降;近期研究则将任务知识融合进统一主干或动态调整主干,但可能造成任务特定分类器与更新后的主干之间存在潜在不匹配。为此,本文提出新的局部分类器对齐(LCA)损失,使分类器更紧密匹配主干。理论上证明该损失能提升分类器对所有已见任务的泛化能力与鲁棒性。结合模型融合策略,构建完整持续学习方案。在多个标准基准上实验表明,本方法性能优异,常达领先水平,部分情况下显著超越现有最先进方法。
原文摘要 · Abstract (English)
A fundamental requirement for intelligent systems is the ability to learn continuously under changing environments. However, models trained in this regime often suffer from catastrophic forgetting. Leveraging pre-trained models has recently emerged as a promising solution, since their generalized feature extractors enable faster and more robust adaptation. While some earlier works mitigate forgetting by fine-tuning only on the first task, this approach quickly deteriorates as the number of tasks grows and the data distributions diverge. More recent research instead seeks to consolidate task knowledge into a unified backbone, or adapting the backbone as new tasks arrive. However, such approaches may create a (potential) \textit{mismatch} between task-specific classifiers and the adapted backbone. To address this issue, we propose a novel \textit{Local Classifier Alignment} (LCA) loss to better align the classifier with backbone. Theoretically, we show that this LCA loss can enable the classifier to not only generalize well for all observed tasks, but also improve robustness. Furthermore, we develop a complete solution for continual learning, following the model merging approach and using LCA. Extensive experiments on several standard benchmarks demonstrate that our method often achieves leading performance, sometimes surpasses the state-of-the-art methods with a large margin.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。