Affiliation:
1. Department of Information and Computer Engineering, Chung Yuan Christian University, Taoyuan City 320314, Taiwan
2. Department of Computer Science and Information Engineering, National Central University, Taoyuan City 320314, Taiwan
Abstract
Conventional deep neural networks face challenges in handling the increasing amount of information in real-world scenarios where it is impractical to gather all the training data at once. Incremental learning, also known as continual learning, provides a solution for lightweight and sustainable learning with neural networks. However, incremental learning encounters issues such as “catastrophic forgetting” and the “stability–plasticity dilemma”. To address these challenges, this study proposes a two-stage training method. In the first stage, dual knowledge distillation is introduced, including feature map-based and response-based knowledge distillation. This approach prevents the model from excessively favoring new tasks during training, thus addressing catastrophic forgetting. In the second stage, an out-of-distribution dataset is incorporated to calculate the discrepancy loss between multiple classifiers. By maximizing the discrepancy loss and minimizing the cross-entropy loss, the model improves the classification accuracy of new tasks. The proposed method is evaluated using the CIFAR100 and RESISC45 benchmark datasets, comparing it to existing approaches. Experimental results demonstrate an overall accuracy improvement of 6.9% and a reduction of 5.1% in the forgetting rate after adding nine consecutive tasks. These findings indicate that the proposed method effectively mitigates catastrophic forgetting and provides a viable solution for image classification in natural and remote sensing images.
Funder
National Science and Technology Council
Reference52 articles.
1. Li, J., Wu, Y., Zhang, H., and Wang, H. (2023). A Novel Unsupervised Segmentation Method of Canopy Images from UAV Based on Hybrid Attention Mechanism. Electronics, 12.
2. McCloskey, M., and Cohen, N.J. (1989). Psychology of Learning and Motivation, Elsevier.
3. Devkota, N., and Kim, B.W. (2023). Deep Learning-Based Small Target Detection for Satellite–Ground Free Space Optical Communications. Electronics, 12.
4. Ma, S., Chen, J., Wu, S., and Li, Y. (2023). Landslide Susceptibility Prediction Using Machine Learning Methods: A Case Study of Landslides in the Yinghu Lake Basin in Shaanxi. Sustainability, 15.
5. Three types of incremental learning;Tuytelaars;Nat. Mach. Intell.,2022
Cited by
1 articles.
订阅此论文施引文献
订阅此论文施引文献,注册后可以免费订阅5篇论文的施引文献,订阅后可以查看论文全部施引文献