Multi-class deep learning architecture for classifying lung diseases from chest X-Ray and CT images

Abstract

Medical imaging is considered a suitable alternative testing method for the detection of lung diseases. Many researchers have been working to develop various detection methods that have aided in the prevention of lung diseases. To better understand the condition of the lung disease infection, chest X-Ray and CT scans are utilized to check the disease’s spread throughout the lungs. This study proposes an automated system for the detection multi lung diseases in X-Ray and CT scans. A customized convolutional neural network (CNN) and two pre-trained deep learning models with a new image enhancement model are proposed for image classification. The proposed lung disease detection comprises two main steps: pre-processing, and deep learning classification. The new image enhancement algorithm is developed in the pre-processing step using k-symbol Lerch transcendent functions model which enhancement images based on image pixel probability. While, in the classification step, the customized CNN architecture and two pre-trained CNN models Alex Net, and VGG16Net are developed. The proposed approach was tested on publicly available image datasets (CT, and X-Ray image dataset), and the results showed classification accuracy, sensitivity, and specificity of 98.60%, 98.40%, and 98.50% for the X-Ray image dataset, respectively, and 98.80%, 98.50%, 98.40% for the CT scans dataset, respectively. Overall, the obtained results highlight the advantages of the image enhancement model as a first step in processing.