arXiv:2502.21130cs.CV2025-02CVPR被引 9

通过分层蒸馏机制,快速准确分类超分辨率病理图像。

Fast and Accurate Gigapixel Pathological Image Classification with Hierarchical Distillation Multi-Instance Learning

  • 用高低分辨率双网络协同筛选无关图像块,降低计算量
  • 在Camelyon16上提升AUC 3.13%,推理时间减少28.6%
  • 适合需要高效病理诊断的临床与科研场景

尽管多实例学习(MIL)在病理图像分类中表现优异,但处理超大规模全切片图像(WSIs)时面临高推理成本问题。为此,我们提出HDMIL框架,通过消除无关图像块实现快速且准确的分类。该框架包含动态多实例网络(DMIN)和轻量级实例预筛选网络(LIPN):DMIN在高分辨率WSI上运行并生成基于注意力得分的掩码以标识无关块;这些掩码用于指导LIPN在对应低分辨率图像上训练,预测每一块的有用性。测试时,LIPN先识别低分辨率图像中的有效区域,间接实现对高分辨率图像无效区域的剔除,从而显著降低推理时间而不损失性能。此外,我们首次设计基于切比雪夫多项式的柯尔莫哥洛夫-阿诺德分类器,在计算病理学中引入可学习激活层以增强模型表现。在三个公开数据集上的实验表明,HDMIL优于现有最优方法,例如在Camelyon16数据集上,AUC提升3.13%,推理时间减少28.6%。

原文摘要 · Abstract (English)

Although multi-instance learning (MIL) has succeeded in pathological image classification, it faces the challenge of high inference costs due to processing numerous patches from gigapixel whole slide images (WSIs). To address this, we propose HDMIL, a hierarchical distillation multi-instance learning framework that achieves fast and accurate classification by eliminating irrelevant patches. HDMIL consists of two key components: the dynamic multi-instance network (DMIN) and the lightweight instance pre-screening network (LIPN). DMIN operates on high-resolution WSIs, while LIPN operates on the corresponding low-resolution counterparts. During training, DMIN are trained for WSI classification while generating attention-score-based masks that indicate irrelevant patches. These masks then guide the training of LIPN to predict the relevance of each low-resolution patch. During testing, LIPN first determines the useful regions within low-resolution WSIs, which indirectly enables us to eliminate irrelevant regions in high-resolution WSIs, thereby reducing inference time without causing performance degradation. In addition, we further design the first Chebyshev-polynomials-based Kolmogorov-Arnold classifier in computational pathology, which enhances the performance of HDMIL through learnable activation layers. Extensive experiments on three public datasets demonstrate that HDMIL outperforms previous state-of-the-art methods, e.g., achieving improvements of 3.13% in AUC while reducing inference time by 28.6% on the Camelyon16 dataset.

病理图像多实例学习高效推理图像分类

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