Automated detection of small bowel lesions based on capsule endoscopy using deep learning algorithm

Abstract

BackgroundIn order to overcome the challenges of lesion detection in capsule endoscopy (CE), we improved the YOLOv5-based deep learning algorithm and established the CE-YOLOv5 algorithm to identify small bowel lesions captured by CE. MethodsA total of 124,678 typical abnormal images from 1,452 patients were enrolled to train the CE-YOLOv5 model. Then 298 patients with suspected small bowel lesions detected by CE were prospectively enrolled in the testing phase of the study. Small bowel images and videos from the above 298 patients were interpreted by the experts, non-experts and CE-YOLOv5, respectively. ResultsThe sensitivity of CE-YOLOv5 in diagnosing vascular lesions, ulcerated/erosive lesions, protruding lesions, parasite, diverticulum, active bleeding and villous lesions based on CE videos was 91.9 %, 92.2 %, 91.4 %, 93.1 %, 93.3 %, 95.1 %, and 100 % respectively. Furthermore, CE-YOLOv5 achieved specificity and accuracy of more than 90 % for all lesions. Compared with experts, the CE-YOLOv5 showed comparable overall sensitivity, specificity and accuracy (all P > 0.05). Compared with non-experts, the CE-YOLOv5 showed significantly higher overall sensitivity (P < 0.0001) and overall accuracy (P < 0.0001), and a moderately higher overall specificity (P = 0.0351). Furthermore, the time for AI-reading (5.62 ± 2.81 min) was significantly shorter than that for the other two groups (both P < 0.0001). ConclusionsCE-YOLOv5 diagnosed small bowel lesions in CE videos with high sensitivity, specificity and accuracy, providing a reliable approach for automated lesion detection in real-world clinical practice.