arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2203.01524cs.LGeess.IV
收藏

基于鲁棒损失的半监督学习

Semi-supervised Learning using Robust Loss

University of Southern California(南加州大学)

Wenhui Cui, Haleh Akrami, Anand A. Joshi, Richard M. Leahy

摘要:

在医疗应用中,手动标记的数据量有限,因此半监督学习和自动标记策略可以成为训练深度神经网络的资产。然而,自动生成的标签的质量可能是不均匀的,并且不如手动标签。在本文中,我们提出了一种半监督训练策略,用于利用手动标记的数据和额外的未标记数据。与现有的方法相比,我们应用鲁棒损失的自动标记的数据,自动补偿不均匀的数据质量使用教师-学生的框架。首先,我们使用在标记数据上预先训练的教师模型为未标记数据生成伪标签。这些伪标签是有噪声的,并且将它们与标记数据一起用于训练深度神经网络会严重降低学习的特征表示和网络的泛化能力。在这里,我们通过使用鲁棒的损失函数来减轻这些伪标签的影响。具体来说,我们使用三个强大的损失函数,即β交叉熵,对称交叉熵,广义交叉熵。我们表明,我们提出的策略提高了模型的性能,通过补偿不均匀的质量标签在图像分类以及分割应用。

英文摘要:

The amount of manually labeled data is limited in medical applications, so semi-supervised learning and automatic labeling strategies can be an asset for training deep neural networks. However, the quality of the automatically generated labels can be uneven and inferior to manual labels. In this paper, we suggest a semi-supervised training strategy for leveraging both manually labeled data and extra unlabeled data. In contrast to the existing approaches, we apply robust loss for the automated labeled data to automatically compensate for the uneven data quality using a teacher-student framework. First, we generate pseudo-labels for unlabeled data using a teacher model pre-trained on labeled data. These pseudo-labels are noisy, and using them along with labeled data for training a deep neural network can severely degrade learned feature representations and the generalization of the network. Here we mitigate the effect of these pseudo-labels by using robust loss functions. Specifically, we use three robust loss functions, namely beta cross-entropy, symmetric cross-entropy, and generalized cross-entropy. We show that our proposed strategy improves the model performance by compensating for the uneven quality of labels in image classification as well as segmentation applications.

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

↑