Automated pneumonia detection on chest X-ray images: A deep learning approach with different optimizers and transfer learning architectures

被引:55
作者
Manickam, Adhiyaman [1 ,2 ]
Jiang, Jianmin [1 ,2 ]
Zhou, Yu [1 ,2 ]
Sagar, Abhinav [3 ]
Soundrapandiyan, Rajkumar [4 ]
Samuel, R. Dinesh Jackson [5 ]
机构
[1] Shenzhen Univ, Coll Comp Sci & Software Engn, Res Inst Future Media Comp, Shenzhen, Peoples R China
[2] Shenzhen Univ, Guangdong Lab Artificial Intelligence & Digital E, Shenzhen, Peoples R China
[3] Vellore Inst Technol, Sch Mech Engn, Vellore, Tamil Nadu, India
[4] Vellore Inst Technol, Sch Comp Sci & Engn, Vellore, Tamil Nadu, India
[5] Oxford Brookes Univ, Visual Artificial Intelligence Lab, Fac Technol Design & Environm, Oxford, England
关键词
Pneumonia Detection; Deep Learning; Transfer Learning; ResNet; 50; Inception V3; Adam Optimizer; Stochastic Gradient Descent Optimizer; U-Net; Convolutional Neural Networks; Accuracy; LUNG-DISEASES; CLASSIFICATION; RADIOGRAPHS;
D O I
10.1016/j.measurement.2021.109953
中图分类号
T [工业技术];
学科分类号
08 ;
摘要
Pneumonia is a disease that leads to the death of individuals within a short period since the flow of fluid in the lungs. Hence, initial diagnosis and drugs are very important to avoid the progress of the disease. This paper proposes a novel deep learning approach for automatic detection of pneumonia using deep transfer learning to simplify the detection process with improved accuracy. This work was aimed to preprocess the input chest X-ray images to identify the presence of pneumonia using U-Net architecture based segmentation and classifies the pneumonia as normal and abnormal (Bacteria, viral) using pre-trained on ImageNet dataset models such as ResNet50, InceptionV3, InceptionResNetV2. Besides, to extract the efficient features and improve accuracy of pre-trained models two optimizers, namely, Adam and Stochastic Gradient Descent (SGD) used and its performances are analyzed with batch sizes of 16 and 32. Based on the values obtained, the performances of undertaken pre-trained models are analyzed and compared with other Convolutional Neural Network (CNN) models such as DenseNet-169+SVM, VGG16, RetinaNet + Mask RCNN, VGG16 and Xception, Fully connected RCNN, etc using various measures. From the results observed that the proposed ResNet50 model work achieved 93.06% accuracy, 88.97 % precision rate, 96.78% Recall rate and 92.71% F1-score rate, which than is higher than the other models aforementioned.
引用
收藏
页数:15
相关论文
共 44 条
[1]  
Acharya Anuja Kumar, 2020, Biomedical & Pharmacology Journal, V13, P449, DOI 10.13005/bpj/1905
[2]  
Ahmed T.S., 2020, SYMMETRY, V1146, P1
[3]  
Albawi S, 2017, I C ENG TECHNOL
[4]  
[Anonymous], 2018, ARXIV180407839
[5]   Lung Pattern Classification for Interstitial Lung Diseases Using a Deep Convolutional Neural Network [J].
Anthimopoulos, Marios ;
Christodoulidis, Stergios ;
Ebner, Lukas ;
Christe, Andreas ;
Mougiakakou, Stavroula .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2016, 35 (05) :1207-1216
[6]   Diagnosis of Pneumonia from Chest X-Ray Images using Deep Learning [J].
Ayan, Enes ;
Unver, Halil Murat .
2019 SCIENTIFIC MEETING ON ELECTRICAL-ELECTRONICS & BIOMEDICAL ENGINEERING AND COMPUTER SCIENCE (EBBT), 2019,
[7]  
Bailer C., ARXIV180503096
[8]   Deep-learning framework to detect lung abnormality - A study with chest X-Ray and lung CT scan images [J].
Bhandary, Abhir ;
Prabhu, G. Ananth ;
Rajinikanth, V ;
Thanaraj, K. Palani ;
Satapathy, Suresh Chandra ;
Robbins, David E. ;
Shasky, Charles ;
Zhang, Yu-Dong ;
Tavares, Joao Manuel R. S. ;
Raja, N. Sri Madhava .
PATTERN RECOGNITION LETTERS, 2020, 129 :271-278
[9]  
Cherian T, 2005, B WORLD HEALTH ORGAN, V83, P353
[10]   Training and Validating a Deep Convolutional Neural Network for Computer-Aided Detection and Classification of Abnormalities on Frontal Chest Radiographs [J].
Cicero, Mark ;
Bilbily, Alexander ;
Dowdell, Tim ;
Gray, Bruce ;
Perampaladas, Kuhan ;
Barfett, Joseph .
INVESTIGATIVE RADIOLOGY, 2017, 52 (05) :281-287