Pneumonia detection based on transfer learning and a combination of VGG19 and a CNN Built from scratch

Oussama Dahmane, Mustapha Khelifi, Mohammed Beladgham, Ibrahim Kadri

Abstract


In this paper, to categorize and detect pneumonia from a collection of chest X-ray picture samples, we propose a deep learning technique based on object detection, convolutional neural networks, and transfer learning. The proposed model is a combination of the pre-trained model (VGG19) and our designed architecture. The Guangzhou Women and Children's Medical Center in Guangzhou, China provided the chest X-ray dataset used in this study. There are 5,000 samples in the data set, with 1,583 healthy samples and 4,273 pneumonia samples. Preprocessing techniques such as contrast limited adaptive histogram equalization (CLAHE) and brightness preserving bi-histogram equalization was also used (BBHE) to improve accuracy. Due to the imbalance of the data set, we adopted some training techniques to improve the learning process of the samples. This network achieved over 99% accuracy due to the proposed architecture that is based on a combination of two models. The pre-trained VGG19 as feature extractor and our designed convolutional neural network (CNN).

Keywords


BBHE; CLAHE; Convolutional neural network; Deep learning; Object detection; Transfer learning; VGG19;

Full Text:

PDF


DOI: http://doi.org/10.11591/ijeecs.v24.i3.pp1469-1480

Refbacks

  • There are currently no refbacks.


Creative Commons License
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.

Indonesian Journal of Electrical Engineering and Computer Science (IJEECS)
p-ISSN: 2502-4752, e-ISSN: 2502-4760
This journal is published by the Institute of Advanced Engineering and Science (IAES) in collaboration with Intelektual Pustaka Media Utama (IPMU).

shopify stats IJEECS visitor statistics