提出新方法,让模型在未知脑瘤类型下仍能准确分类。
Hierarchical Generalized Category Discovery for Brain Tumor Classification in Digital Pathology
- 结合层次聚类与对比学习,识别已知和未知脑瘤类别。
- 在病理图像上比顶尖方法提升28%准确率,尤其擅长发现新类别。
- 适用于不同成像方式,适合医学影像研发者参考。
精准的脑肿瘤分类对神经肿瘤手术中的术中决策至关重要。然而,现有方法仅限于预定义类别,无法识别训练阶段未见的肿瘤模式。无监督学习可提取通用特征,但缺乏利用标注数据先验知识的能力;半监督方法通常假设所有潜在类别均出现在标注数据中。广义类别发现(GCD)旨在填补这一空白,对未标记数据中的已知与未知类别进行分类。为反映脑肿瘤分类体系的层次结构,本文提出层次化广义类别发现脑肿瘤分类方法(HGCD-BT),将层次聚类与对比学习相结合,并引入一种新型半监督层次聚类损失。在模拟拉曼组织学脑肿瘤图像数据集OpenSRH上,HGCD-BT在切片级分类任务中相较当前最优的GCD方法提升28%准确率,尤其在识别未见肿瘤类别方面表现突出。此外,我们在数字脑肿瘤图谱(Digital Brain Tumor Atlas)的苏木精-伊红染色全切片图像上验证了HGCD-BT的泛化能力,证明其在多种成像模态下的适用性。
原文摘要 · Abstract (English)
Accurate brain tumor classification is critical for intra-operative decision making in neuro-oncological surgery. However, existing approaches are restricted to a fixed set of predefined classes and are therefore unable to capture patterns of tumor types not available during training. Unsupervised learning can extract general-purpose features, but it lacks the ability to incorporate prior knowledge from labelled data, and semi-supervised methods often assume that all potential classes are represented in the labelled data. Generalized Category Discovery (GCD) aims to bridge this gap by categorizing both known and unknown classes within unlabelled data. To reflect the hierarchical structure of brain tumor taxonomies, in this work, we introduce Hierarchical Generalized Category Discovery for Brain Tumor Classification (HGCD-BT), a novel approach that integrates hierarchical clustering with contrastive learning. Our method extends contrastive learning based GCD by incorporating a novel semi-supervised hierarchical clustering loss. We evaluate HGCD-BT on OpenSRH, a dataset of stimulated Raman histology brain tumor images, achieving a +28% improvement in accuracy over state-of-the-art GCD methods for patch-level classification, particularly in identifying previously unseen tumor categories. Furthermore, we demonstrate the generalizability of HGCD-BT on slide-level classification of hematoxylin and eosin stained whole-slide images from the Digital Brain Tumor Atlas, confirming its utility across imaging modalities.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。