Optimized Convolutional Neural Networks for Enhanced Detection of Acute Lymphoblastic Leukemia from Grayscale Cell Images

Authors

  • Yuan Zhang

DOI:

https://doi.org/10.61173/vzhk8s94

Keywords:

Acute Lymphoblastic Leukemia (ALL), Convolutional Neural Networks (CNN), Deep Learning, Image Classification

Abstract

Acute Lymphoblastic Leukemia (ALL) is a highly aggressive blood cancer that predominantly affects young children and the elderly, with significantly varied cure rates. Traditional diagnostic methods, such as the Complete Blood Count (CBC) and peripheral blood smear, while effective, are increasingly complemented by advanced machine learning techniques for enhanced accuracy in diagnosis. This study explores the application of Convolutional Neural Networks (CNNs) to improve the prediction accuracy of ALL diagnoses by systematically tuning various parameters of the CNN model. Using a dataset from Kaggle, which includes grayscale images of cancer cells, this employed a data preparation pipeline that involved image resizing, normalization, and feature extraction through Histogram of Oriented Gradients (HOG), followed by dimensionality reduction with Principal Component Analysis (PCA). The CNN model was trained using TensorFlow and Keras, focusing on optimizing key hyperparameters such as the number of epochs, batch size, and loss functions. The findings demonstrate that a configuration using 15 epochs, a batch size of 64, and the definite cross-entropy loss function achieves the highest accuracy and efficiency in classifying leukemia images. This research not only contributes to the enhancement of leukemia detection technology but also provides valuable insights into optimizing deep learning models for broader medical applications.

References

[1] Zhang B, Shi H, Wang H. Machine learning and AI in cancer prognosis, prediction, and treatment selection: a critical approach. J Multidiscip Healthc. 2023;16:1779-91.

[2] Schroeder B. Using machine learning to identify undiagnosable cancers. MIT News [Internet]. 2022 Sep 1 [cited 2024 Sep 5]. Available from: https://news.mit.edu/2022/usingmachine-learning-identify-undiagnosable-cancers-0901.

[3] Jiang Y, Zhang Z, Wang W, Huang W, Chen C, Xi S, et al. Biology-guided deep learning predicts prognosis and cancer immunotherapy response. Nat Commun. 2023.

[4] Albaradei S, Alganmi N, Albaradie A, Alharbi E, Motwalli O, Thafar MA, et al. A deep learning model predicts the presence of diverse cancer types using circulating tumor cells. Sci Rep. 2023;13(1):21114.

[5] Elsayed B, Elhadary M, Elshoeibi RM, Elshoeibi AM, Badr A, Metwally O, et al. Deep learning enhances acute lymphoblastic leukemia diagnosis and classification using bone marrow images. Front Oncol. 2023;13:1330977. PMID: 38125946; PMCID: PMC10731043.

[6] Mehradaria M. Leukemia dataset [Internet]. Kaggle. 2021 [cited 2024 Sep 5]. Available from: https://www.kaggle.com/ datasets/mehradaria/leukemia.

[7] Li Z, Liu F, Yang W, Peng S, Zhou J. A survey of convolutional neural networks: analysis, applications, and prospects. IEEE transactions on neural networks and learning systems. 2021 Jun 10;33(12):6999-7019.

[8] Gu J, Wang Z, Kuen J, Ma L, Shahroudy A, Shuai B, Liu T, Wang X, Wang G, Cai J, Chen T. Recent advances in convolutional neural networks. Pattern recognition. 2018 May 1;77:354-77.

[9] Yamashita R, Nishio M, Do RK, Togashi K. Convolutional neural networks: an overview and application in radiology. Insights into imaging. 2018 Aug;9:611-29.

[10] Wu J. Introduction to convolutional neural networks. National Key Lab for Novel Software Technology. Nanjing University. China. 2017 May 1;5(23):495.

Downloads

Published

2024-12-31