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 条
  • [71] Shin H, 2017, ADV NEUR IN, V30
  • [72] Medical Image Synthesis for Data Augmentation and Anonymization Using Generative Adversarial Networks
    Shin, Hoo-Chang
    Tenenholtz, Neil A.
    Rogers, Jameson K.
    Schwarz, Christopher G.
    Senjem, Matthew L.
    Gunter, Jeffrey L.
    Andriole, Katherine P.
    Michalski, Mark
    [J]. SIMULATION AND SYNTHESIS IN MEDICAL IMAGING, 2018, 11037 : 1 - 11
  • [73] Singh A, 2021, ADV NEURAL INFORM PR, V34
  • [74] Always Be Dreaming: A New Approach for Data-Free Class-Incremental Learning
    Smith, James
    Hsu, Yen-Chang
    Balloch, Jonathan
    Shen, Yilin
    Jin, Hongxia
    Kira, Zsolt
    [J]. 2021 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2021), 2021, : 9354 - 9364
  • [75] CODA-Prompt: COntinual Decomposed Attention-based Prompting for Rehearsal-Free Continual Learning
    Smith, James Seale
    Karlinsky, Leonid
    Gutta, Vyshnavi
    Cascante-Bonilla, Paola
    Kim, Donghyun
    Arbelle, Assaf
    Panda, Rameswar
    Feris, Rogerio
    Kira, Zsolt
    [J]. 2023 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2023, : 11909 - 11919
  • [76] Global and Local Feature Reconstruction for Medical Image Segmentation
    Song, Jiahuan
    Chen, Xinjian
    Zhu, Qianlong
    Shi, Fei
    Xiang, Dehui
    Chen, Zhongyue
    Fan, Ying
    Pan, Lingjiao
    Zhu, Weifang
    [J]. IEEE TRANSACTIONS ON MEDICAL IMAGING, 2022, 41 (09) : 2273 - 2284
  • [77] Continual Domain Incremental Learning for Chest X-Ray Classification in Low-Resource Clinical Settings
    Srivastava, Shikhar
    Yaqub, Mohammad
    Nandakumar, Karthik
    Ge, Zongyuan
    Mahapatra, Dwarikanath
    [J]. DOMAIN ADAPTATION AND REPRESENTATION TRANSFER, AND AFFORDABLE HEALTHCARE AND AI FOR RESOURCE DIVERSE GLOBAL HEALTH (DART 2021), 2021, 12968 : 226 - 238
  • [78] Exemplar-free class incremental learning via discriminative and comparable parallel one-class classifiers
    Sun, Wenju
    Li, Qingyong
    Zhang, Jing
    Wang, Danyu
    Wang, Wen
    Geng, YangLi-ao
    [J]. PATTERN RECOGNITION, 2023, 140
  • [79] Txurio Maialen Stephens, 2023, Innovation in Medicine and Healthcare: Proceedings of 11th KES-InMed 2023. Smart Innovation, Systems and Technologies (357), P335, DOI 10.1007/978-981-99-3311-2_30
  • [80] Ulyanov D, 2017, Arxiv, DOI [arXiv:1607.08022, 10.48550/arxiv.1607.08022, DOI 10.48550/ARXIV.1607.08022]