arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2308.03374cs.CV
收藏

面向类别增量学习的异构遗忘补偿

Heterogeneous Forgetting Compensation for Class-Incremental Learning

State Key Laboratory of Robotics, Shenyang Institute of AutomationChinese Academy of Sciences(中国科学院沈阳自动化研究所机器人学国家重点实验室);Institutes for Robotics and Intelligent ManufacturingChinese Academy of Sciences(中国科学院机器人与智能制造研究所);University of Chinese Academy of Sciences(中国科学院大学);South China University of Technology(华南理工大学)

Jiahua Dong, Wenqi Liang, Yang Cong, Gan Sun

AI总结:

针对类别增量学习中现有方法忽略旧类别遗忘异质性的问题,提出异构遗忘补偿模型,从表征与梯度层面设计模块与损失函数,有效提升增量学习性能。

摘要:

类别增量学习(CIL)在连续学习新类别的同时克服旧类别的灾难性遗忘方面已取得显著成效。然而,现有大多数CIL方法不合理地假设所有旧类别的遗忘速率相同,且忽略了不同旧类别间的遗忘异质性对遗忘补偿带来的负面影响。为解决上述挑战,我们提出了一种全新的异构遗忘补偿(Heterogeneous Forgetting Compensation, HFC)模型,该模型可从表征和梯度两个层面解决易遗忘与难遗忘旧类别的异构遗忘问题。具体而言,我们设计了任务语义聚合模块,从表征层面缓解异构遗忘,它通过聚合每个任务内的局部类别信息来学习任务共享的全局表征。此外,我们提出了两种全新的即插即用损失函数:梯度平衡遗忘补偿损失与梯度平衡关系蒸馏损失,从梯度层面缓解遗忘,二者通过梯度平衡补偿修正旧类别的遗忘异质性与异构关系一致性。在多个代表性数据集上的实验验证了HFC模型的有效性,代码可在https://github.com/JiahuaDong/HFC获取。

英文摘要:

Class-incremental learning (CIL) has achieved remarkable successes in learning new classes consecutively while overcoming catastrophic forgetting on old categories. However, most existing CIL methods unreasonably assume that all old categories have the same forgetting pace, and neglect negative influence of forgetting heterogeneity among different old classes on forgetting compensation. To surmount the above challenges, we develop a novel Heterogeneous Forgetting Compensation (HFC) model, which can resolve heterogeneous forgetting of easy-to-forget and hard-to-forget old categories from both representation and gradient aspects. Specifically, we design a task-semantic aggregation block to alleviate heterogeneous forgetting from representation aspect. It aggregates local category information within each task to learn task-shared global representations. Moreover, we develop two novel plug-and-play losses: a gradient-balanced forgetting compensation loss and a gradient-balanced relation distillation loss to alleviate forgetting from gradient aspect. They consider gradient-balanced compensation to rectify forgetting heterogeneity of old categories and heterogeneous relation consistency. Experiments on several representative datasets illustrate effectiveness of our HFC model. The code is available at https://github.com/JiahuaDong/HFC.

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

↑