通过双教师框架提升医学图像分类的半监督学习效果
Judge Like a Real Doctor: Dual Teacher Sample Consistency Framework for Semi-supervised Medical Image Classification
- 引入绝对与相对位置一致性,增强模型对样本间关系的感知
- 在多个数据集上实现更高准确率,显著优于现有方法
- 适合医学图像标注成本高的场景,尤其适用于小样本学习
半监督学习(SSL)是缓解医学图像分类中高标注成本的常用方法。作为SSL的主要分支,一致性正则化通过同一样本在不同视图下的预测一致性(即绝对位置一致性,AL-c)来提升性能。然而,仅依赖AL-c可能忽略样本间的相对差异,我们将其称为相对位置。为此,提出样本一致性均值教师(SCMT),不仅保留AL-c,还引入样本与其相关样本间相似性的相对位置一致性(RL-c),从双重视角提取更丰富的语义信息。此外,由于医学图像结构高度相似,特征空间中样本分布过密,导致相对位置易受噪声影响。因此进一步提出样本散度均值教师(SSMT),利用对比学习稀疏化样本分布,获得更鲁棒的相对位置。大量实验表明,该方法在多个数据集上均表现优越。
原文摘要 · Abstract (English)
Semi-supervised learning (SSL) is a popular solution to alleviate the high annotation cost in medical image classification. As a main branch of SSL, consistency regularization engages in imposing consensus between the predictions of a single sample from different views, termed as Absolute Location consistency (AL-c). However, only AL-c may be insufficient. Just like when diagnosing a case in practice, besides the case itself, the doctor usually refers to certain related trustworthy cases to make more reliable decisions.Therefore, we argue that solely relying on AL-c may ignore the relative differences across samples, which we interpret as relative locations, and only exploit limited information from one perspective. To address this issue, we propose a Sample Consistency Mean Teacher (SCMT) which not only incorporates AL c but also additionally enforces consistency between the samples' relative similarities to its related samples, called Relative Location consistency (RL c). AL c and RL c conduct consistency regularization from two different perspectives, jointly extracting more diverse semantic information for classification. On the other hand, due to the highly similar structures in medical images, the sample distribution could be overly dense in feature space, making their relative locations susceptible to noise. To tackle this problem, we further develop a Sample Scatter Mean Teacher (SSMT) by utilizing contrastive learning to sparsify the sample distribution and obtain robust and effective relative locations. Extensive experiments on different datasets demonstrate the superiority of our method.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。