Ensemble of deep capsule neural networks: an application to pediatric pneumonia prediction
Tóm tắt
Pneumonia disease accounts for 15% of all deaths in children under the age of five and early detection of the disease significantly improves survival chances. In this work, we introduce a novel deep neural network model for evaluating pediatric pneumonia from chest radio-graph images. The proposed network is an ensemble of multiple candidate networks, each with interleaved convolutional and capsule layers. Individual networks are stitched together with dense layers and trained as a single model to minimize joint loss. The proposed approach is validated through extensive experimentation on the benchmark pneumonia dataset, and the results demonstrate that the model captures higher level abstractions as well as hidden low-level features from the input radio-graphic images. Our comparison studies reveal that the proposed model produces more generic predictions than existing approaches, with an accuracy of 94.84%. The proposed model produces better scores than the existing models and is extremely useful in assisting clinicians in pneumonia diagnosis.
Tài liệu tham khảo
Naralasetti V, Shaik RK, Katepalli G, Bodapati JD (2021) Deep learning models for pneumonia identification and classification based on X-ray images. Traitement Signal 38(3):903
Saraiva AA, Ferreira NMF, de Sousa LL, Costa NJC, Sousa JVM, Santos D, Valente A, Soares S (2019) Classification of images of childhood pneumonia using convolutional neural networks. In: BIOIMAGING, pp. 112–119
Feng Z, Yu Q, Yao S, Luo L, Zhou W, Mao X, Li J, Duan J, Yan Z, Yang M et al (2020) Early prediction of disease progression in Covid-19 pneumonia patients with chest CT and clinical characteristics. Nat Commun 11(1):1–9
Ozturk T, Talo M, Yildirim EA, Baloglu UB, Yildirim O, Acharya UR (2020) Automated detection of Covid-19 cases using deep neural networks with X-ray images. Comput Biol Med 121:103792
Ayan E, Ünver HM (2019) Diagnosis of pneumonia from chest X-ray images using deep learning. In: 2019 Scientific Meeting on Electrical-Electronics & Biomedical Engineering and Computer Science (EBBT). IEEE, pp. 1–5
Irfan M, Iftikhar MA, Yasin S, Draz U, Ali T, Hussain S, Bukhari S, Alwadie AS, Rahman S, Glowacz A et al (2021) Role of hybrid deep neural networks (HDNNS), computed tomography, and chest x-rays for the detection of Covid-19. Int J Environ Res Public Health 18(6):3056
Sharma H, Jain J.S, Bansal P, Gupta S (2020) Feature extraction and classification of chest x-ray images using cnn to detect pneumonia. In: 2020 10th International Conference on Cloud Computing, Data Science & Engineering (Confluence) . IEEE, pp. 227–231
Narin A, Kaya C, Pamuk Z (2021) Automatic detection of coronavirus disease (Covid-19) using X-ray images and deep convolutional neural networks. Pattern Anal Appl 24:1207
Kermany DS, Goldbaum M, Cai W, Valentim CC, Liang H, Baxter SL, McKeown A, Yang G, Wu X, Yan F et al (2018) Identifying medical diagnoses and treatable diseases by image-based deep learning. Cell 172(5):1122–1131
Stephen O, Sain M, Maduh UJ, Jeong D-U (2019) An efficient deep learning approach to pneumonia classification in healthcare. J Healthcare Eng. https://doi.org/10.1155/2019/4180949
Yadav SS, Jadhav SM (2019) Deep convolutional neural network based medical image classification for disease diagnosis. J Big Data 6(1):1–18
Ibrahim AU, Ozsoz M, Serte S, Al-Turjman F, Yakoi PS (2021) Pneumonia classification using deep learning from chest X-ray images during Covid-19. Cogn Comput. https://doi.org/10.1007/s12559-020-09787-5
Rajpurkar P, Irvin J, Zhu K, Yang B, Mehta H, Duan T, Ding D, Bagul A, Langlotz C, Shpanskaya K, et al (2017) Chexnet: radiologist-level pneumonia detection on chest x-rays with deep learning. http://arxiv.org/abs/1711.05225
Altan A, Karasu S (2020) Recognition of Covid-19 disease from X-ray images by hybrid model consisting of 2d curvelet transform, chaotic salp swarm algorithm and deep learning technique. Chaos Solitons Fractals 140:110071
Chowdhury ME, Rahman T, Khandakar A, Mazhar R, Kadir MA, Mahbub ZB, Islam KR, Khan MS, Iqbal A, Al Emadi N et al (2020) Can I help in screening viral and Covid-19 pneumonia? IEEE Access 8:132665–132676
Almalki YE, Qayyum A, Irfan M, Haider N, Glowacz A, Alshehri F.M, Alduraibi SK, Alshamrani K, Alkhalik Basha MA, Alduraibi A, et al (2021) A novel method for covid-19 diagnosis using artificial intelligence in chest X-ray images. In: Healthcare, vol. 9. Multidisciplinary Digital Publishing Institute, p. 522
Jain R, Nagrath P, Kataria G, Kaushik VS, Hemanth DJ (2020) Pneumonia detection in chest X-ray images using convolutional neural networks and transfer learning. Measurement 165:108046
George PM, Barratt SL, Condliffe R, Desai SR, Devaraj A, Forrest I, Gibbons MA, Hart N, Jenkins RG, McAuley DF et al (2020) Respiratory follow-up of patients with Covid-19 pneumonia. Thorax 75(11):1009–1016
Dondeti V, Bodapati JD, Shareef SN, Veeranjaneyulu N (2020) Deep convolution features in non-linear embedding space for fundus image classification. Rev d’Intell Artif 34(3):307–313
Banerjee T, Batta D, Jain A (2020) Deep belief convolutional neural network with artificial image creation by gans based diagnosis of pneumonia in radiological samples of the pectoralis major. Technical report, EasyChair
Afshar P, Heidarian S, Naderkhani F, Oikonomou A, Plataniotis KN, Mohammadi A (2020) Covid-caps: a capsule network-based framework for identification of Covid-19 cases from X-ray images. Pattern Recogn Lett 138:638–643
Sabour S, Frosst N, Hinton GE (2017) Dynamic routing between capsules. http://arxiv.org/abs/1710.09829
Bodapati JD (2022) Stacked convolutional auto-encoder representations with spatial attention for efficient diabetic retinopathy diagnosis. Multimedia Tools Appl. https://doi.org/10.1007/s11042-022-12811-5
Bodapati JD, Rohith V (2022) Chxcapsnet: deep capsule network with transfer learning for evaluating pneumonia in paediatric chest radiographs. Measurement 188:110491
Shorten C, Khoshgoftaar TM (2019) A survey on image data augmentation for deep learning. J Big Data 6(1):1–48
Mandrekar JN (2010) Receiver operating characteristic curve in diagnostic test assessment. J Thoracic Oncol 5(9):1315–1316
Sahlol AT, Yousri D, Ewees AA, Al-Qaness MA, Damasevicius R, Abd Elaziz M (2020) Covid-19 image classification using deep features and fractional-order marine predators algorithm. Sci Rep 10(1):1–15