Differentiation and risk stratification for gastrointestinal stromal tumors with endoscopic images using deep learning
摘要
Deep learning may be helpful to differentiate gastrointestinal stromal tumors and risk stratification using endoscopic images.
MethodsA total of 494 patients with gastrointestinal stromal tumors and 1010 patients with gastric cancer from author’s hospital were trained and validated. Another 99 patients with gastrointestinal stromal tumors and 100 patients with gastric cancer from second hospital were enrolled as external validation. Two deep learning networks, Swin Transformer and ConvNeXt were adapted for the differentiation, tumor size, mitotic index and risk stratification prediction with image level, patient level and combined level.
ResultsIn the internal validation dataset, ConvNeXt achieved area under curve of 0.985 in the differentiation gastrointestinal stromal tumors from gastric cancer. Swin Transformer achieved a best area under curve of 0.927, 0.806 and 0.765 in the prediction of tumor size, mitotic index and risk stratification for gastrointestinal stromal tumors, respectively. In the external validation dataset, ConvNeXt achieved a best area under curve of 0.999 in the differentiation gastrointestinal stromal tumors from gastric cancer. Swin Transformer achieved a best area under curve of 0.872, 0.856 and 0.693 in the prediction of tumor size, mitotic index and risk stratification for gastrointestinal stromal tumors.
ConclusionsEndoscopic images-based deep learning models demonstrated excellent performance in the differentiation gastrointestinal stromal tumors from gastric cancer, and in the prediction the tumor size, mitotic index and risk stratification of gastrointestinal stromal tumors. They are promising in the differentiation and risk stratification to improve the management of gastrointestinal stromal tumors.