An Accurate Deep Learning-Based Computer-Aided Diagnosis System for Gastrointestinal Disease Detection Using Wireless Capsule Endoscopy Image Analysis
Source abstract
Abstract
Peptic ulcers and stomach cancer are common conditions that impact the gastrointestinal (GI) system. Wireless capsule endoscopy (WCE) has emerged as a widely used, noninvasive technique for diagnosing these issues, providing valuable insights through the detailed imaging of the GI tract. Therefore, an early and accurate diagnosis of GI diseases is crucial for effective treatment. This paper introduces the Intelligent Learning Rate Controller (ILRC) mechanism that optimizes the training of deep learning (DL) models by adaptively adjusting the learning rate (LR) based on training progress. This helps improve convergence speed and reduce the risk of overfitting. The ILRC was applied to four DL models: EfficientNet-B0, ResNet101v2, InceptionV3, and InceptionResNetV2. These models were further enhanced using transfer learning, freezing layers, fine-tuning techniques, residual learning, and modern regularization methods. The models were evaluated on two datasets, the Kvasir-Capsule and KVASIR v2 datasets, which contain WCE images. The results demonstrated that the models, particularly when using ILRC, outperformed existing state-of-the-art methods in accuracy. On the Kvasir-Capsule dataset, the models achieved accuracies of up to 99.906%, and on the Kvasir-v2 dataset, they achieved up to 98.062%. This combination of techniques offers a robust solution for automating the detection of GI abnormalities in WCE images, significantly enhancing diagnostic efficiency and accuracy in clinical settings.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Gastrointestinal Bleeding Diagnosis and Treatment
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
Department of Information Systems, College of Computer and Information Sciences, Jouf University, Sakakah 72388, Saudi Arabia
Funding
Deanship of Graduate Studies and Scientific Research at Jouf University
References (33)
Show source-provided references
1. Kvasir-Capsule, a video capsule endoscopy dataset10.1038/s41597-021-00920-z ↗
2. WHO (2022, March 25). Gastrointestinal Cancer. Available online: https://www.who.int/news-room/fact-sheets/detail/cancer.
3. Pre-processing technique for wireless capsule endoscopy image enhancement
4. Amiri, Z., Hassanpour, H., and Beghdadi, A. (2019, January 28–31). A computer-aided method to detect bleeding frames in capsule Endoscopy images. Proceedings of the 8th European Workshop on Visual Information Processing, Roma, Italy.10.1109/EUVIP47703.2019.8946168 ↗
5. Bleeding frame and region detection in the Wireless Capsule Endoscopy video10.1109/JBHI.2015.2399502 ↗
6. Magnetically guided capsule endoscopy10.1002/mp.12299 ↗
7. Bai, L., Wang, L., Chen, T., Zhao, Y., and Ren, H. (2022). Transformer-based disease identification for small-scale imbalanced capsule endoscopy dataset. Electronics, 11.10.3390/electronics11172747 ↗
8. Management of gastric cancer in Asia: Resource-stratified guidelines10.1016/S1470-2045(13)70436-4 ↗
9. Accuracy of magnetically controlled capsule endoscopy, compared with conventional gastroscopy, in detection of gastric diseases10.1016/j.cgh.2016.05.013 ↗
10. EndoSLAM dataset and an unsupervised monocular visual odometry and depth estimation approach for endoscopic videos10.1016/j.media.2021.102058 ↗
11. Using single-cell technologies to map the human immune system—Implications for nephrology10.1038/s41581-019-0227-3 ↗
12. A saliency-based unsupervised method for angiectasia detection in endoscopic video frames10.1007/s40846-017-0299-0 ↗
13. An automatic bleeding frame and region detection scheme for wireless capsule endoscopy videos based on interplane intensity variation profile in normalized RGB color space10.1155/2018/9423062 ↗
14. Endoscopic submucosal dissection comparing with surgical resection in patients with early gastric cancer—A single center experience in Taiwan10.1016/j.jfma.2020.08.027 ↗
15. Kumar, V., Jain, S., and Singh, N. (2022). A Hybrid Convolutional Neural Network with Meta Feature Learning for Abnormality Detection in Wireless Capsule Endoscopy Images. arXiv.
16. Digestive tract abnormalities classification using wireless capsule endoscopy data
17. WCE-DCGAN: A data augmentation method based on wireless capsule endoscopy images for gastrointestinal disease detection10.1049/ipr2.12704 ↗
18. Mahmood, V., Fareed, M.M.S., Ahmed, G., Dawood, F., and Zikria, S. (2022). A robust deep model for classification of peptic ulcer and other digestive tract disorders using endoscopic images. Biomedicines, 10.10.3390/biomedicines10092195 ↗
19. Fan, S., Xu, L., Fan, Y., Wei, K., and Li, L. (2018). Computer-aided detection of small intestinal ulcer and erosion in wireless capsule endoscopy images. Phys. Med. Biol., 63.10.1088/1361-6560/aad51c ↗
20. Computer-aided diagnosis system for ulcer detection in wireless capsule endoscopy images10.1049/iet-ipr.2018.6232 ↗
21. Deep transfer learning approaches for bleeding detection in endoscopy images10.1016/j.compmedimag.2020.101852 ↗
22. Gunasekaran, H., Ramalakshmi, K., Swaminathan, D.K., and Mazzara, M. (2023). GIT-Net: An Ensemble Deep Learning-Based GI Tract Classification of Endoscopic Images. Bioengineering, 10.10.3390/bioengineering10070809 ↗
23. Yogapriya, J., Chandran, V., Sumithra, M., Anitha, P., Jenopaul, P., and Dhas, C.S.G. (2021). Gastrointestinal tract disease classification from wireless endoscopy images using pretrained deep-learning model. Comput. Math. Methods Med., 2021.10.1155/2021/5940433 ↗
24. Pogorelov, K., Randel, K.R., Griwodz, C., Eskeland, S.L., de Lange, T., Johansen, D., Spampinato, C., Dang-Nguyen, D.-T., Lux, M., and Schmidt, P.T. (2017, January 20–23). KVASIR: A multi-class image dataset for computer-aided gastrointestinal disease detection. Proceedings of the 8th ACM on Multimedia Systems Conference, New York, NY, USA.10.1145/3083187.3083212 ↗
25. Saponara, S., and Elhanashi, A. (2022). Impact of image resizing on deep learning detectors for training time and model performance. ApplePies, Lecture Notes in Electrical Engineering, Springer.10.1007/978-3-030-95498-7_2 ↗
26. How transferable are features in deep neural networks?
27. Goodfellow, I., Shlens, J., and Szegedy, C. (2015, January 7–9). Explaining and harnessing adversarial examples. Proceedings of the International Conference on Learning Representations (ICLR), San Diego, CA, USA.
28. Transfer-Learning: Bridging the Gap between Real and Simulation Data for Machine Learning in Injection Molding
29. Transfer learning-based efficient spatiotemporal human action recognition framework for long and overlapping action classes10.1007/s11227-021-03957-4 ↗
30. Tan, M., and Le, Q. (2019, January 9–15). EfficientNet: Rethinking model scaling for convolutional neural networks. Proceedings of the 36th International Conference on Machine Learning, Long Beach, CA, USA.
31. Classification of remote sensing images using efficientnet-b3 cnn model with attention10.1109/ACCESS.2021.3051085 ↗
32. Sandler, M., Howard, A., Zhu, M., Zhmoginov, A., and Chen, L.-C. (2018, January 18–23). MobileNetV2: Inverted residuals and linear bottlenecks. Proceedings of the 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition, Salt Lake City, UT, USA.10.1109/CVPR.2018.00474 ↗
33. Enhanced skin condition prediction through machine learning using dynamic training and testing augmentation10.1109/ACCESS.2020.2976045 ↗