Deep Learning and X-Ray Imaging Innovations for Pneumonia Infection Diagnosis: Introducing DeepPneuNet

被引:2
作者
Chakraborty, Sanjay [1 ,2 ]
Nag, Tirthajyoti [3 ]
Pandey, Saroj Kumar [4 ]
Ghosh, Jayasree [3 ]
Dey, Lopamudra [5 ]
机构
[1] Linkoping Univ, Dept Comp & Informat Sci IDA, REAL, AIICS, Linkoping, Sweden
[2] Techno Int New Town, Dept Comp Sci & Engn, Kolkata, India
[3] JIS Univ, Dept Comp Sci & Engn, Kolkata, India
[4] GLA Univ, Dept Comp Engn & Applicat, Mathura, India
[5] Linkoping Univ, Dept Biomed & Clin Sci BKV, Linkoping, Sweden
关键词
deep learning; DeepPneuNet; diagnosis prediction; infections; pneumonia; x-ray imaging; TUBERCULOSIS;
D O I
10.1111/coin.70029
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
This paper aims to develop a new deep learning model (DeepPneuNet) and evaluate its performance in predicting Pneumonia infection diagnosis based on patients' chest x-ray images. We have collected 5856 chest x-ray images that are labeled as either "pneumonia" or "normal" from a public forum. Before applying the DeepPneuNet model, a necessary feature extraction and feature mapping have been done on the input images. Conv2D layers with a 1 x 1 kernel size are followed by ReLU activation functions to make up the model. These layers are in charge of recognizing important patterns and features in the images. A MaxPooling 2D procedure is applied to minimize the spatial size of the feature maps after every two Conv2D layers. The sparse categorical cross-entropy loss function trains the model, and the Adam optimizer with a learning rate of 0.001 is used to optimize it. The DeepPneuNet provides 90.12% accuracy for diagnosis of the Pneumonia infection for a set of real-life test images. With 9,445,586 parameters, the DeepPneuNet model exhibits excellent parameter efficiency. DeepPneuNet is a more lightweight and computationally efficient alternative when compared to the other pre-trained models. We have compared accuracies for predicting Pneumonia diagnosis of our proposed DeepPneuNet model with some state-of-the-art deep learning models. The proposed DeepPneuNet model is more advantageous than the existing state-of-the-art learning models for Pneumonia diagnosis with respect to accuracy, precision, recall, F-score, training parameters, and training execution time.
引用
收藏
页数:19
相关论文
共 43 条
[1]  
Agarap A.F., 2018, arXiv, DOI DOI 10.48550/ARXIV.1803.08375
[2]   Pneumonia Detection Using Chest Radiographs With Novel EfficientNetV2L Model [J].
Ali, Mudasir ;
Shahroz, Mobeen ;
Akram, Urooj ;
Mushtaq, Muhammad Faheem ;
Altamiranda, Stefania Carvajal ;
Obregon, Silvia Aparicio ;
Diez, Isabel De La Torre ;
Ashraf, Imran .
IEEE ACCESS, 2024, 12 :34691-34707
[3]   Multi-label classification of Alzheimer's disease stages from resting-state fMRI-based correlation connectivity data and deep learning [J].
Alorf, Abdulaziz ;
Khan, Muhammad Usman Ghani .
COMPUTERS IN BIOLOGY AND MEDICINE, 2022, 151
[4]  
Baldi P., 2013, Adv. Neural Inform. Process. Syst., V26
[5]   Predicting the diagnosis of HIV and sexually transmitted infections among men who have sex with men using machine learning approaches [J].
Bao, Yining ;
Medland, Nicholas A. ;
Fairley, Christopher K. ;
Wu, Jinrong ;
Shang, Xianwen ;
Chow, Eric P. F. ;
Xu, Xianglong ;
Ge, Zongyuan ;
Zhuang, Xun ;
Zhang, Lei .
JOURNAL OF INFECTION, 2021, 82 (01) :48-59
[6]  
Bjorck J, 2018, ADV NEUR IN, V31
[7]  
Chaithanya B., 2021, Indonesian Journal of Electrical Engineering and Computer Science, P1700, DOI DOI 10.11591/IJEECS.V24.I3.PP1700-1710
[8]  
Chen Y., 2020, Sensors, V20, P1585
[9]   Deep Learning for Automatic Pneumonia Detection [J].
Gabruseva, Tatiana ;
Poplavskiy, Dmytro ;
Kalinin, Alexandr .
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS (CVPRW 2020), 2020, :1436-1443
[10]   Recent advances in convolutional neural networks [J].
Gu, Jiuxiang ;
Wang, Zhenhua ;
Kuen, Jason ;
Ma, Lianyang ;
Shahroudy, Amir ;
Shuai, Bing ;
Liu, Ting ;
Wang, Xingxing ;
Wang, Gang ;
Cai, Jianfei ;
Chen, Tsuhan .
PATTERN RECOGNITION, 2018, 77 :354-377