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

SequenceMatch:重新审视半监督学习的弱-强增强设计

SequenceMatch: Revisiting the design of weak-strong augmentations for Semi-supervised learning

Sungkyunkwan University(成均馆大学)

Khanh-Binh Nguyen

AI总结:

SequenceMatch通过引入中等强度数据增强和双重一致性约束,有效缓解了半监督学习中的确认偏差问题,在保持高效率的同时显著提升了多个标准及大规模数据集上的分类准确率。

摘要:

半监督学习(SSL)近年来变得流行,因为它允许使用大量未标记数据来训练模型。然而,许多SSL方法面临的一个问题是确认偏差,当模型过拟合于小型带标签训练数据集并产生过度自信的错误预测时,就会发生这种情况。为了解决这个问题,我们提出了SequenceMatch,一种利用多种数据增强的高效SSL方法。SequenceMatch的关键要素是为未标记数据引入了中等强度增强。通过利用不同的增强方式以及每对增强样本之间的一致性约束,SequenceMatch有助于减小模型对弱增强和强增强样本预测分布之间的差异。此外,SequenceMatch针对高置信度和低置信度预测定义了两种不同的一致性约束。因此,SequenceMatch比ReMixMatch更具数据效率,并且比ReMixMatch(4倍)和CoMatch(2倍)更具时间效率,同时具有更高的准确率。尽管简单,SequenceMatch在CIFAR-10/100、SVHN和STL-10等标准基准上始终优于先前的方法。它还在ImageNet等大规模数据集上以38.46%的错误率大幅超越了先前最先进的方法。代码可在https://github.com/beandkay/SequenceMatch获取。

英文摘要:

Semi-supervised learning (SSL) has become popular in recent years because it allows the training of a model using a large amount of unlabeled data. However, one issue that many SSL methods face is the confirmation bias, which occurs when the model is overfitted to the small labeled training dataset and produces overconfident, incorrect predictions. To address this issue, we propose SequenceMatch, an efficient SSL method that utilizes multiple data augmentations. The key element of SequenceMatch is the inclusion of a medium augmentation for unlabeled data. By taking advantage of different augmentations and the consistency constraints between each pair of augmented examples, SequenceMatch helps reduce the divergence between the prediction distribution of the model for weakly and strongly augmented examples. In addition, SequenceMatch defines two different consistency constraints for high and low-confidence predictions. As a result, SequenceMatch is more data-efficient than ReMixMatch, and more time-efficient than both ReMixMatch ($\times4$) and CoMatch ($\times2$) while having higher accuracy. Despite its simplicity, SequenceMatch consistently outperforms prior methods on standard benchmarks, such as CIFAR-10/100, SVHN, and STL-10. It also surpasses prior state-of-the-art methods by a large margin on large-scale datasets such as ImageNet, with a 38.46\% error rate. Code is available at https://github.com/beandkay/SequenceMatch.

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

↑