中文

自松弛联合训练:面向有序噪声标签严重度估计的样本选择

计算机视觉与模式识别 2024-10-30 v1

摘要

严重度估计是医学图像诊断中的一项关键任务。然而,为单个图像准确分配严重度类别标签成本高昂且极具挑战性。因此,所附带的标签往往是含噪的。在本文中,我们提出了一种使用“有序”噪声标签进行训练的新框架。由于严重度级别具有有序关系,我们可以利用这一点来训练分类器,同时减轻噪声标签的负面影响。我们的框架使用两种技术:干净样本选择和双网络架构。我们方法的一个技术亮点是使用从噪声硬标签中导出的软标签。通过在两种技术中恰当地使用软标签和硬标签,我们实现了更准确的样本选择和更鲁棒的网络训练。在使用两个内镜溃疡性结肠炎(UC)数据集和一个视网膜糖尿病视网膜病变(DR)数据集的实验中,所提出的方法优于多种最先进的方法。我们的代码可在https://github.com/shumpei-takezaki/Self-Relaxed-Joint-Training获取。

关键词

引用

@article{arxiv.2410.21885,
  title  = {Self-Relaxed Joint Training: Sample Selection for Severity Estimation with Ordinal Noisy Labels},
  author = {Shumpei Takezaki and Kiyohito Tanaka and Seiichi Uchida},
  journal= {arXiv preprint arXiv:2410.21885},
  year   = {2024}
}

备注

Accepted at WACV2025