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.

Mitigating Diagnostic Errors in Lung Cancer Classification: A Multi-Eyes Principle to Uncertainty Quantification

Joint Dermatological Lesion Classification and Confidence Modeling with Uncertainty Estimation

Uncertainty-aware image classification on 3D CT lung

Quantifying and leveraging predictive uncertainty for medical image assessment

Quantifying and Leveraging Classification Uncertainty for Chest Radiograph Assessment

Uncertainty Quantification in Computer-Aided Diagnosis: Make Your Model say "I don't know" for Ambiguous Cases

Lung Cancer Diagnosis Using Deep Attention‐based Multiple Instance Learning and Radiomics

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

Lung Cancer Diagnosis Using Deep Attention Based on Multiple Instance Learning and Radiomics

Predictive uncertainty estimation in deep learning for lung carcinoma classification in digital pathology under real dataset shifts

Uncertainty-informed deep learning models enable high-confidence predictions for digital histopathology

A 3D Probabilistic Deep Learning System for Detection and Diagnosis of Lung Cancer Using Low-Dose CT Scans

Towards radiologist-level cancer risk assessment in CT lung screening using deep learning

End-to-end lung cancer screening with three-dimensional deep learning on low-dose chest computed tomography

Uncertainty Quantification in Segmenting Tuberculosis-Consistent Findings in Frontal Chest X-rays

Uncertainty quantification in skin cancer classification using three-way decision-based Bayesian deep learning

Probabilistic Radiomics: Ambiguous Diagnosis with Controllable Shape Analysis

Implementing Trust in Non-Small Cell Lung Cancer Diagnosis with a Conformalized Uncertainty-Aware AI Framework in Whole-Slide Images

Improving Image-Based Precision Medicine with Uncertainty-Aware Causal Models

An integrated method for detecting lung cancer via CT scanning via optimization, deep learning, and IoT data transmission

Deep learning ensemble approach with explainable AI for lung and colon cancer classification using advanced hyperparameter tuning