Improved classification and localization approach to small bowel capsule endoscopy using convolutional neural network

Yunseob Hwang,Han Hee Lee,Chunghyun Park,Bayu Adhi Tama,Jin Su Kim,Dae Young Cheung,Woo Chul Chung,Young‐Seok Cho,Kang‐Moon Lee,Myung‐Gyu Choi,Seungchul Lee,Bo‐In Lee
DOI: https://doi.org/10.1111/den.13787
2020-08-31
Digestive Endoscopy
Abstract:<section class="article-section__content"><h3 class="article-section__sub-title section1"> Background</h3><p>Although great advances in artificial intelligence for interpreting small bowel capsule endoscopy (SBCE) images have been made in recent years, its practical use is still limited. The aim of this study was to develop a more practical convolutional neural network (CNN) algorithm for the automatic detection of various small bowel lesions.</p></section><section class="article-section__content"><h3 class="article-section__sub-title section1"> Methods</h3><p>A total of 7,556 images were collected for the training dataset from 526 SBCE videos. Abnormal images were classified into two categories: hemorrhagic lesions (red spot/angioectasia/active bleeding) and ulcerative lesions (erosion/ulcer/stricture). A CNN algorithm based on VGGNet was trained in two different ways: the combined model (hemorrhagic and ulcerative lesions trained separately) and the binary model (all abnormal images trained without discrimination). The detected lesions were visualized using a gradient class activation map (Grad‐CAM). The two models were validated using 5,760 independent images taken at two other academic hospitals.</p></section><section class="article-section__content"><h3 class="article-section__sub-title section1"> Results</h3><p>Both the combined and binary models acquired high accuracy for lesion detection, and the difference between the two models was not significant (96.83% vs. 96.62%, p=0.122). However, the combined model showed higher sensitivity (97.61% vs. 95.07%, p&lt;0.001) and higher accuracy for individual lesions from the hemorrhagic and ulcerative categories than the binary model. The combined model also revealed more accurate localization of the culprit area on images evaluated by the Grad‐CAM.</p></section><section class="article-section__content"><h3 class="article-section__sub-title section1"> Conclusions</h3><p>Diagnostic sensitivity and classification of small bowel lesions using a convolutional neural network are improved by the independent training for hemorrhagic and ulcerative lesions. Grad‐CAM is highly effective in localizing the lesions.</p></section>
gastroenterology & hepatology,surgery
What problem does this paper attempt to address?