arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2403.12486cs.LGcs.AI
收藏

NTK引导的小样本类别增量学习

NTK-Guided Few-Shot Class Incremental Learning

Shanghai Artificial Intelligence Laboratory(上海人工智能实验室);College of Information Science and Electronic Engineering, Zhejiang University(浙江大学信息科学与电子工程学院)

Jingren Liu, Zhong Ji, Yanwei Pang, YunLong Yu

AI总结:

本文从NTK视角建模FSCIL抗遗忘问题,通过元学习、自监督预训练、课程对齐和双重NTK正则化实现最优收敛与更低泛化损失,准确率较现有方法提升2.9%至9.3%。

摘要:

小样本类别增量学习(FSCIL)方法的激增,凸显了在FSCIL学习器中保持强大抗遗忘能力这一关键挑战。本文借助神经正切核(NTK)视角,从数学泛化的角度提出了一种新的抗遗忘概念化方法。我们的方法聚焦两个关键方面:确保最优NTK收敛,以及最小化与NTK相关的泛化损失,二者构成跨任务泛化的理论基础。为实现全局NTK收敛,我们引入了一种有原则的元学习机制,在扩展的网络架构内引导优化。同时,为降低与NTK相关的泛化损失,我们系统性地优化其构成因素。具体而言,我们在基础会话上启动自监督预训练,以增强与NTK相关的泛化潜力。随后,这些自监督权重通过课程对齐被精细调整,接着应用专门针对卷积层和线性层定制的双重NTK正则化。通过这些措施的共同作用,我们的网络获得了稳健的NTK特性,确保NTK矩阵的最优收敛性和稳定性,并最小化与NTK相关的泛化损失,显著增强其理论泛化能力。在流行的FSCIL基准数据集上,我们的NTK-FSCIL超越了同期最先进方法,将会话结束准确率提升了2.9%至9.3%。

英文摘要:

The proliferation of Few-Shot Class Incremental Learning (FSCIL) methodologies has highlighted the critical challenge of maintaining robust anti-amnesia capabilities in FSCIL learners. In this paper, we present a novel conceptualization of anti-amnesia in terms of mathematical generalization, leveraging the Neural Tangent Kernel (NTK) perspective. Our method focuses on two key aspects: ensuring optimal NTK convergence and minimizing NTK-related generalization loss, which serve as the theoretical foundation for cross-task generalization. To achieve global NTK convergence, we introduce a principled meta-learning mechanism that guides optimization within an expanded network architecture. Concurrently, to reduce the NTK-related generalization loss, we systematically optimize its constituent factors. Specifically, we initiate self-supervised pre-training on the base session to enhance NTK-related generalization potential. These self-supervised weights are then carefully refined through curricular alignment, followed by the application of dual NTK regularization tailored specifically for both convolutional and linear layers. Through the combined effects of these measures, our network acquires robust NTK properties, ensuring optimal convergence and stability of the NTK matrix and minimizing the NTK-related generalization loss, significantly enhancing its theoretical generalization. On popular FSCIL benchmark datasets, our NTK-FSCIL surpasses contemporary state-of-the-art approaches, elevating end-session accuracy by 2.9\% to 9.3\%.

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

↑