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 条
  • [31] LEARNING CLASS-SPECIFIC POOLING SHAPES FOR IMAGE CLASSIFICATION
    Wang, Jinzhuo
    Wang, Wenmin
    Wang, Ronggang
    Gao, Wen
    2015 IEEE INTERNATIONAL CONFERENCE ON MULTIMEDIA & EXPO (ICME), 2015,
  • [32] Learning a Class-Specific Dictionary for Facial Expression Recognition
    Zhang, Shiqing
    Zhang, Gang
    Cui, Yueli
    Zhao, Xiaoming
    CYBERNETICS AND INFORMATION TECHNOLOGIES, 2016, 16 (04) : 55 - 62
  • [33] WEAKLY-SUPERVISED CONTINUAL LEARNING FOR CLASS-INCREMENTAL SEGMENTATION
    Lenczner, Gaston
    Chan-Hon-Tong, Adrien
    Luminari, Nicola
    Le Saux, Bertrand
    2022 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM (IGARSS 2022), 2022, : 4843 - 4846
  • [34] Continual Learning for Class- and Domain-Incremental Semantic Segmentation
    Kalb, Tobias
    Roschani, Masoud
    Ruf, Miriam
    Beyerer, Juergen
    2021 32ND IEEE INTELLIGENT VEHICLES SYMPOSIUM (IV), 2021, : 1345 - 1351
  • [35] An empirical evaluation of tinyML architectures for Class-Incremental Continual Learning
    Tremonti, Matteo
    Pezze, Davide Dalle
    Paissan, Francesco
    Farella, Elisabetta
    Susto, Gian Antonio
    2024 IEEE INTERNATIONAL CONFERENCE ON PERVASIVE COMPUTING AND COMMUNICATIONS WORKSHOPS AND OTHER AFFILIATED EVENTS, PERCOM WORKSHOPS, 2024, : 690 - 695
  • [36] Class-Incremental Continual Learning Into the eXtended DER-Verse
    Boschini, Matteo
    Bonicelli, Lorenzo
    Buzzega, Pietro
    Porrello, Angelo
    Calderara, Simone
    IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2023, 45 (05) : 5497 - 5512
  • [37] Online Class-Incremental Continual Learning with Adversarial Shapley Value
    Shim, Dongsub
    Mai, Zheda
    Jeong, Jihwan
    Sanner, Scott
    Kim, Hyunwoo
    Jang, Jongseong
    THIRTY-FIFTH AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE, THIRTY-THIRD CONFERENCE ON INNOVATIVE APPLICATIONS OF ARTIFICIAL INTELLIGENCE AND THE ELEVENTH SYMPOSIUM ON EDUCATIONAL ADVANCES IN ARTIFICIAL INTELLIGENCE, 2021, 35 : 9630 - 9638
  • [38] Class-specific discriminative metric learning for scene recognition
    Wang, Chen
    Peng, Guohua
    De Baets, Bernard
    PATTERN RECOGNITION, 2022, 126
  • [39] Learning class-specific edges for object detection and segmentation
    Prasad, Mukta
    Zisserman, Andrew
    Fitzgibbon, Andrew
    Kumar, M. Pawan
    Torr, P. H. S.
    COMPUTER VISION, GRAPHICS AND IMAGE PROCESSING, PROCEEDINGS, 2006, 4338 : 94 - +
  • [40] ConStruct-VL: Data-Free Continual Structured VL Concepts Learning
    Smith, James Seale
    Cascante-Bonilla, Paola
    Arbelle, Assaf
    Kim, Donghyun
    Panda, Rameswar
    Cox, David
    Yang, Diyi
    Kira, Zsolt
    Feris, Rogerio
    Karlinsky, Leonid
    2023 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2023, : 14994 - 15004