Journal of Information Technology Management

Journal of Information Technology Management

Implementing a Convolutional Neural Network with Adam Optimizer for Pneumonia Detection Through Chest X-ray Imaging

Document Type : Research Paper

Authors
1 Department of Electrical and Electronics Engineering, College of Engineering, University of Thi-Qar, Thi-Qar, Iraq
2 University of Sumer, Thi-Qar, Iraq.
3 Presidency of the University, Shatrah University, Thi-Qar, Iraq.
4 Presidency of the University, Shatrah University, Thi-Qar, Iraq; Mazaya University College, Thi-Qar, Iraq.
5 Immediate Ambulance Department, Al-Dour Technical Institute, Northern Technical University, Tikrit, Iraq
6 General Directorate of Thi-Qar Education, Ministry of Education, Thi-Qar, Iraq.
Abstract
Pneumonia occurs frequently and can be life-threatening all over the world, mainly in locations where radiologists are hard to find. Even though chest X-ray imaging is widely used, correctly and promptly interpreting it is difficult due to human error and inadequate resources. It aims to overcome the problem of limited, reliable resources for advanced radiology by designing and evaluating a CNN model that can distinguish chest X-rays with and without pneumonia. To increase the usability of the model, the chest X-ray images were processed by being resized to a size of 224×224×3 and then enlarged through random rotation, flipping, and translation. The data was split so that 80 percent went to the training set and 20 percent to the testing set. Several layers were built into the custom CNN for overfitting prevention, such as convolution, pooling, batch normalization, ReLU, and dropout. For 20 epochs, I used Adam as the optimizer and set the mini-batch size to 32 and the learning rate to 1e-4. The model correctly distinguished between pneumonia and normal lung cases over 95% of the time. We found that the model was robust thanks to the confusion matrix and a close study of sample images after prediction. Microwave-based deep learning is introduced in this paper, which demonstrates strong accuracy in detecting pneumonia and makes it possible to use the tool as an automatic device for medical and remote healthcare screening.
Keywords

Alalwan, N., Taloba, A. I., Abozeid, A., Alzahrani, A. I., & Al-Bayatti, A. H. (2023). A hybrid classification and identification of pneumonia using African buffalo optimization and CNN from chest X-ray images. CMES-Computer Modeling in Engineering and Sciences138(3), 2497-2517.
Alsharif, R., Al-Issa, Y., Alqudah, A. M., Qasmieh, I. A., Mustafa, W. A., & Alquran, H. (2021). PneumoniaNet: Automated detection and classification of pediatric pneumonia using chest X-ray images and CNN approach. Electronics10(23), 2949.
Brownlee, J. (2022). Machine learning mastery. Machine Learning Mastery.
Chakrabarty, N., & Biswas, S. (2018, October). A statistical approach to adult census income level prediction. In 2018 International Conference on Advances in Computing, Communication Control and Networking (ICACCCN) (pp. 207-212). IEEE.
Chen, J., Li, Y., Guo, L., Zhou, X., Zhu, Y., He, Q., ... & Feng, Q. (2024). Machine learning techniques for CT imaging diagnosis of novel coronavirus pneumonia: a review. Neural Computing and Applications, 36(1), 181-199.
Goel, S., & Sobti, R. (2024). Deep Cognitive Learning for Enhanced Pneumonia Detection: Employing CNNs. Innovative Computing and Communications: Proceedings of ICICC 2024, Volume 33, 213.
Hasan, M. R., Ullah, S. M. A., & Islam, S. M. R. (2024). Recent advancement of deep learning techniques for pneumonia prediction from chest X-ray image. Medical Reports7, 100106.
Hussin Adam Khatir, A. A., & Bee, M. (2022). Machine learning models and data-balancing techniques for credit scoring: What is the best combination?. Risks10(9), 169.
Joloudari, J. H., Marefat, A., Nematollahi, M. A., Oyelere, S. S., & Hussain, S. (2023). Effective class-imbalance learning based on SMOTE and convolutional neural networks. Applied Sciences, 13(6), 4006.
Kozhay, K., Turarbek, S., Asselbekova, T., Ali, M. H., & Shehab, E. (2024). Convolutional neural network-based defect detection technique in FDM technology. Procedia Computer Science, 231, 119-128.
Lim, W. S. (2021). Pneumonia—overview. Encyclopedia of respiratory medicine, 185.
Lohmander, F. (2021). Breast reconstruction with biological graft: clinical outcomes in the setting of breast cancer treatment. Karolinska Institutet (Sweden).
Mateo, J., Rius-Peris, J. M., Maraña-Pérez, A. I., Valiente-Armero, A., & Torres, A. M. (2021). Extreme gradient boosting machine learning method for predicting medical treatment in patients with acute bronchiolitis. Biocybernetics and Biomedical Engineering41(2), 792-801.
Rahman, T., Chowdhury, M. E., Khandakar, A., Islam, K. R., Islam, K. F., Mahbub, Z. B., ... & Kashem, S. (2020). Transfer learning with deep convolutional neural network (CNN) for pneumonia detection using chest X-ray. Applied Sciences10(9), 3233.
Rehman, A., Khan, A., Fatima, G., Naz, S., & Razzak, I. (2023). Review on chest pathogies detection systems using deep learning techniques: A. Rehman et al. Artificial Intelligence Review56(11), 12607-12653.
Tian, Z., Yi, S., Zhiwen, P., Nan, L., & Xiaohu, Y. (2021). Matching theory based physical layer secure transmission strategy for cognitive radio networks. IEEE Access9, 46201-46209.