arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2301.00524cs.CVcs.HCcs.LG
收藏

在标签噪声存在时学习置信分类器

Learning Confident Classifiers in the Presence of Label Noise

Institute of Informatics, LMU Munich(慕尼黑大学信息学研究所);MBZUAI(穆罕默德·本·扎耶德人工智能大学);Skoltech(斯科尔科沃科学技术研究院)

Asma Ahmed Hashmi, Aigerim Zhumabayeva, Nikita Kotelevskii, Artem Agafonov, Mohammad Yaqub, Maxim Panov, Martin Takáč

AI总结:

针对标签噪声问题,本文提出一种概率模型,通过显式建模噪声、引入信息正则化及调整分割损失函数,在分类和分割任务上均优于现有最优方法,可恢复真实标签并构建置信模型。

摘要:

深度神经网络(DNN)模型的成功很大程度上取决于所提供标注的质量。例如,在医学图像分割中,为每个数据点获取多个专家标注以最小化主观标注偏差是常见做法。此时,估计的目标是滤除标签噪声并恢复未明确给出的真实掩码。本文提出一种针对噪声观测的概率模型,可用于构建置信分类和分割模型。为实现该目标,我们显式建模标签噪声,并引入一种新的基于信息的正则化方法,以促使网络恢复真实标签。此外,针对分割任务,我们通过优先在所有标注者达成一致的高置信区域进行学习来调整损失函数。我们在一系列分类任务上评估所提方法,包括MNIST、CIFAR-10、Fashion-MNIST的噪声版本以及带有真实人类标注噪声的CIFAR-10N数据集。同时,针对分割任务,我们考虑了多个医学成像数据集,如反映真实标注者间差异的LIDC和RIGA数据集。实验表明,我们的算法在所考虑的分类和分割问题上优于现有最优解决方案。

英文摘要:

The success of Deep Neural Network (DNN) models significantly depends on the quality of provided annotations. In medical image segmentation, for example, having multiple expert annotations for each data point is common to minimize subjective annotation bias. Then, the goal of estimation is to filter out the label noise and recover the ground-truth masks, which are not explicitly given. This paper proposes a probabilistic model for noisy observations that allows us to build a confident classification and segmentation models. To accomplish it, we explicitly model label noise and introduce a new information-based regularization that pushes the network to recover the ground-truth labels. In addition, for segmentation task we adjust the loss function by prioritizing learning in high-confidence regions where all the annotators agree on labeling. We evaluate the proposed method on a series of classification tasks such as noisy versions of MNIST, CIFAR-10, Fashion-MNIST datasets as well as CIFAR-10N, which is real-world dataset with noisy human annotations. Additionally, for segmentation task, we consider several medical imaging datasets, such as, LIDC and RIGA that reflect real-world inter-variability among multiple annotators. Our experiments show that our algorithm outperforms state-of-the-art solutions for the considered classification and segmentation problems.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑