Abstract
Leukemia is a severe blood disorder that poses a threat to the life and well-being of patients. Accurate diagnosis of leukemia typing is crucial for treatment and prognosis. However, manual diagnosis requires significant resources and is subject to observer variation. Computer-aided diagnostic (CAD) technologies have emerged as a promising approach to complement manual diagnosis and alleviate the workload of specialized physicians. With the recent advances in deep learning, the performance of CAD has been substantially enhanced. In this paper, we propose a novel classification model, AMLNet, for acute myeloid leukemia (AML) cells. AMLNet is a lightweight model based on the ShuffleNet V2 architecture. We first validate the performance of the convolutional neural network and Transformer architecture models on the largest AML cell dataset to date. Then, we enhance the data augmentation approach and model structure of the ShuffleNet V2 model to develop AMLNet. The AUC, sensitivity, and accuracy of AMLNet are up to 0.9935%, 63.91%, and 96.43%, respectively, which achieves the optimal classification performance. Finally, the classification performance of AMLNet is verified on other public datasets, which is superior to other methods. AMLNet presents a promising solution for practical diagnosis scenarios since the end devices of assisted diagnosis often have limited computational power. The proposed AMLNet model effectively improves the classification performance of ShuffleNet V2 while offering the advantages of high performance and low computational consumption. Overall, our study demonstrates the potential of lightweight deep learning models in improving the accuracy of leukemia typing and reducing the workload of specialized physicians.

















Similar content being viewed by others
Data availability
The datasets used and/or analyzed during the current study are available from the corresponding author upon reasonable request.
References
Saleem S, Amin J, Sharif M, Anjum MA, Iqbal M, Wang SH (2022) A deep network designed for segmentation and classification of leukemia using fusion of the transfer learning models. Complex Intell Syst 8:3105–3120. https://doi.org/10.1007/s40747-021-00473-z
Claro ML, de Vseras MS, Santana AM, Vogado LH, Junior GB, de Medeiros FN, Tavares JM (2022) Assessing the impact of data augmentation and a combination of CNNs on leukemia classification. Inf Sci 609:1010–29
Rastogi P, Khanna K, Singh V (2022) LeuFeatx: deep learning–based feature extractor for the diagnosis of acute leukemia from microscopic images of peripheral blood smear. Comput Biol Med 142:105236. https://doi.org/10.1016/j.compbiomed.2022.105236
Raina R, Gondhi NK, Chaahat D, Singh M, Lee Kaur H.N. (2020) A systematic review on acute Leukemia detection using deep learning techniques. Arch Comput Methods Eng 30:251–270. https://doi.org/10.1007/s11831-022-09796-7
Abhishek A, Jha RK, Sinha R, Jha K (2022) Automated classification of acute leukemia on a heterogeneous dataset using machine learning and deep learning techniques. Biomed Signal Process Control 72:103341. https://doi.org/10.1016/j.bspc.2021.103341
Rodrigues LF, Backes AR, Travençolo BAN, de Oliveira GMB (2022) Optimizing a deep residual neural network with genetic algorithm for acute lymphoblastic Leukemia classification. J Digit Imaging 35:623–637. https://doi.org/10.1007/s10278-022-00600-3
Liu K, Hu J (2022) Classification of acute myeloid leukemia M1 and M2 subtypes using machine learning. Comput Biol Med 147:105741. https://doi.org/10.1016/j.compbiomed.2022.105741
Laosai J, Chamnongthai K (2018) Classification of acute leukemia using medical-knowledge-based morphology and CD marker. Biomed Signal Process Control 44:127–137. https://doi.org/10.1016/j.bspc.2018.01.020
Merino A, Boldú L, Ermens A (2018) Acute myeloid leukaemia: how to combine multiple tools. Int J Lab Hematol 40:109–119. https://doi.org/10.1111/ijlh.12831
Döhner H, Weisdorf DJ, Bloomfield CD (2015) Acute myeloid leukemia. N Engl J Med 373:1136–1152. https://doi.org/10.1056/NEJMra1406184
Weinberg OK, Porwit A, Orazi A, Hasserjian RP, Foucar K, Duncavage EJ, Arber DA (2022) The International consensus classification of acute myeloid leukemia. Virchows Arch. https://doi.org/10.1007/s00428-022-03430-4
Arber DA, Orazi A, Hasserjian R, Thiele J, Borowitz MJ, Le Beau MM, Bloomfield CD, Cazzola M, Vardiman JW (2016) The 2016 revision to the World Health Organization classification of myeloid neoplasms and acute leukemia. Blood 127:2391–2405. https://doi.org/10.1182/blood-2016-03-643544
Reichard KK, Tefferi A, Abdelmagid M, Orazi A, Alexandres C, Haack J, Greipp PT (2022) Pure (acute) erythroid leukemia: morphology, immunophenotype, cytogenetics, mutations, treatment details, and survival data among 41 Mayo Clinic cases, Blood. Cancer J 12:1–8. https://doi.org/10.1038/s41408-022-00746-x
Das PK, Diya VA, Meher S, Panda R, Abraham A (2022) A Systematic review on recent advancements in deep and machine learning based detection and classification of acute lymphoblastic Leukemia. IEEE Access 10:81741–81763. https://doi.org/10.1109/ACCESS.2022.3196037
Matek C, Schwarz S, Spiekermann K, Marr C (2019) Human-level recognition of blast cells in acute myeloid leukaemia with convolutional neural networks. Nat Mach Intell 1:538–544. https://doi.org/10.1038/s42256-019-0101-9
Eckardt JN, Middeke JM, Riechert S, Schmittmann T, Sulaiman AS, Kramer M, Sockel K, Kroschinsky F, Schuler U, Schetelig J, Röllig C, Thiede C, Wendt K, Bornhäuser M (2022) Deep learning detects acute myeloid leukemia and predicts NPM1 mutation status from bone marrow smears. Leukemia 36:111–118. https://doi.org/10.1038/s41375-021-01408-w
Eckardt JN, Schmittmann T, Riechert S, Kramer M, Sulaiman AS, Sockel K, Kroschinsky F, Schetelig J, Wagenführ L, Schuler U, Platzbecker U, Thiede C, Stölzel F, Röllig C, Bornhäuser M, Wendt K, Middeke JM (2022) Deep learning identifies acute Promyelocytic Leukemia in bone marrow smears. BMC Cancer 22:1–11. https://doi.org/10.1186/s12885-022-09307-8
Boldú L, Merino A, Acevedo A, Molina A, Rodellar J (2021) A deep learning model (ALNet) for the diagnosis of acute leukaemia lineage using peripheral blood cell images. Comput Methods Programs Biomed 202:105999. https://doi.org/10.1016/j.cmpb.2021.105999
Alsalem MA, Zaidan AA, Zaidan BB, Hashim M, Madhloom HT, Azeez ND, Alsyisuf S (2018) A review of the automated detection and classification of acute leukaemia: Coherent taxonomy, datasets, validation and performance measurements, motivation, open challenges and recommendations. Comput Methods Programs Biomed 158:93–112. https://doi.org/10.1016/j.cmpb.2018.02.005
Kratz A, Lee SH, Zini G, Riedl JA, Hur M, Machin S (2019) Digital morphology analyzers in hematology: ICSH review and recommendations. Int J Lab Hematol 41(4):437–47. https://doi.org/10.1111/ijlh.13042
Mishra S, Majhi B, Sa PK (2019) Texture feature based classification on microscopic blood smear for acute lymphoblastic leukemia detection, Biomed. Signal Process. Control 47:303–311. https://doi.org/10.1016/j.bspc.2018.08.012
Das PK, Meher S (2021) An efficient deep convolutional neural network based detection and classification of acute lymphoblastic Leukemia. Expert Syst Appl 183:115311. https://doi.org/10.1016/j.eswa.2021.115311
Ma L, Shuai R, Ran X, Liu W, Ye C (2020) Combining DC-GAN with ResNet for blood cell image classification. Med Biol Eng Comput 58:1251–1264. https://doi.org/10.1007/s11517-020-02163-3
Labati RD, Piuri V, Scotti F (2011) All-IDB: THE ACUTE LYMPHOBLASTIC LEUKEMIA IMAGE DATABASE FOR IMAGE PROCESSING. In: 2011 18th IEEE International Conference on Image Processing 2011 Sep 11 (pp. 2045-2048). IEEE
Gehlot S, Gupta A, Gupta R (2020) SDCT-AuxNetθ: DCT augmented stain deconvolutional CNN with auxiliary classifier for cancer diagnosis. Med Image Anal 61:101661. https://doi.org/10.1016/j.media.2020.101661
Raghu M, Zhang C, Kleinberg J, Bengio S (2019) Transfusion: understanding transfer learning for medical imaging. Adv Neural Inf Process Syst. https://doi.org/10.48550/arXiv.1902.07208
Cubuk ED, Zoph B, Mane D, Vasudevan V, Le QV (2019) AUTOAUGMENT: LEARNING AUGMENTATION STRATEGIES FROM DATA. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition 2019 (pp. 113-123). https://doi.org/10.1109/CVPR.2019.00020.
Howard AG, Zhu M, Chen B, Kalenichenko D, Wang W, Weyand T, Andreetto M, Adam H (2017) MobileNets: efficient convolutional neural networks for mobile vision applications. arX prepr arX. https://doi.org/10.48550/arXiv.1704.04861
Zhang X, Zhou X, Lin M, Sun J (2018) SHUFFLENET: AN EXTREMELY EFFICIENT CONVOLUTIONAL NEURAL NETWORK FOR MOBILE DEVICES. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition 2018 (pp. 6848-6856). https://doi.org/10.1109/CVPR.2018.00716.
Ma N, Zhang X, Zheng HT, Sun J (2018) SHUFFLENET V2: PRACTICAL GUIDELINES FOR EFFICIENT CNN ARCHITECTURE DESIGN. In: Proceedings of the European Conference on Computer Vision (ECCV) 2018 (pp. 116–131).122–138. https://doi.org/10.1007/978-3-030-0 1264-9_8.
Hu J, Shen L, Sun G (2018) SQUEEZE-AND-EXCITATION NETWORKS, In: 2018 IEEE/CVF Conf. Comput. Vis. Pattern Recognit., IEEE, 2018: pp. 7132–7141. https://doi.org/10.1109/CVPR.2018. 00745
Shenggan, BCCD dataset, https://Github.Com/Shenggan/BCCD_Dataset. (2021).
Leng B, Leng M, Ge M, Dong W (2022) Knowledge distillation-based deep learning classification network for peripheral blood leukocytes. Biomed Signal Process Control 75:103590. https://doi.org/10.1016/j.bspc.2022.103590
Selvaraju RR, Cogswell M, Das A, Vedantam R, Parikh D, Batra D (2017) GRAD-CAM: VISUAL EXPLANATIONS FROM DEEP NETWORKS VIA GRADIENT-BASED LOCALIZATION. In: Proceedings of the IEEE International Conference on Computer Vision 2017 (pp. 618-626) https://doi.org/10.1109/ICCV.2017.74.
Deng J, Dong W, Socher R, Li LJ, Li K, Fei-Fei L (2009) IMAGENET: A LARGE-SCALE HIERARCHICAL IMAGE DATABASE. In: 2009 IEEE Conference on Computer Vision and Pattern Recognition 2009 Jun 20 (pp. 248–255).https://doi.org/10.1109/CVPR.2009.5206848
McInnes L, Healy J, Saul N, Großberger L (2018) UMAP uniform manifold approximation and projection. J Open Source Softw 3:861. https://doi.org/10.21105/joss.00861
Yang D, Zhao H, Han T, Kang Q, Ma J, Lu H (2020) Leukocyte subtypes identification using bilinear self-attention convolutional neural network. Measurement 173:108643. https://doi.org/10.1016/j.measurement.2020.108643
Khan A, Eker A, Chefranov A, Demirel H (2021) White blood cell type identification using multi-layer convolutional features with an extreme-learning machine. Biomed Signal Process Control 69:102932. https://doi.org/10.1016/j.bspc.2021.102932
Funding
This research was funded by CAS Engineering Laboratory for In Vitro Diagnostic (Grant No. 20210419115032678), the Strategic Priority Research Program of the Chinese Academy of Sciences (Grant No. XDA16021102), and Suzhou Key Technology (Research) Project of Critical and Infectious Diseases Precaution and Control (Grant No. GWZX202102).
Author information
Authors and Affiliations
Contributions
BL was involved in the investigation, formal analysis, and writing—original draft. HJ contributed to writing—review and editing. BW performed the conceptualization. JW assisted in the investigation and supervision. GL contributed to the project administration and funding acquisition.
Corresponding authors
Ethics declarations
Competing interest
The authors have no conflicts of interest to declare that are relevant to the content of this article.
Ethical approval
The article does not contain any studies with human participants or animals performed by any of the authors.
Additional information
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.
About this article
Cite this article
Leng, B., Jiang, H., Wang, B. et al. A lightweight deep learning model for acute myeloid leukemia-related blast cell identification. J Supercomput 80, 15215–15244 (2024). https://doi.org/10.1007/s11227-024-06063-3
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s11227-024-06063-3