通过辅助光学染色内镜的对齐无关密集蒸馏实现整体白光息肉分类
计算机视觉与模式识别
2025-07-15 v3
摘要
白光成像(WLI)和窄带成像(NBI)是用于息肉分类的两种主要结肠镜检查模态。虽然 NBI 作为光学染色内镜提供了有价值的血管细节,但在资源有限的环境中,WLI 仍然是最常见且通常是唯一可用的模态。然而,基于 WLI 的方法通常表现不佳,限制了其临床适用性。现有方法通过全局特征对齐将知识从 NBI 转移到 WLI,但通常依赖于裁剪的病灶区域,这容易受到检测误差的影响并忽略了上下文和微妙的诊断线索。为了解决这个问题,本文提出了一种新的整体分类框架,该框架利用全图像诊断而无需息肉定位。关键创新在于对齐无关密集蒸馏(ADD)模块,该模块能够在 WLI 和 NBI 图像之间不对齐的情况下实现细粒度的跨域知识蒸馏。无需显式图像对齐,ADD 学习像素级跨域亲和性以建立特征图之间的对应关系,沿着最相关的像素连接引导蒸馏。为了进一步增强蒸馏的可靠性,ADD 结合了类激活映射(CAM)来过滤跨域亲和性,确保蒸馏路径仅连接那些对息肉诊断具有同等贡献的语义一致区域。在公开和内部数据集上的广泛结果表明,我们的方法实现了 SOTA 性能,在 AUC 上分别比其他方法相对高出至少 2.5% 和 16.2%。代码可在以下地址获取:https://github.com/Huster-Hq/ADD。
引用
@article{arxiv.2505.19319,
title = {Holistic White-light Polyp Classification via Alignment-free Dense Distillation of Auxiliary Optical Chromoendoscopy},
author = {Qiang Hu and Qimei Wang and Jia Chen and Xuantao Ji and Mei Liu and Qiang Li and Zhiwei Wang},
journal= {arXiv preprint arXiv:2505.19319},
year = {2025}
}
备注
Early Accepted by MICCAI 2025