arXiv:2507.17323cs.IR2025-07

用三维场景表征实现可解释的结肠息肉在线检索

EndoFinder: Online Lesion Retrieval for Explainable Colorectal Polyp Diagnosis Leveraging Latent Scene Representations

  • 将息肉视为三维场景,融合多视角特征生成隐空间表示
  • 在公开与自采数据集上诊断准确率超越现有方法
  • 适合需要可解释性AI辅助诊断的内镜医生使用

结直肠癌(CRC)仍是癌症相关死亡的主要原因,及时发现和诊断息肉至关重要。尽管深度学习提升了光学辅助诊断能力,但通常依赖大量标注数据,且输出为“黑箱”,缺乏可解释性。本文提出EndoFinder,一种基于多视角场景表征的在线息肉检索框架,实现可解释且可扩展的CRC诊断。首先,我们设计了一种息肉感知图像编码器,结合对比学习与重建任务,利用息肉分割掩码进行自监督训练,无需大规模标注数据即可提取鲁棒特征。其次,将每个息肉视为三维“场景”,引入场景表征变换器,融合多个视角信息生成统一隐空间表示。通过哈希层离散化该表示,实现在历史病例数据库中的实时检索,诊断信息作为可解释参考用于新案例分析。我们在公开及自建息肉数据集上评估了端到端的再识别与病理分类性能。结果表明,EndoFinder在准确性上优于现有方法,并提供透明的检索式洞察,支持临床决策。通过贡献一个新数据集和可扩展的可解释框架,本工作解决了息肉诊断中的关键挑战,为更高效的AI驱动结肠镜流程提供了新方向。源代码见:https://github.com/ku262/EndoFinder-Scene。

原文摘要 · Abstract (English)

Colorectal cancer (CRC) remains a leading cause of cancer-related mortality, underscoring the importance of timely polyp detection and diagnosis. While deep learning models have improved optical-assisted diagnostics, they often demand extensive labeled datasets and yield "black-box" outputs with limited interpretability. In this paper, we propose EndoFinder, an online polyp retrieval framework that leverages multi-view scene representations for explainable and scalable CRC diagnosis. First, we develop a Polyp-aware Image Encoder by combining contrastive learning and a reconstruction task, guided by polyp segmentation masks. This self-supervised approach captures robust features without relying on large-scale annotated data. Next, we treat each polyp as a three-dimensional "scene" and introduce a Scene Representation Transformer, which fuses multiple views of the polyp into a single latent representation. By discretizing this representation through a hashing layer, EndoFinder enables real-time retrieval from a compiled database of historical polyp cases, where diagnostic information serves as interpretable references for new queries. We evaluate EndoFinder on both public and newly collected polyp datasets for re-identification and pathology classification. Results show that EndoFinder outperforms existing methods in accuracy while providing transparent, retrieval-based insights for clinical decision-making. By contributing a novel dataset and a scalable, explainable framework, our work addresses key challenges in polyp diagnosis and offers a promising direction for more efficient AI-driven colonoscopy workflows. The source code is available at https://github.com/ku262/EndoFinder-Scene.

结直肠癌可解释AI内镜诊断图像检索

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。