中文

HydraMix-Net:一种用于细胞检测与分类的深度多任务半监督学习方法

计算机视觉与模式识别 2020-08-12 v1

摘要

半监督技术通过利用无标签数据提升模型性能,消除了对大规模标签集的依赖。本文提出一种医学影像领域的半监督深度多任务分类与定位方法 HydraMix-Net,该领域标注耗时且昂贵。首先,利用模型在增强后的无标签图像集上的预测进行平均生成伪标签。进一步对高熵预测进行锐化以降低熵,然后与有标签集混合用于训练。模型以多任务学习方式训练,采用噪声容忍联合损失进行分类与定位,在给定有限数据时相比简单深度模型取得更优性能。在 DLBCL 数据上,仅给定 100 个有标签样本时,其准确率达 80%,而简单 CNN 为 70%。

关键词

引用

@article{arxiv.2008.04753,
  title  = {HydraMix-Net: A Deep Multi-task Semi-supervised Learning Approach for Cell Detection and Classification},
  author = {R. M. Saad Bashir and Talha Qaiser and Shan E Ahmed Raza and Nasir M. Rajpoot},
  journal= {arXiv preprint arXiv:2008.04753},
  year   = {2020}
}