Exploring the effect of image enhancement techniques on COVID-19 detection using chest X-ray images

• Published in: Computers in biology and medicine Vol. 132; p. 104319
• Main Authors: Rahman, Tawsifur, Khandakar, Amith, Qiblawey, Yazan, Tahir, Anas, Kiranyaz, Serkan, Abul Kashem, Saad Bin, Islam, Mohammad Tariqul, Al Maadeed, Somaya, Zughaier, Susu M., Khan, Muhammad Salman, Chowdhury, Muhammad E.H.
• Format: Journal Article
• Language: English
• Published: United States Elsevier Ltd 01-05-2021; Elsevier Limited
• Subjects: Accuracy; Algorithms; Artificial intelligence; Artificial neural networks; Bronchopulmonary infection; Chest; Chest X-ray images; Classification; Convolutional neural networks; Coronaviruses; COVID-19; Datasets; Disease control; Disease transmission; Equalization; Fuzzy sets; Histograms; Image classification; Image contrast; Image enhancement; Image processing; Image segmentation; Infections; Lung segmentation; Lungs; Machine learning; Masks; Medical imaging; Model accuracy; Network reliability; Neural networks; Noise; Pandemics; Patients; Pneumonia; Viral diseases; Viruses; X ray imagery; X-rays; COVID-19; Lung segmentation; Convolutional neural networks; Chest X-ray images; Image enhancement
• ISSN: 0010-4825; 1879-0534
• DOI: 10.1016/j.compbiomed.2021.104319

Computer-aided diagnosis for the reliable and fast detection of coronavirus disease (COVID-19) has become a necessity to prevent the spread of the virus during the pandemic to ease the burden on the healthcare system. Chest X-ray (CXR) imaging has several advantages over other imaging and detection techniques. Numerous works have been reported on COVID-19 detection from a smaller set of original X-ray images. However, the effect of image enhancement and lung segmentation of a large dataset in COVID-19 detection was not reported in the literature. We have compiled a large X-ray dataset (COVQU) consisting of 18,479 CXR images with 8851 normal, 6012 non-COVID lung infections, and 3616 COVID-19 CXR images and their corresponding ground truth lung masks. To the best of our knowledge, this is the largest public COVID positive database and the lung masks. Five different image enhancement techniques: histogram equalization (HE), contrast limited adaptive histogram equalization (CLAHE), image complement, gamma correction, and balance contrast enhancement technique (BCET) were used to investigate the effect of image enhancement techniques on COVID-19 detection. A novel U-Net model was proposed and compared with the standard U-Net model for lung segmentation. Six different pre-trained Convolutional Neural Networks (CNNs) (ResNet18, ResNet50, ResNet101, InceptionV3, DenseNet201, and ChexNet) and a shallow CNN model were investigated on the plain and segmented lung CXR images. The novel U-Net model showed an accuracy, Intersection over Union (IoU), and Dice coefficient of 98.63%, 94.3%, and 96.94%, respectively for lung segmentation. The gamma correction-based enhancement technique outperforms other techniques in detecting COVID-19 from the plain and the segmented lung CXR images. Classification performance from plain CXR images is slightly better than the segmented lung CXR images; however, the reliability of network performance is significantly improved for the segmented lung images, which was observed using the visualization technique. The accuracy, precision, sensitivity, F1-score, and specificity were 95.11%, 94.55%, 94.56%, 94.53%, and 95.59% respectively for the segmented lung images. The proposed approach with very reliable and comparable performance will boost the fast and robust COVID-19 detection using chest X-ray images.
•The effects of various CXR enhancement techniques were extensively studied on plain and segmented CXR image classification.•This is the largest CXR and lung segmented image dataset comprising COVID-19, normal, and Non-COVID images.•A modified version of the U-Net model is proposed, it outperforms standard U-Net for the lung segmentation of CXR images.•The outcome of this study was verified by image visualization technique to confirm the findings of the deep networks.•Enhancement techniques, transfer learning, and lung segmentation resulted in superior results of COVID-19 detection.