提出新方法检测胃肠道影像中的未知病变,提升AI诊断可靠性。
NCDD: Nearest Centroid Distance Deficit for Out-Of-Distribution Detection in Gastrointestinal Vision
- 基于特征中心距差异设计新型检测评分
- 在两个公开数据集上超越现有最佳方法
- 适合医学影像领域需要高可信度的AI系统
深度学习在胃肠道视觉分析中的应用有望推动诊断、治疗和患者护理的重大进展。然而,这些工具在面对未见过或新出现的疾病模式时,常做出过度自信的预测,严重影响其可靠性。本文将这一可靠性问题建模为分布外(OOD)检测任务,识别出未见的新兴疾病。由于胃肠道图像中正常与异常样本特征重叠明显,现有方法(主要针对自然图像设计)难以适用。我们提出假设:正常样本特征更接近其真实类别的中心,而分布外样本到各中心距离相等。据此提出新颖的最近中心距缺陷(NCDD)评分,在特征空间中实现胃肠道分布外检测。在多个深度网络架构及两个公开基准数据集(Kvasir2 和 Gastrovision)上的评估表明,该方法优于多种先进方法。代码与实现细节已开源。
原文摘要 · Abstract (English)
The integration of deep learning tools in gastrointestinal vision holds the potential for significant advancements in diagnosis, treatment, and overall patient care. A major challenge, however, is these tools' tendency to make overconfident predictions, even when encountering unseen or newly emerging disease patterns, undermining their reliability. We address this critical issue of reliability by framing it as an out-of-distribution (OOD) detection problem, where previously unseen and emerging diseases are identified as OOD examples. However, gastrointestinal images pose a unique challenge due to the overlapping feature representations between in- Distribution (ID) and OOD examples. Existing approaches often overlook this characteristic, as they are primarily developed for natural image datasets, where feature distinctions are more apparent. Despite the overlap, we hypothesize that the features of an in-distribution example will cluster closer to the centroids of their ground truth class, resulting in a shorter distance to the nearest centroid. In contrast, OOD examples maintain an equal distance from all class centroids. Based on this observation, we propose a novel nearest-centroid distance deficit (NCCD) score in the feature space for gastrointestinal OOD detection. Evaluations across multiple deep learning architectures and two publicly available benchmarks, Kvasir2 and Gastrovision, demonstrate the effectiveness of our approach compared to several state-of-the-art methods. The code and implementation details are publicly available at: https://github.com/bhattarailab/NCDD
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。