Incremental class learning using variational autoencoders with similarity learning

被引:0
|
作者
Huo, Jiahao [1 ]
van Zyl, Terence L. [1 ]
机构
[1] Univ Johannesburg, Inst Intelligent Syst, Johannesburg, South Africa
基金
新加坡国家研究基金会;
关键词
Catastrophic forgetting; Incremental learning; Similarity learning; Convolutional neural network (CNN); IMAGE SIMILARITY; RECOGNITION; NETWORKS;
D O I
10.1007/s00521-023-08485-1
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Catastrophic forgetting in neural networks during incremental learning remains a challenging problem. Previous research investigated catastrophic forgetting in fully connected networks, with some earlier work exploring activation functions and learning algorithms. Applications of neural networks have been extended to include similarity learning. Understanding how similarity learning loss functions would be affected by catastrophic forgetting is of significant interest. Our research investigates catastrophic forgetting for four well-known similarity-based loss functions during incremental class learning. The loss functions are Angular, Contrastive, Center, and Triplet loss. Our results show that the catastrophic forgetting rate differs across loss functions on multiple datasets. The Angular loss was least affected, followed by Contrastive, Triplet loss, and Center loss with good mining techniques. We implemented three existing incremental learning techniques, iCaRL, EWC, and EBLL. We further proposed a novel technique using Variational Autoencoders (VAEs) to generate representation as exemplars passed through the network's intermediate layers. Our method outperformed three existing state-of-the-art techniques. We show that one does not require stored images (exemplars) for incremental learning with similarity learning. The generated representations from VAEs help preserve regions of the embedding space used by prior knowledge so that new knowledge does not "overwrite" it.
引用
收藏
页码:769 / 784
页数:16
相关论文
共 50 条
  • [41] CGoFed: Constrained Gradient Optimization Strategy for Federated Class Incremental Learning
    Feng, Jiyuan
    Yang, Xu
    Liang, Liwen
    Han, Weihong
    Fang, Binxing
    Liao, Qing
    IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2025, 37 (05) : 2282 - 2295
  • [42] Class-Incremental Learning: Survey and Performance Evaluation on Image Classification
    Masana, Marc
    Liu, Xialei
    Twardowski, Bartlomiej
    Menta, Mikel
    Bagdanov, Andrew D.
    van de Weijer, Joost
    IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2023, 45 (05) : 5513 - 5533
  • [43] Automatic Security Classification Based on Incremental Learning and Similarity Comparison
    Liang, Yan
    Wen, Zepeng
    Tao, Yizheng
    Li, GongLiang
    Guo, Bing
    PROCEEDINGS OF 2019 IEEE 8TH JOINT INTERNATIONAL INFORMATION TECHNOLOGY AND ARTIFICIAL INTELLIGENCE CONFERENCE (ITAIC 2019), 2019, : 812 - 817
  • [44] Improve the Performance and Stability of Incremental Learning by a Similarity Harmonizing Mechanism
    Ma, Jing
    Liao, Mingjie
    Zhang, Lei
    IEEE ACCESS, 2022, 10 : 117429 - 117438
  • [45] Continuous transfer of neural network representational similarity for incremental learning
    Tian, Songsong
    Li, Weijun
    Ning, Xin
    Ran, Hang
    Qin, Hong
    Tiwari, Prayag
    NEUROCOMPUTING, 2023, 545
  • [46] Use of Variational Autoencoders with Unsupervised Learning to Detect Incorrect Organ Segmentations at CT
    Sandfort, Veit
    Yan, Ke
    Graffy, Peter M.
    Pickhardt, Perry J.
    Summers, Ronald M.
    RADIOLOGY-ARTIFICIAL INTELLIGENCE, 2021, 3 (04)
  • [47] Few-Shot Class Incremental Learning with Generative Feature Replay
    Shankarampeta, Abhilash Reddy
    Yamauchi, Koichiro
    PROCEEDINGS OF THE 10TH INTERNATIONAL CONFERENCE ON PATTERN RECOGNITION APPLICATIONS AND METHODS (ICPRAM), 2021, : 259 - 267
  • [48] Mitigate forgetting in few-shot class-incremental learning using different image views
    Mazumder, Pratik
    Singh, Pravendra
    NEURAL NETWORKS, 2023, 165 : 999 - 1009
  • [49] A class-incremental learning approach for learning feature-compatible embeddings
    An, Hongchao
    Yang, Jing
    Zhang, Xiuhua
    Ruan, Xiaoli
    Wu, Yuankai
    Li, Shaobo
    Hu, Jianjun
    NEURAL NETWORKS, 2024, 180
  • [50] Class-incremental learning with causal relational replay
    Nguyen, Toan
    Kieu, Duc
    Duong, Bao
    Kieu, Tung
    Do, Kien
    Nguyen, Thin
    Le, Bac
    EXPERT SYSTEMS WITH APPLICATIONS, 2024, 250