Convolutional neural network-based object detection model to identify gastrointestinal stromal tumors in endoscopic ultrasound images

Abstract

Background and Aim We aimed to develop a convolutional neural network (CNN)-based object detection model for the discrimination of gastric subepithelial tumors, such as gastrointestinal stromal tumors (GISTs), and leiomyomas, in endoscopic ultrasound (EUS) images. Methods We used 376 images from 114 patients with histologically confirmed gastric GIST or leiomyoma to train the EUS-CNN. We constructed the EUS-CNN using an EfficientNet CNN model for feature extraction and a weighted bi-directional feature pyramid network for object detection. We assessed the performance of our EUS-CNN by calculating its accuracy, sensitivity, specificity, and area under receiver operating characteristic curve (AUC) using a validation set of 170 images from 54 patients. Four EUS experts and 15 EUS trainees were asked to judge the same validation dataset, and the diagnostic yields were compared between the EUS-CNN and human assessments. Results In the per-image analysis, the sensitivity, specificity, accuracy, and AUC of our EUS-CNN were 95.6%, 82.1%, 91.2%, and 0.9234, respectively. In the per-patient analysis, the sensitivity, specificity, accuracy, and AUC for our object detection model were 100.0%, 85.7%, 96.3%, and 0.9929, respectively. The EUS-CNN outperformed human assessment in terms of accuracy, sensitivity, and negative predictive value. Conclusions We developed the EUS-CNN system, which demonstrated high diagnostic ability for gastric GIST prediction. This EUS-CNN system can be helpful not only for less-experienced endoscopists but also for experienced ones. Additional EUS image accumulation and prospective studies are required alongside validation in a large multicenter trial.