arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2403.10658cs.CVcs.LG

InterLUDE:标记数据与未标记数据之间的交互以增强半监督学习

InterLUDE: Interactions between Labeled and Unlabeled Data to Enhance Semi-Supervised Learning

Tufts University(塔夫茨大学);University of Texas at Arlington(德克萨斯大学阿灵顿分校)

Zhe Huang, Xiaowei Yu, Dajiang Zhu, Michael C. Hughes

AI总结:

InterLUDE通过标记与未标记数据的嵌入融合和一致性正则化损失,增强半监督学习,在STL-10仅40标签时错误率降至3.2%,显著优于先前方法。

摘要:

半监督学习(SSL)旨在通过同时使用标记和未标记数据进行训练来提升任务性能。主流的SSL图像分类方法大多优化一个损失函数,该函数将监督分类目标与仅从未标记数据导出的正则化项相加组合。这种公式忽略了标记与未标记图像之间交互的潜力。在本文中,我们提出InterLUDE,一种增强SSL的新方法,由两部分组成,每部分都受益于标记-未标记交互。第一部分,嵌入融合,在标记和未标记嵌入之间进行插值以改善表示学习。第二部分是一个新的损失函数,基于一致性正则化原则,旨在最小化模型在标记与未标记输入之间预测的差异。在标准闭集SSL基准和一个具有未整理未标记集的医学SSL任务上的实验表明,我们的方法具有明显优势。在仅有40个标签的STL-10数据集上,InterLUDE实现了3.2%的错误率,而之前最佳方法报告为14.9%。

英文摘要:

Semi-supervised learning (SSL) seeks to enhance task performance by training on both labeled and unlabeled data. Mainstream SSL image classification methods mostly optimize a loss that additively combines a supervised classification objective with a regularization term derived solely from unlabeled data. This formulation neglects the potential for interaction between labeled and unlabeled images. In this paper, we introduce InterLUDE, a new approach to enhance SSL made of two parts that each benefit from labeled-unlabeled interaction. The first part, embedding fusion, interpolates between labeled and unlabeled embeddings to improve representation learning. The second part is a new loss, grounded in the principle of consistency regularization, that aims to minimize discrepancies in the model's predictions between labeled versus unlabeled inputs. Experiments on standard closed-set SSL benchmarks and a medical SSL task with an uncurated unlabeled set show clear benefits to our approach. On the STL-10 dataset with only 40 labels, InterLUDE achieves 3.2% error rate, while the best previous method reports 14.9%.

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

↑