A deep learning-based system to identify originating mural layer of upper gastrointestinal submucosal tumors under EUS.

Abstract

EUS is the most accurate procedure to determine the originating mural layer and subsequently select the treatment of submucosal tumors (SMTs). However, it requires superb technical and cognitive skills. In this study, we propose a system named SMT Master to determine the originating mural layer of SMTs under EUS. We developed 3 models: deep convolutional neural network (DCNN) 1 for lesion segmentation, DCNN2 for mural layer segmentation, and DCNN3 for the originating mural layer classification. A total of 2721 EUS images from 201 patients were used to train the 3 models. We validated our model internally and externally using 283 images from 26 patients and 172 images from 26 patients, respectively. We applied 368 images from 30 patients for the man-machine contest and used 30 video clips to test the originating mural layer classification. In the originating mural layer classification task, DCNN3 achieved a classification accuracy of 84.43% and 80.68% at internal and external validations, respectively. In the video test, the accuracy was 80.00%. DCNN1 achieved Dice coefficients of 0.956 and 0.776 for lesion segmentation at internal and external validations, respectively, whereas DCNN2 achieved Dice coefficients of 0.820 and 0.740 at internal and external validations, respectively. The system achieved 90.00% accuracy in classification, which is comparable with that of EUS experts. Our proposed system has the potential to solve difficulties in determining the originating mural layer of SMTs in EUS procedures, which relieves the EUS learning pressure of physicians.