Multi-classification deep learning models for detection of ulcerative colitis, polyps, and dyed-lifted polyps using wireless capsule endoscopy images

Abstract

Wireless capsule endoscopy (WCE) enables imaging and diagnostics of the gastrointestinal (GI) tract to be performed without any discomfort. Despite this, several characteristics, including efficacy, tolerance, safety, and performance, make it difficult to apply and modify widely. The use of automated WCE to collect data and perform the analysis is essential for finding anomalies. Medical specialists need a significant amount of time and expertise to examine the data generated by WCE imaging of the patient’s digestive tract. To address these challenges, several computer vision-based solutions have been designed; nevertheless, they do not achieve an acceptable level of accuracy, and more advancements are required. Thus, in this study, we proposed four multi-classification deep learning (DL) models i.e., Vgg-19 + CNN, ResNet152V2, Gated Recurrent Unit (GRU) + ResNet152V2, and ResNet152V2 + Bidirectional GRU (Bi-GRU) and applied it on different publicly available databases for diagnosing ulcerative colitis, polyps, and dyed-lifted polyps using WCE images. To our knowledge, this is the only study that uses a single DL model for the classification of three different GI diseases. We compared the classification performance of the proposed DL classifiers in terms of many parameters such as accuracy, loss, Matthew's correlation coefficient (MCC), recall, precision, negative predictive value (NPV), positive predictive value (PPV), and F1-score. The results revealed that the Vgg-19 + CNN outperforms the three other proposed DL models in classifying GI diseases using WCE images. The Vgg-19 + CNN model achieved an accuracy of 99.45%. The results of four proposed DL classifiers are also compared with recent state-of-the-art classifiers and the proposed Vgg-19 + CNN model has performed better in terms of improved accuracy.