跳转至

SC MIL

SC-MIL: Supervised Contrastive Multiple Instance Learning for Imbalanced Classification in Pathology

data: TCGA-NSCLC、TCGA-RCC,及域外来源的同类亚型数据

WACV 2024 · 论文

引言

将监督式对比学习应用于包表示,并逐步从表征学习转向分类器学习,处理数据集类别不平衡及阳性切片中病灶稀少的问题。

方法

(1)包级对比:在切片嵌入之间应用监督对比损失,避免给所有 patch 赋同一标签。

(2)分类学习:以交叉熵学习最终决策边界。

(3)平滑过渡:训练中逐步调整两类目标的侧重,处理标签不平衡。

SC-MIL:方法相关图示

原文Figure 2,PDF 第 3 页。

实验

实验设置

按病例隔离划分 60:15:25;只调整训练集不平衡比例,固定验证/测试集。

主要结果

核心证据是在多种不平衡程度和域外数据上比较稳定收益,而不是只在平衡训练集比较平均 AUC。

SC-MIL:实验对照表

原文Table 3,PDF 第 4 页。

消融与讨论

比较对比损失与交叉熵的联合/顺序训练,并考察训练权重调度;必须防止患者跨划分泄漏。

参考文献

SC-MIL: Supervised Contrastive Multiple Instance Learning for Imbalanced Classification in Pathology

WACV 2024 · 论文