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

被引:1
|
作者
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
相关论文
共 50 条
  • [1] Class Impression for Data-Free Incremental Learning
    Ayromlou, Sana
    Abolmaesumi, Purang
    Tsang, Teresa
    Li, Xiaoxiao
    MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION, MICCAI 2022, PT IV, 2022, 13434 : 320 - 329
  • [2] Maintaining Distinction and Fairness in Data-Free Class Incremental Learning
    Feng, Yu
    Zhao, Pengcheng
    Guo, Yadong
    Zhao, Xue
    Hao, Wangli
    Zhao, Ran
    Li, Fuzhong
    2023 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS, IJCNN, 2023,
  • [3] 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
    2021 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2021), 2021, : 9354 - 9364
  • [4] Data-Free Class-Incremental Hand Gesture Recognition
    Aich, Shubhra
    Ruiz-Santaquiteria, Jesus
    Lu, Zhenyu
    Garg, Prachi
    Joseph, K. J.
    Garcia, Alvaro Fernandez
    Balasubramanian, Vineeth N.
    Kin, Kenrick
    Wan, Chengde
    Camgoz, Necati Cihan
    Ma, Shugao
    De la Torre, Fernando
    2023 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2023), 2023, : 20901 - 20910
  • [5] Dual-Teacher Class-Incremental Learning With Data-Free Generative Replay
    Choi, Yoojin
    El-Khamy, Mostafa
    Lee, Jungwon
    2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS, CVPRW 2021, 2021, : 3538 - 3547
  • [6] R-DFCIL: Relation-Guided Representation Learning for Data-Free Class Incremental Learning
    Gao, Qiankun
    Zhao, Chen
    Ghanem, Bernard
    Zhang, Jian
    COMPUTER VISION, ECCV 2022, PT XXIII, 2022, 13683 : 423 - 439
  • [7] Is Class-Incremental Enough for Continual Learning?
    Cossu, Andrea
    Graffieti, Gabriele
    Pellegrini, Lorenzo
    Maltoni, Davide
    Bacciu, Davide
    Carta, Antonio
    Lomonaco, Vincenzo
    FRONTIERS IN ARTIFICIAL INTELLIGENCE, 2022, 5
  • [8] ICICLE: Interpretable Class Incremental Continual Learning
    Rymarczyk, Dawid
    van de Weijer, Joost
    Zielinski, Bartosz
    Twardowski, Bartlomiej
    2023 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION, ICCV, 2023, : 1887 - 1898
  • [9] Learning Class-Specific Features with Class Regularization for Videos
    Stergiou, Alexandros
    Poppe, Ronald
    Veltkamp, Remco C.
    APPLIED SCIENCES-BASEL, 2020, 10 (18):
  • [10] Memory efficient data-free distillation for continual learning
    Li, Xiaorong
    Wang, Shipeng
    Sun, Jian
    Xu, Zongben
    PATTERN RECOGNITION, 2023, 144