Enhancing Pulmonary Disease Classification in Diseases: A Comparative Study of CNN and Optimized MobileNet Architectures
Abstract
Background: Deep learning technologies, especially Convolutional Neural Networks (CNNs), are revolutionizing the field of medical imaging by providing advanced tools for the accurate classification of pulmonary diseases from chest X-ray (CXR) images. In our study, we employed both traditional CNN models and MobileNet architectures to classify various chest diseases using CXR images. Initially, a conventional CNN model was utilized to estab- lish a baseline accuracy. Subsequently, we adopted MobileNet, known for its efficiency in processing image data, to enhance classification performance. To further optimize the system, we applied Energy Valley Optimization (EVO) for hyperparameter tuning. The baseline CNN model achieved an accuracy of 85.91%. The implementation of MobileNet significantly improved this metric, reaching a pre-optimization accuracy of 93.30%. Post-EVO optimization, the accuracy was further enhanced to 94.18%. Comparative analysis of accuracy, precision, recall, F1-score, and ROC curves was conducted to illustrate the impact of hyperparameter tuning on model performance in medical diagnostics. Our findings demonstrate that while standard CNNs provide a solid foundation for CXR image classification, the integration of MobileNet architectures and EVO for hyperparameter adjustment significantly boosts diagnostic accuracy. This advancement in automated medical image analysis could potentially transform the landscape of pulmonary disease diagnosis, offering a more robust framework for accurate and efficient patient care.
Keywords
Full Text:
PDFReferences
S. Albahli, "Efficient gan-based chest radiographs (cxr) augmentation to diagnose coronavirus disease pneumonia," International Journal of Medical Sciences, vol. 17, pp. 1439–1448, 2020.
K. H. Shibly, S. K. Dey, M. T. U. Islam, and M. M. Rahman, "Covid faster r-cnn: A novel framework to diagnose novel coronavirus disease (covid-19) in x-ray images," Informatics in Medicine Unlocked, vol. 20, p. 100405, 2020.
S. Basu and R. H. Campbell, "Going by the numbers: Learning and modeling covid-19 disease dynamics," Chaos, Solitons & Fractals, vol. 138, p. 110140, 2020.
P. K. Sethy and S. K. Behera. Detection of coronavirus disease (covid-19) based on deep features. Preprints, 2020.
S. Hassantabar, M. Ahmadi, and A. Sharifi, "Diagnosis and detection of infected tissue of covid-19 patients based on lung x-ray image using convolutional neural network approaches," Chaos, Solitons & Fractals, vol. 140, p. 110170, 2020.
D. A. Ragab, M. Sharkas, S. Marshall, and J. Ren, "Breast cancer detection using deep convolutional neural networks and support vector machines," PeerJ, vol. 7, p. e6201, 2019.
Z. Yao, J. Li, Z. Guan, Y. Ye, and Y. Chen, "Liver disease screening based on densely connected deep neural networks," Neural Networks, vol. 123, pp. 299– 304, 2020.
I. Pacal, D. Karaboga, A. Basturk, B. Akay, and U. Nalbantoglu, "A comprehensive review of deep learning in colon cancer," Computers in Biology and Medicine, vol. 126, p. 104003, 2020.
X. W. Gao, R. Hui, and Z. Tian, "Classification of ct brain images based on deep learning networks," Computer Methods and Programs in Biomedicine, vol. 138, pp. 49–56, 2017.
A. Esteva, B. Kuprel, R. A. Novoa, J. Ko, S. M. Swetter, H. M. Blau, and S. Thrun, "Dermatologist-level classification of skin cancer with deep neural networks," Nature, vol. 542, pp. 115–118, 2017.
W. Ausawalaithong, A. Thirach, S. Marukatat, and T. Wilaiprasitporn, "Automatic lung cancer prediction from chest x-ray images using the deep learning approach," in 2018 11th Biomedical Engineering International Conference (BMEiCON), pp. 1–5, 2018.
N. M. Elshennawy and D. M. Ibrahim, "Deep-pneumonia framework using deep learning models based on chest x-ray images," Diagnostics, vol. 10, p. 649, 2020.
A. Abbas, M. M. Abdelsamea, and M. M. Gaber, "Classification of covid- 19 in chest x-ray images using detrac deep convolutional neural network," arXiv preprint arXiv:2003.13815, 2020.
S. Asif, Y. Wenhui, H. Jin, and S. Jinhai, "Classification of COVID-19 from Chest X-ray images using Deep Convolutional Neural Network," 2020 IEEE 6th International Conference on Computer and Communications (ICCC), pp. 426-433, 2020, doi: 10.1109/ICCC51575.2020.9344870.
E. E.-D. Hemdan, M. A. Shouman, and M. E. Karar, "Covidx-net: A framework of deep learning classifiers to diagnose covid-19 in x-ray images," arXiv preprint arXiv:2003.11055, 2020.
C. Zheng et al., "Deep learning-based detection for covid-19 from chest ct using weak label," medRxiv, 2020.
J. Zhao, Y. Zhang, X. He, and P. Xie, "Covid-ct-dataset: A ct scan dataset about covid-19," arXiv preprint arXiv:2003.13865, 2020.
O. Gozes, M. Frid-Adar, N. Sagie, H. Zhang, W. Ji, and H. Greenspan, "Coronavirus detection and analysis on chest ct with deep learning," arXiv preprint arXiv:2004.02640, 2020.
S. Wang et al., "A deep learning algorithm using ct images to screen for coronavirus disease (covid-19)," European radiology, vol. 31, pp. 6096-6104, 2020.
L. Li et al., "Artificial intelligence distinguishes covid-19 from community- acquired pneumonia on chest ct," Radiology, vol. 296, no. 2, pp. 65–71, 2020.
J. Chen et al., "Deep learning-based model for detecting 2019 novel coronavirus pneumonia on high-resolution computed tomography," Scientific Reports, vol. 10, pp. 1–11, 2020.
B. Wang et al., "Ai-assisted ct imaging analysis for covid-19 screening: building and deploying a medical ai system," Applied Soft Computing, vol. 98, p. 106897, 2020.
M. Yamac, M. Ahishali, A. Degerli, S. Kiranyaz, M. E. Chowdhury, and M. Gabbouj, "Convolutional sparse support estimator based covid-19 recognition from x-ray images," arXiv preprint arXiv:2005.04014, 2020.
Y. Oh, S. Park, and J. C. Ye, "Deep Learning COVID-19 Features on CXR Using Limited Training Data Sets," in IEEE Transactions on Medical Imaging, vol. 39, no. 8, pp. 2688-2700, Aug. 2020, doi: 10.1109/TMI.2020.2993291.
M. Loey, F. Smarandache, and N. E. M. Khalifa, "Within the lack of chest covid-19 x-ray dataset: a novel detection model based on gan and deep transfer learning," Symmetry, vol. 12, p. 651, 2020.
T. Ozturk, M. Talo, E. A. Yildirim, U. B. Baloglu, O. Yildirim, and U. R. Acharya, "Automated detection of covid-19 cases using deep neural networks with x-ray images," Computers in biology and medicine, vol. 121, p. 103792, 2020.
N. E. M. Khalifa, M. H. N. Taha, A. E. Hassanien, and S. Elghamrawy, "Detection of coronavirus (covid-19) associated pneumonia based on generative adversarial networks and a fine-tuned deep transfer learning model using chest x-ray dataset," in International Conference on Advanced Intelligent Systems and Informatics, pp. 234-247, 2022.
A. A. Ardakani, A. R. Kanafi, U. R. Acharya, N. Khadem, and A. Mohammadi, "Application of deep learning technique to manage covid-19 in routine clinical practice using ct images: results of 10 convolutional neural networks," Computers in biology and medicine, vol. 121, p. 103795, 2020.
H. S. Maghdid, A. T. Asaad, K. Z. Ghafoor, A. S. Sadiq, and M. K. Khan, "Diagnosing covid-19 pneumonia from x-ray and ct images using deep learning and transfer learning algorithms," arXiv preprint arXiv:2004.00038, 2020.
T. Yan, P.K. Wong, H. Ren, H. Wang, J. Wang, and Y. Li, "Automatic distinction between covid-19 and common pneumonia using multi-scale convolutional neural network on chest ct scans," Chaos, Solitons Fractals, vol. 140, p. 110153, 2020.
V. Perumal, V. Narayanan, and S. J. S. Rajasekar, "Detection of covid-19 using cxr and ct images using transfer learning and haralick features," Applied Intelligence, vol. 51, pp. 341-358, 2020.
A. Guarnera, E. Santini, and P. Podda, "Covid-19 pneumonia and lung cancer: A challenge for the radiologist review of the main radiological features, differential diagnosis and overlapping pathologies," Tomography, vol. 8, no. 1, pp. 513–528, 2022.
R. Li, C. Xiao, Y. Huang, H. Hassan, and B. Huang, "Deep learning applications in computed tomography images for pulmonary nodule detection and diagnosis: A review," Diagnostics, vol. 12, no. 2, p. 298, 2022.
H. Farhat, G. E. Sakr, and R. Kilany, "Deep learning applications in pulmonary medical imaging: recent updates and insights on covid-19," Machine vision and applications, vol. 31, pp. 1–42, 2020.
S. Bharati, P. Podder, M. Mondal, and V. B. Prasath, "Medical imaging with deep learning for covid-19 diagnosis: a comprehensive review," arXiv preprint arXiv:2107.09602, 2021.
B. Liu, W. Chi, X. Li, P. Li, W. Liang, H. Liu, and J. He, "Evolving the pulmonary nodules diagnosis from classical approaches to deep learning-aided decision support: three decades’ development course and future prospect," Journal of cancer research and clinical oncology, vol. 146, pp. 153–185, 2020.
F. Bushra et al., "Deep learning in computed tomography pulmonary angiography imaging: A dual-pronged approach for pulmonary embolism detection," Expert Systems with Applications, vol. 245, p. 123029, 2024.
M. A. Talukder, M. A. Layek, M. Kazi, M. A. Uddin, and S. Aryal, "Empowering covid-19 detection: Optimizing performance through fine-tuned efficientnet deep learning architecture," Computers in Biology and Medicine, vol. 168, p. 107789, 2024.
M. V. Sanida, T. Sanida, A. Sideris, and M. Dasygenis, "An advanced deep learning framework for multi-class diagnosis from chest x-ray images," J, vol. 7, no. 1, pp. 48–71, 2024.
S. Umirzakova, S. Ahmad, L. U. Khan, and T. Whangbo, "Medical image super resolution for smart healthcare applications: A comprehensive survey," Information Fusion, p. 102075, 2023.
S. Atasever, N. Azginoglu, D. S. Terzi, and R. Terzi, "A comprehensive survey of deep learning research on medical image analysis with focus on transfer learning," Clinical Imaging, vol. 94, pp. 18–41, 2023.
T. Mahmood, A. Rehman, T. Saba, L. Nadeem, and S. A. O. Bahaj, "Recent Advancements and Future Prospects in Active Deep Learning for Medical Image Segmentation and Classification," in IEEE Access, vol. 11, pp. 113623-113652, 2023, doi: 10.1109/ACCESS.2023.3313977.
M. Lubbad, D. Karaboga, A. Basturk, B. Akay, U. Nalbantoglu, and I. Pacal, "Machine learning applications in detection and diagnosis of urology cancers: a systematic literature review," Neural Computing and Applications, pp. 1–25, 2024.
N. Naik et al., "Role of deep learning in prostate cancer management: past, present and future based on a comprehensive literature Review," Journal of Clinical Medicine, vol. 11, no. 13, p. 3575, 2022.
H. E. Sarog˘lu et al., "Machine learning, iot and 5g technologies for breast cancer studies: A review," Alexandria Engineering Journal, vol. 89, pp. 210– 223, 2024.
D. Tian et al., "A review of traditional chinese medicine diagnosis using machine learning: Inspection, auscultation-olfaction, inquiry, and palpation," Computers in Biology and Medicine, p. 108074, 2024.
S. Wassan et al, "Deep convolutional neural network and iot technology for healthcare," Digital Health, vol. 10, 2024.
K. Sarkar, A. Shiuly, and K. G. Dhal, "Revolutionizing concrete analysis: An in-depth survey of ai-powered insights with image-centric approaches on comprehensive quality control, advanced crack detection and concrete property exploration," Construction and Building Materials, vol. 411, p. 134212, 2024.
S. K. Zhou et al., "A review of deep learning in medical imaging: Imaging traits, technology trends, case studies with progress highlights, and future promises," Proceedings of the IEEE, vol. 109, no. 5, pp. 820– 838, 2021.
X. Chen, "Recent advances and clinical applications of deep learning in medical image analysis," Medical Image Analysis, vol. 79, p. 102444, 2022.
A. S. Panayides et al., "Ai in medical imaging informatics: current challenges and future directions," IEEE journal of biomedical and health informatics, vol. 24, no. 7, pp. 1837–1857, 2020.
D. Karimi, H. Dou, S. K. Warfield, and A. Gholipour, "Deep learning with noisy labels: Exploring techniques and remedies in medical image analysis," Medical Image Analysis, vol. 65, p. 101759, 2020.
A. Singh, S. Sengupta, and V. Lakshminarayanan, "Explainable deep learning models in medical image analysis," Journal of imaging, vol. 6, no. 6, p. 52, 2020.
N. Tajbakhsh, L. Jeyaseelan, Q. Li, J. N. Chiang, Z. Wu, and X. Ding, "Embracing imperfect datasets: A review of deep learning solutions for medical image segmentation," Medical Image Analysis, vol. 63, p. 101693, 2020.
M. B. Sushma and S. Prusty, "The effect of covid-19 on public trans- portation sectors and conceptualizing the shifting paradigm: A report on indian scenario," in Advances in Sustainable Materials and Resilient Infrastructure, pp. 21–40, 2022.
A. Narin, C. Kaya, and Z. Pamuk, "Automatic detection of coronavirus disease (covid-19) using x-ray images and deep convolutional neural networks," arXiv preprint arXiv:2003.10849, 2020.
M. E. Chowdhury et al., "Can ai help in screening viral and covid-19 pneumonia?," arXiv preprint arXiv:2003.13145, 2020.
X. Xu et al., "A deep learning system to screen novel coronavirus disease 2019 pneumonia," Engineering, vol. 6, no. 10, pp. 1122–1129, 2020.
R. M. Pereira, D. Bertolini, L. O. Teixeira, C. N. Silla Jr., and Y. M. Costa, "Covid-19 identification in chest x-ray images on flat and hierarchical classification scenarios," Computational Methods and Programs in Biomedicine, p. 105532, 2020.
J. Zhang, Y. Xie, Y. Li, C. Shen, and Y. Xia, "Covid-19 screening on chest x-ray images using deep learning based anomaly detection," arXiv preprint arXiv:2003.12338, 2020.
DOI: https://doi.org/10.18196/jrc.v5i2.21422
Refbacks
Copyright (c) 2024 Omar Nadhim Mohammed
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.
Journal of Robotics and Control (JRC)
P-ISSN: 2715-5056 || E-ISSN: 2715-5072
Organized by Peneliti Teknologi Teknik Indonesia
Published by Universitas Muhammadiyah Yogyakarta in collaboration with Peneliti Teknologi Teknik Indonesia, Indonesia and the Department of Electrical Engineering
Website: http://journal.umy.ac.id/index.php/jrc
Email: jrcofumy@gmail.com