CCSI: Continual Class-Specific Impression for data-free class incremental learning

被引:2
作者
Ayromlou, Sana [1 ,3 ]
Tsang, Teresa [2 ]
Abolmaesumi, Purang [1 ]
Li, Xiaoxiao [1 ,3 ]
机构
[1] Univ British Columbia, Elect & Comp Engn Dept, Vancouver, BC V6T 1Z4, Canada
[2] Vancouver Gen Hosp, Vancouver, BC V5Z 1M9, Canada
[3] Vector Inst, Toronto, ON M5G 0C6, Canada
基金
加拿大自然科学与工程研究理事会;
关键词
Class incremental learning; Data synthesis; Echo-cardiograms; Computed tomography; Microscopy imaging; CLASSIFICATION;
D O I
10.1016/j.media.2024.103239
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
In real -world clinical settings, traditional deep learning -based classification methods struggle with diagnosing newly introduced disease types because they require samples from all disease classes for offline training. Class incremental learning offers a promising solution by adapting a deep network trained on specific disease classes to handle new diseases. However, catastrophic forgetting occurs, decreasing the performance of earlier classes when adapting the model to new data. Prior proposed methodologies to overcome this require perpetual storage of previous samples, posing potential practical concerns regarding privacy and storage regulations in healthcare. To this end, we propose a novel data -free class incremental learning framework that utilizes data synthesis on learned classes instead of data storage from previous classes. Our key contributions include acquiring synthetic data known as Continual Class -Specific Impression (CCSI) for previously inaccessible trained classes and presenting a methodology to effectively utilize this data for updating networks when introducing new classes. We obtain CCSI by employing data inversion over gradients of the trained classification model on previous classes starting from the mean image of each class inspired by common landmarks shared among medical images and utilizing continual normalization layers statistics as a regularizer in this pixelwise optimization process. Subsequently, we update the network by combining the synthesized data with new class data and incorporate several losses, including an intra-domain contrastive loss to generalize the deep network trained on the synthesized data to real data, a margin loss to increase separation among previous classes and new ones, and a cosine -normalized cross -entropy loss to alleviate the adverse effects of imbalanced distributions in training data. Extensive experiments show that the proposed framework achieves state-of-theart performance on four of the public MedMNIST datasets and in-house echocardiography cine series, with an improvement in classification accuracy of up to 51% compared to baseline data -free methods. Our code is available at https://github.com/ubc-tea/Continual-Impression-CCSI.
引用
收藏
页数:14
相关论文
共 112 条
  • [1] Rusu AA, 2016, Arxiv, DOI [arXiv:1606.04671, 10.48550/arXiv.1606.04671, DOI 10.48550/ARXIV.1606.04671]
  • [2] Deep Learning with Differential Privacy
    Abadi, Martin
    Chu, Andy
    Goodfellow, Ian
    McMahan, H. Brendan
    Mironov, Ilya
    Talwar, Kunal
    Zhang, Li
    [J]. CCS'16: PROCEEDINGS OF THE 2016 ACM SIGSAC CONFERENCE ON COMPUTER AND COMMUNICATIONS SECURITY, 2016, : 308 - 318
  • [3] A dataset of microscopic peripheral blood cell images for development of automatic recognition systems
    Acevedo, Andrea
    Merino, Anna
    Alferez, Santiago
    Molina, Angel
    Boldu, Laura
    Rodellar, Jose
    [J]. DATA IN BRIEF, 2020, 30
  • [4] SS-IL: Separated Softmax for Incremental Learning
    Ahn, Hongjoon
    Kwak, Jihwan
    Lim, Subin
    Bang, Hyeonsu
    Kim, Hyojun
    Moon, Taesup
    [J]. 2021 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2021), 2021, : 824 - 833
  • [5] Expert Gate: Lifelong Learning with a Network of Experts
    Aljundi, Rahaf
    Chakravarty, Punarjay
    Tuytelaars, Tinne
    [J]. 30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, : 7120 - 7129
  • [6] Class Impression for Data-Free Incremental Learning
    Ayromlou, Sana
    Abolmaesumi, Purang
    Tsang, Teresa
    Li, Xiaoxiao
    [J]. MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION, MICCAI 2022, PT IV, 2022, 13434 : 320 - 329
  • [7] Ba J. L., 2016, arXiv, DOI DOI 10.48550/ARXIV.1607.06450
  • [8] Rainbow Memory: Continual Learning with a Memory of Diverse Samples
    Bang, Jihwan
    Kim, Heesu
    Yoo, YoungJoon
    Ha, Jung-Woo
    Choi, Jonghyun
    [J]. 2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, : 8214 - 8223
  • [9] Generating Highly Realistic Images of Skin Lesions with GANs
    Baur, Christoph
    Albarqouni, Shadi
    Navab, Nassir
    [J]. OR 2.0 CONTEXT-AWARE OPERATING THEATERS, COMPUTER ASSISTED ROBOTIC ENDOSCOPY, CLINICAL IMAGE-BASED PROCEDURES, AND SKIN IMAGE ANALYSIS, OR 2.0 2018, 2018, 11041 : 260 - 267
  • [10] Culprit-Prune-Net: Efficient Continual Sequential Multi-domain Learning with Application to Skin Lesion Classification
    Bayasi, Nourhan
    Hamarneh, Ghassan
    Garbi, Rafeef
    [J]. MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION - MICCAI 2021, PT VII, 2021, 12907 : 165 - 175