Artificial intelligence-assisted diagnosis system of benign and malignant gastric ulcer based on deep learning / 中华消化内镜杂志

ABSTRACT

Objective:To construct an artificial intelligence-assisted diagnosis system to detect gastric ulcer lesions and identify benign and malignant gastric ulcers automatically.Methods:A total of 1 885 endoscopy images were collected from November 2016 to April 2019 in the Digestive Endoscopy Center of Renmin Hospital of Wuhan University. Among them, 636 were normal images, 630 were with benign gastric ulcers, and 619 were with malignant gastric ulcers. A total of 1 735 images belonged to training data set and 150 images were used for validation. These images were input into the Res-net50 model based on the fastai framework, the Res-net50 model based on the Keras framework, and the VGG-16 model based on the Keras framework respectively. Three separate binary classification models of normal gastric mucosa and benign ulcers, normal gastric mucosa and malignant ulcers, and benign and malignant ulcers were constructed.Results:The VGG-16 model showed the best ability of classification. The accuracy of the validation set was 98.0%, 98.0% and 85.0%, respectively, for distinguishing normal gastric mucosa from benign ulcers, normal gastric mucosa from malignant ulcers, and benign ulcers from malignant ulcers.Conclusion:The artificial intelligence-assisted diagnosis system obtained in this study shows noteworthy ability of detection of ulcerous lesions, and is expected to be used in clinical to assist doctors to detect ulcer and identify benign and malignant ulcers.