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
来源
NEURAL COMPUTING & APPLICATIONS | 2023年 / 37卷 / 2期
基金
新加坡国家研究基金会;
关键词
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 条
  • [1] Synthesizing Data Using Variational Autoencoders for Handling Class Imbalanced Deep Learning
    Sheikh, Taimoor Shakeel
    Khan, Adil
    Fahim, Muhammad
    Ahmad, Muhammad
    ANALYSIS OF IMAGES, SOCIAL NETWORKS AND TEXTS (AIST 2019), 2020, 1086 : 270 - 281
  • [2] Learning Latent Subspaces in Variational Autoencoders
    Klys, Jack
    Snell, Jake
    Zemel, Richard
    ADVANCES IN NEURAL INFORMATION PROCESSING SYSTEMS 31 (NIPS 2018), 2018, 31
  • [3] Deep learning for photovoltaic defect detection using variational autoencoders
    Westraadt, Edward J.
    Brettenny, Warren J.
    Clohessy, Chantelle M.
    SOUTH AFRICAN JOURNAL OF SCIENCE, 2023, 119 (1-2)
  • [4] Learning a meaningful representation of cell fate using variational autoencoders
    Lowe, A. R.
    Soelistyo, C.
    Vallardi, G.
    Charras, G.
    MOLECULAR BIOLOGY OF THE CELL, 2023, 34 (02) : 71 - 71
  • [5] Class incremental learning of remote sensing images based on class similarity distillation
    Shen, Mingge
    Chen, Dehu
    Hu, Silan
    Xu, Gang
    PEERJ COMPUTER SCIENCE, 2023, 9
  • [6] Learning Manifold Dimensions with Conditional Variational Autoencoders
    Zheng, Yijia
    He, Tong
    Qiu, Yixuan
    Wipf, David
    ADVANCES IN NEURAL INFORMATION PROCESSING SYSTEMS 35, NEURIPS 2022, 2022,
  • [7] Variational Autoencoders with Triplet Loss for Representation Learning
    Isil, Cagatay
    Solmaz, Berkan
    Koc, Aykut
    2018 26TH SIGNAL PROCESSING AND COMMUNICATIONS APPLICATIONS CONFERENCE (SIU), 2018,
  • [8] Learning hard quantum distributions with variational autoencoders
    Rocchetto, Andrea
    Grant, Edward
    Strelchuk, Sergii
    Carleo, Giuseppe
    Severini, Simone
    NPJ QUANTUM INFORMATION, 2018, 4
  • [9] Learning conditional variational autoencoders with missing covariates
    Ramchandran, Siddharth
    Tikhonov, Gleb
    Lonnroth, Otto
    Tiikkainen, Pekka
    Lahdesmaki, Harri
    PATTERN RECOGNITION, 2024, 147
  • [10] InfoVAE: Balancing Learning and Inference in Variational Autoencoders
    Zhao, Shengjia
    Song, Jiaming
    Ermon, Stefano
    THIRTY-THIRD AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE / THIRTY-FIRST INNOVATIVE APPLICATIONS OF ARTIFICIAL INTELLIGENCE CONFERENCE / NINTH AAAI SYMPOSIUM ON EDUCATIONAL ADVANCES IN ARTIFICIAL INTELLIGENCE, 2019, : 5885 - 5892