Diagnosing Batch Normalization in Class Incremental Learning
This addresses catastrophic forgetting in class incremental learning for deep neural networks, offering an incremental improvement to existing rehearsal-based methods.
The paper identifies that standard batch normalization (BN) harms representation learning and classifier training in class incremental learning, exacerbating catastrophic forgetting, and proposes BN Tricks to improve feature extraction and reduce bias, achieving significant performance gains on benchmark datasets.
Extensive researches have applied deep neural networks (DNNs) in class incremental learning (Class-IL). As building blocks of DNNs, batch normalization (BN) standardizes intermediate feature maps and has been widely validated to improve training stability and convergence. However, we claim that the direct use of standard BN in Class-IL models is harmful to both the representation learning and the classifier training, thus exacerbating catastrophic forgetting. In this paper we investigate the influence of BN on Class-IL models by illustrating such BN dilemma. We further propose BN Tricks to address the issue by training a better feature extractor while eliminating classification bias. Without inviting extra hyperparameters, we apply BN Tricks to three baseline rehearsal-based methods, ER, DER++ and iCaRL. Through comprehensive experiments conducted on benchmark datasets of Seq-CIFAR-10, Seq-CIFAR-100 and Seq-Tiny-ImageNet, we show that BN Tricks can bring significant performance gains to all adopted baselines, revealing its potential generality along this line of research.