Multi-label recognition of cancer-related lesions with clinical priors on white-light endoscopy

作者：Tao Yu, Ne Lin, Xingwei Zhong, Xiaoyan Zhang, Xinsen Zhang, Yihe Chen, Jiquan Liu, Weiling Hu, Huilong Duan, Jianmin Si · 发表于：Computers in Biology and Medicine · 年份：2022 · DOI：10.1016/j.compbiomed.2022.105255 · 被引用次数：10 · 研究领域：Colorectal Cancer Screening and Detection、Esophageal Cancer Research and Treatment、AI in cancer detection

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 ...