Online Class-Incremental Learning in Image Classification Based on Attention

Baoyu Du,Zhonghe Wei,Jinyong Cheng,Guohua Lv, Xiaoyu Dai

PATTERN RECOGNITION AND COMPUTER VISION, PRCV 2023, PT VII(2024)

引用 0|浏览0
暂无评分
摘要
Online class-incremental learning aims to learn from continuous and single-pass data streams. However, during the learning process, catastrophic forgetting often occurs, leading to the loss of knowledge about previous classes. Most of the current methods to solve the problem of catastrophic forgetting do not fully exploit the semantic information in single-pass data streams. To effectively address this issue, we propose an Attention-based Dual-View Consistency (ADVC) strategy. Specifically, we generate an attention score for each region through the attention mechanism to identify those important features that are useful for classification. This enables the model to thoroughly explore the semantic information in the single-pass data stream and improve prediction accuracy. At the same time, we consider the effectiveness of sample retrieval from another perspective (the samples themselves) by using data augmentation methods in memory to generate variants of old classes of samples to alleviate the problem of catastrophic forgetting. Through extensive experimental results, we demonstrate that our approach outperforms state-of-the-art methods on a range of benchmark datasets.
更多
查看译文
关键词
Class-incremental learning,attention,ADVC,data augmentation
AI 理解论文
溯源树
样例
生成溯源树,研究论文发展脉络
Chat Paper
正在生成论文摘要