A Lightweight Machine Learning Model for High Precision Gastrointestinal Stromal Tumors Identification

一种用于高精度识别胃肠道间质瘤的轻量级机器学习模型

阅读:1

Abstract

Gastrointestinal stromal tumors (GISTs), which usually develop with a significant malignant potential, are a serious challenge in stromal health. With Endoscopic ultrasound (EUS), GISTs can appear similar to other tumors. This study introduces a lightweight convolutional neural network model optimized for the classification of GISTs and leiomyomas using EUS images only. Models are constructed based on a dataset that comprises 13277 augmented grayscale images derived from 703 patients, ensuring a balanced representation between GIST and leiomyoma cases. The optimized model architecture includes seven convolutional units followed by fully connected layers. After being trained and evaluated with a 5-fold cross-validation, the optimized model achieves an average validation accuracy of 96.2%. The model achieved a sensitivity, specificity, positive predictive value, and negative predictive value of 97.7%, 94.7%, 94.6%, and 97.7%, respectively, and significantly outperformed endoscopists' assessments. The study highlights the model's robustness and consistency. Our results suggest that instead of using developed deep models with fine-tuning, lightweight models with their simpler designs may grasp the essence and drop speckle noise. A lightweight model as a hypothesis with fewer model parameters is preferable to a deeper model with 10 times the model parameters according to Occam's razor statement.

特别声明

1、本页面内容包含部分的内容是基于公开信息的合理引用;引用内容仅为补充信息,不代表本站立场。

2、若认为本页面引用内容涉及侵权,请及时与本站联系,我们将第一时间处理。

3、其他媒体/个人如需使用本页面原创内容,需注明“来源:[生知库]”并获得授权;使用引用内容的,需自行联系原作者获得许可。

4、投稿及合作请联系:info@biocloudy.com。