Open AccessPneumonia detection based on transfer learning and a combination of VGG19 and a CNN Built from scratch
Author(s) -
Dahmane Oussama,
Mustapha Khelifi,
Mohammed Beladgham,
Kamarulzaman Ibrahim
Publication year - 2021
Publication title -
indonesian journal of electrical engineering and computer science
Language(s) - English
Resource type - Journals
SCImago Journal Rank - 0.241
H-Index - 17
eISSN - 2502-4760
pISSN - 2502-4752
DOI - 10.11591/ijeecs.v24.i3.pp1469-1480
Subject(s) - convolutional neural network , artificial intelligence , computer science , transfer of learning , adaptive histogram equalization , pattern recognition (psychology) , deep learning , preprocessor , feature (linguistics) , histogram , histogram equalization , machine learning , image (mathematics) , linguistics , philosophy
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).