• Laser & Optoelectronics Progress
  • Vol. 59, Issue 6, 0617028 (2022)
Hongxiao Li1, Shu Li2, Xiafei Shi1, Xiaoxi Dong1, Ge Jin2, Lanping Zhu2, Yingxin Li1, and Huijuan Yin1、*
Author Affiliations
  • 1Institute of Biomedical Engineering, Chinese Academy of Medical Sciences & Peking Union Medical College, Tianjin 300192, China
  • 2Department of Gastroenterology, General Hospital of Tianjin Medical University, Tianjin 300050, China
  • show less
    DOI: 10.3788/LOP202259.0617028 Cite this Article Set citation alerts
    Hongxiao Li, Shu Li, Xiafei Shi, Xiaoxi Dong, Ge Jin, Lanping Zhu, Yingxin Li, Huijuan Yin. BiT-Based Early Gastric Cancer Classification Using Endoscopic Images[J]. Laser & Optoelectronics Progress, 2022, 59(6): 0617028 Copy Citation Text show less

    Abstract

    Gastric cancer is one of the significant lethal cancers in China. Most patients are diagnosed at an advanced stage, and if gastric cancer can be detected at an early stage through large-scale screening, patient survival can be considerably improved. In China, there are two obstacles toward the large-scale screening of early gastric cancer. One is that endoscopy is overly invasive, resulting in low patient acceptance, and the other is that the number of endoscopists is too small compared with China's large population. A capsule endoscopic robot can alleviate the first obstacle, and the second obstacle is expected to be solved using artificial intelligence. We transferred the state-of-the-art Big Transfer (BiT) to a small dataset of early gastric cancer endoscopic images and built an early gastric cancer classification model based on white-light endoscopic images. We customized the BiT hyperparameter rules in transfer learning based on local situations. The batch size was selected according to the GPU memory limit, and based on the batch size, the linear scale rules were used to adjust the optimizer's initial learning rate dynamically. The total number of training images for the small dataset was set at 256000, on which other hyperparameters of the transfer learning were set. This study experimented with multiple models having the same structure of ResNet-v2 but different depths and widths. The best model has a depth of 101 and a width three times the original one. It has an accuracy of 97.14%, an F1 score of 94.77%, a sensitivity of 90.67%, and a specificity of 99.73% on the test set. Furthermore, the results show that the effect of batch size on the model training is statistically insignificant. This paper transferred a large model to a small dataset of endoscopic images with the BiT customization. This will promote the use of large-scale models in the field of endoscopic image analysis, which can help realize a large-scale screening of early gastric cancer.
    Hongxiao Li, Shu Li, Xiafei Shi, Xiaoxi Dong, Ge Jin, Lanping Zhu, Yingxin Li, Huijuan Yin. BiT-Based Early Gastric Cancer Classification Using Endoscopic Images[J]. Laser & Optoelectronics Progress, 2022, 59(6): 0617028
    Download Citation