Abstract:Deep learning-based computer-aided diagnosis techniques have demonstrated encouraging performance in endoscopic lesion identification and detection, and have reduced the rate of missed and false detections of disease during endoscopy. However, the interpretability of the model-based results has not been adequately addressed by existing methods. This phenomenon is directly manifested by a significant bias in the representation of feature localization. Good recognition models experience severe feature localization errors, particularly for lesions with subtle morphological features, and such unsatisfactory performance hinders the clinical deployment of models. To effectively alleviate this problem, we proposed a solution to optimize the localization bias in feature representations of cancer-related recognition models that is difficult to accurately label and identify in clinical practice. Optimization was performed in the training phase of the model through the proposed data augmentation method and auxiliary loss function based on clinical priors. The data augmentation method, called partial jigsaw, can “break” the spatial structure of lesion-independent image blocks and enrich the data feature space to decouple the interference of background features on the space and focus on fine-grained lesion features. The annotation-based auxiliary loss function used class activation maps for sample distribution correction and led the model to present localization representation converging on the gold standard annotation of visualization maps. The results show that with the improvement of our method, the precision of model recognition reached an average of 92.79%, an F1-score of 92.61%, and accuracy of 95.56% based on a dataset constructed from 23 hospitals. In addition, we quantified the evaluation representation of visualization feature maps. The improved model yielded significant offset correction results for visualized feature maps compared with the baseline model. The average visualization-weighted positive coverage improved from 51.85% to 83.76%. The proposed approach did not change the deployment capability and inference speed of the original model and can be incorporated into any state-of-the-art neural network. It also shows the potential to provide more accurate localization inference results and assist in clinical examinations during endoscopies.

Multi-scale Contrastive Learning for Gastroenteroscopy Classification.

Combining Convolutional and Recurrent Neural Networks for Classification of Focal Liver Lesions in Multi-phase CT Images.

Real-Time Multi-Label Upper Gastrointestinal Anatomy Recognition from Gastroscope Videos

CasCRNN-GL-Net: Cascaded Convolutional and Recurrent Neural Networks with Global and Local Pathways for Classification of Focal Liver Lesions in Multi-Phase CT Images

Residual Convolutional Neural Networks With Global And Local Pathways For Classification Of Focal Liver Lesions

Supervised Contrastive Learning for Gastrointestinal Lesions Classification in Endoscopic Images

Multi-label Recognition of Cancer-Related Lesions with Clinical Priors on White-Light Endoscopy

Large-scale Gastric Cancer Screening and Localization Using Multi-task Deep Neural Network

Deep Learning-Based Computer-Aided Diagnosis System for Gastroscopy Image Classification Using Synthetic Data

Multi-level Asymmetric Contrastive Learning for Volumetric Medical Image Segmentation Pre-training

A Hierarchical Multi-Task Approach to Gastrointestinal Image Analysis

Multi-Scale Context-Guided Deep Network for Automated Lesion Segmentation With Endoscopy Images of Gastrointestinal Tract

Improving the Classification Performance of Esophageal Disease on Small Dataset by Semi-supervised Efficient Contrastive Learning

Multi-scale Multi-instance Multi-feature Joint Learning Broad Network (M3JLBN) for Gastric Intestinal Metaplasia Subtype Classification

DCDS-Net: Deep Transfer Network Based on Depth-Wise Separable Convolution with Residual Connection for Diagnosing Gastrointestinal Diseases

Gastrointestinal Stromal Tumors Diagnosis on Multi-Center Endoscopic Ultrasound Images Using Multi-Scale Image Normalization and Transfer Learning

Convolutional‐capsule network for gastrointestinal endoscopy image classification

Hierarchical Deep Convolutional Neural Networks for Multi-category Diagnosis of Gastrointestinal Disorders on Histopathological Images

Clustering-Guided Twin Contrastive Learning for Endomicroscopy Image Classification

GCLDNet: Gastric Cancer Lesion Detection Network Combining Level Feature Aggregation and Attention Feature Fusion

Multi-Class Gastroesophageal Reflux Disease Classification System Using Deep Learning Techniques