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.

Cancer classification using a local reconstruction and global preserving based semi-supervised dimensionality reduction method

Category-Level Regularized Unlabeled-to-Labeled Learning for Semi-supervised Prostate Segmentation with Multi-site Unlabeled Data

Improving the Generalization of Unsupervised Feature Learning by Using Data from Different Sources on Gene Expression Data for Cancer Diagnosis

A Self-Training Subspace Clustering Algorithm under Low-Rank Representation for Cancer Classification on Gene Expression Data

An Unsupervised Feature Learning Method for Enhancing the Generalization of Cancer Diagnosis

Semi-Supervised Projective Non-Negative Matrix Factorization for Cancer Classification.

Histopathology Classification and Localization of Colorectal Cancer Using Global Labels by Weakly Supervised Deep Learning.

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

Breast Tissue Image Classification Based on Semi-supervised Locality Discriminant Projection with Kernels.

Enhancing Characteristic Gene Selection and Tumor Classification by the Robust Laplacian Supervised Discriminative Sparse PCA

Comparison among dimensionality reduction techniques based on Random Projection for cancer classification.

A semi-supervised deep learning method based on stacked sparse auto-encoder for cancer prediction using RNA-seq data

Semi-supervised lung adenocarcinoma histopathology image classification based on multi-teacher knowledge distillation

RPCA-Based Tumor Classification Using Gene Expression Data

Optimal Combination of Feature Selection and Classification Via Local Hyperplane Based Learning Strategy

Based on an Improved Pre-Pca + LDA Classifier Design in Tumor Cells

Sparse Representation for Tumor Classification Based on Feature Extraction Using Latent Low-Rank Representation.

A Novel Local Learning Based Approach with Application to Breast Cancer Diagnosis

Unsupervised Local Discrimination for Medical Images

Semi-supervised Enhanced Discriminative Local Constraint Preserving Projection for Dimensionality Reduction of Medical Hyperspectral Images

Prognostic outcome prediction by semi-supervised least squares classification