Dual Balanced Class-Incremental Learning With im-Softmax and Angular Rectification
文献类型:期刊论文
作者 | Ruicong Zhi3; Yicheng Meng3; Junyi Hou2; Jun, Wan1![]() |
刊名 | IEEE Transactions on Neural Networks and Learning Systems
![]() |
出版日期 | 2024 |
页码 | 1-16 |
英文摘要 | Owing to the superior performances, exemplarbased methods with knowledge distillation (KD) are widely applied in class incremental learning (CIL). However, it suffers from two drawbacks: 1) data imbalance between the old/learned and new classes causes the bias of the new classifier toward the head/new classes and 2) deep neural networks (DNNs) suffer from distribution drift when learning sequence tasks, which results in narrowed feature space and deficient representation of old tasks. For the first problem, we analyze the insufficiency of softmax loss when dealing with the problem of data imbalance in theory and then propose the imbalance softmax (im-softmax) loss to relieve the imbalanced data learning, where we re-scale the output logits to underfit the head/new classes. For another problem, we calibrate the feature space by incremental-adaptive angular margin (IAAM) loss. The new classes form a complete distribution in feature space yet the old are squeezed. To recover the old feature space, we first compute the included angle of normalized features and normalized anchor prototypes, and use the angle distribution to represent the class distribution, then we replenish the old distribution with the deviation from the new. Each anchor prototype is predefined as a learnable vector for a designated class. The proposed im-softmax reduces the bias in the linear classification layer. IAAM rectifies the representation learning, reduces the intra-class distance, and enlarges the inter-class margin. Finally, we seamlessly combine the im-softmax and IAAM in an end-to-end training framework, called the dual balanced class incremental learning (DBL), for further improvements. |
URL标识 | 查看原文 |
源URL | [http://ir.ia.ac.cn/handle/173211/57466] ![]() |
专题 | 自动化研究所_模式识别国家重点实验室_生物识别与安全技术研究中心 |
通讯作者 | Jun, Wan |
作者单位 | 1.Institute of Automation, Chinese Academy of Sciences, China 2.National University of Singapore 3.University of Science and Technology Beijing |
推荐引用方式 GB/T 7714 | Ruicong Zhi,Yicheng Meng,Junyi Hou,et al. Dual Balanced Class-Incremental Learning With im-Softmax and Angular Rectification[J]. IEEE Transactions on Neural Networks and Learning Systems,2024:1-16. |
APA | Ruicong Zhi,Yicheng Meng,Junyi Hou,&Jun, Wan.(2024).Dual Balanced Class-Incremental Learning With im-Softmax and Angular Rectification.IEEE Transactions on Neural Networks and Learning Systems,1-16. |
MLA | Ruicong Zhi,et al."Dual Balanced Class-Incremental Learning With im-Softmax and Angular Rectification".IEEE Transactions on Neural Networks and Learning Systems (2024):1-16. |
入库方式: OAI收割
来源:自动化研究所
其他版本
除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。