Abstract:Background: Wireless capsule endoscopy (WCE) has significantly advanced the diagnosis of gastrointestinal (GI) diseases by allowing for the non-invasive visualization of the entire small intestine. However, machine learning-based methods for organ classification in WCE often rely on color information, leading to decreased performance when obstacles such as food debris are present. This study proposes a novel model that integrates convolutional neural networks (CNNs) and long short-term memory (LSTM) networks to analyze multiple frames and incorporate temporal information, ensuring that it performs well even when visual information is limited. Methods: We collected data from 126 patients using PillCam™ SB3 (Medtronic, Minneapolis, MN, USA), which comprised 2,395,932 images. Our deep learning model was trained to identify organs (stomach, small intestine, and colon) using data from 44 training and 10 validation cases. We applied calibration using a Gaussian filter to enhance the accuracy of detecting organ boundaries. Additionally, we estimated the transit time of the capsule in the gastric and small intestine regions using a combination of a convolutional neural network (CNN) and a long short-term memory (LSTM) designed to be aware of the sequence information of continuous videos. Finally, we evaluated the model's performance using WCE videos from 72 patients. Results: Our model demonstrated high performance in organ classification, achieving an accuracy, sensitivity, and specificity of over 95% for each organ (stomach, small intestine, and colon), with an overall accuracy and F1-score of 97.1%. The Matthews Correlation Coefficient (MCC) and Geometric Mean (G-mean) were used to evaluate the model's performance on imbalanced datasets, achieving MCC values of 0.93 for the stomach, 0.91 for the small intestine, and 0.94 for the colon, and G-mean values of 0.96 for the stomach, 0.95 for the small intestine, and 0.97 for the colon. Regarding the estimation of gastric and small intestine transit times, the mean time differences between the model predictions and ground truth were 4.3 ± 9.7 min for the stomach and 24.7 ± 33.8 min for the small intestine. Notably, the model's predictions for gastric transit times were within 15 min of the ground truth for 95.8% of the test dataset (69 out of 72 cases). The proposed model shows overall superior performance compared to a model using only CNN. Conclusions: The combination of CNN and LSTM proves to be both accurate and clinically effective for organ classification and transit time estimation in WCE. Our model's ability to integrate temporal information allows it to maintain high performance even in challenging conditions where color information alone is insufficient. Including MCC and G-mean metrics further validates the robustness of our approach in handling imbalanced datasets. These findings suggest that the proposed method can significantly improve the diagnostic accuracy and efficiency of WCE, making it a valuable tool in clinical practice for diagnosing and managing GI diseases.

A Systematic Evaluation and Optimization of Automatic Detection of Ulcers in Wireless Capsule Endoscopy on a Large Dataset Using Deep Convolutional Neural Networks.

Second glance framework (secG): Enhanced ulcer detection with deep learning on a large wireless capsule endoscopy dataset

Computer-aided detection of small intestinal ulcer and erosion in wireless capsule endoscopy images.

Deep Convolutional Neural Network for Ulcer Recognition in Wireless Capsule Endoscopy: Experimental Feasibility and Optimization

A Deep Convolutional Neural Network for Bleeding Detection in Wireless Capsule Endoscopy Images

Automated Detection of Small Bowel Lesions Based on Capsule Endoscopy Using Deep Learning Algorithm

Ulcer detection in wireless capsule endoscopy images

A CNN-Based Disease Detection Framework for Wireless Capsule Endoscopy Videos

Automated detection of ulcers and erosions in capsule endoscopy images using a convolutional neural network

Modified residual attention network for abnormalities segmentation and detection in WCE images

Efficacy of a comprehensive binary classification model using a deep convolutional neural network for wireless capsule endoscopy

Capsule Endoscopy Image Classification with Deep Convolutional Neural Networks

Infinite Curriculum Learning for Efficiently Detecting Gastric Ulcers in WCE Images

Efficiency Meets Accuracy: Benchmarking Object Detection Models for Pathology Detection in Wireless Capsule Endoscopy

Assessment of Crohn’s Disease Lesions in Wireless Capsule Endoscopy Images

Detection of Protruding Lesion in Wireless Capsule Endoscopy Videos of Small Intestine.

Computer-aided Bleeding Detection in WCE Video.

Fusion of Selected Deep CNN and Handcrafted Features for Gastritis Detection from Wireless Capsule Endoscopy Images

Deep Learning-Based Real-Time Organ Localization and Transit Time Estimation in Wireless Capsule Endoscopy

Deep convolutional neural networks with ensemble learning and transfer learning for automated detection of gastrointestinal diseases

Attention Mechanism Guided SE + ResNet-H Model for Gastrointestinal Endoscopy Image Classification