A Continual Learning Approach for Cross-Domain White Blood Cell Classification
Accurate classification of white blood cells in peripheral blood is essential for diagnosing hematological diseases. Due to constantly evolving clinical settings, data sources, and disease classifications, it is necessary to update machine learning classification models regularly for practical real-world use. Such models significantly benefit from sequentially learning from incoming data streams without forgetting previously acquired knowledge. However, models can suffer from catastrophic forgetting, causing a drop in performance on previous tasks when fine-tuned on new data. Here, we propose a rehearsal-based continual learning approach for class incremental and domain incremental scenarios in white blood cell classification. To choose representative samples from previous tasks, we employ exemplar set selection based on the model's predictions. This involves selecting the most confident samples and the most challenging samples identified through uncertainty estimation of the model. We thoroughly evaluated our proposed approach on three white blood cell classification datasets that differ in color, resolution, and class composition, including scenarios where new domains or new classes are introduced to the model with every task. We also test a long class incremental experiment with both new domains and new classes. Our results demonstrate that our approach outperforms established baselines in continual learning, including existing iCaRL and EWC methods for classifying white blood cells in cross-domain environments.
Code (0)
등록된 구현이 없습니다.
Tasks
ClassificationContinual LearningMethods 이 논문이 사용한 방법론
Similar Papers 제목 키워드 기반
Domain-incremental White Blood Cell Classification with Privacy-aware Continual Learning
White blood cell (WBC) classification plays a vital role in hematology for diagnosing various medical conditions. However, it faces significant challenges due to domain shifts caused by variations in sample sources (e.g.…
Continual LearningPrivacy PreservingUnsupervised Cross-Domain Feature Extraction for Single Blood Cell Image Classification
Diagnosing hematological malignancies requires identification and classification of white blood cells in peripheral blood smears. Domain shifts caused by different lab procedures, staining, illumination, and microscope s…
image-classificationImage ClassificationPrognosisWhite blood cell subtype detection and classification
Machine learning has endless applications in the health care industry. White blood cell classification is one of the interesting and promising area of research. The classification of the white blood cells plays an import…
ClassificationMedical Diagnosisobject-detectionObject DetectionNeural Cellular Automata for Lightweight, Robust and Explainable Classification of White Blood Cell Images
Diagnosis of hematological malignancies depends on accurate identification of white blood cells in peripheral blood smears. Deep learning techniques are emerging as a viable solution to scale and optimize this process by…
Classificationimage-classificationImage ClassificationClassification of White Blood Cells Using Machine and Deep Learning Models: A Systematic Review
Machine learning (ML) and deep learning (DL) models have been employed to significantly improve analyses of medical imagery, with these approaches used to enhance the accuracy of prediction and classification. Model pred…
ClassificationData AugmentationMedical Image Analysis