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.

Interpretable Lung Cancer Diagnosis with Nodule Attribute Guidance and Online Model Debugging

A Hybrid CNN Feature Model for Pulmonary Nodule Malignancy Risk Differentiation.

Towards Reliable and Explainable AI Model for Solid Pulmonary Nodule Diagnosis

Faithful learning with sure data for lung nodule diagnosis

Trustworthy learning with (un)sure annotation for lung nodule diagnosis with CT

An Interpretable Deep Hierarchical Semantic Convolutional Neural Network for Lung Nodule Malignancy Classification

A Simple Method to Train the AI Diagnosis Model of Pulmonary Nodules.

An Interpretable Three-Dimensional Artificial Intelligence Model for Computer-Aided Diagnosis of Lung Nodules in Computed Tomography Images

Unsupervised Representation Learning for Pulmonary Nodule Classification

Parse and Recall: Towards Accurate Lung Nodule Malignancy Prediction like Radiologists

Interpretative Computer-aided Lung Cancer Diagnosis: from Radiology Analysis to Malignancy Evaluation

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

An Explainable Classification Method Based on Complex Scaling in Histopathology Images for Lung and Colon Cancer

Detection-Guided Deep Learning-Based Model with Spatial Regularization for Lung Nodule Segmentation

Joint Learning for Pulmonary Nodule Segmentation, Attributes and Malignancy Prediction

Explainable Classification of Benign-Malignant Pulmonary Nodules With Neural Networks and Information Bottleneck

Reducing Annotation Need in Self-Explanatory Models for Lung Nodule Diagnosis

Lung cancer classification and identification framework with automatic nodule segmentation screening using machine learning

Highly accurate model for prediction of lung nodule malignancy with CT scans

Learning efficient, explainable and discriminative representations for pulmonary nodules classification

Lung Nodule Segmentation Combining Sobel Operator and Mask R-CNN