arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2203.04560cs.CV

面向少量标记样本的主动自半监督学习

Active Self-Semi-Supervised Learning for Few Labeled Samples

Australian Centre for Robotics, The University of Sydney(悉尼大学澳大利亚机器人中心);Department of Marine Technology, Norwegian University of Science and Technology(挪威科技大学海洋技术系)

Ziting Wen, Oscar Pizarro, Stefan Williams

AI总结:

针对有限标注下深度模型训练难题,提出主动自半监督学习框架AS3L,通过主动学习与标签传播获取准确先验伪标签引导半监督训练,在图像分类任务上性能提升显著且收敛更快。

摘要:

在多样的实际领域中应用深度模型时,使用有限标注数据训练深度模型是一项重大挑战。将半监督学习与自监督模型结合使用,有望提升标签效率。但该方法在降低标签需求方面存在瓶颈。我们观察到,当仅有有限标签可用时,半监督模型会破坏自监督学习中的有价值信息。为解决这一问题,本文提出了一种简单且有效的框架——主动自半监督学习(active self-semi-supervised learning,AS3L)。AS3L利用先验伪标签(prior pseudo-labels,PPL)引导半监督模型训练,这些PPL通过在自监督特征上进行标签传播得到。基于“PPL的准确率不仅受特征质量影响,还受标记样本选择影响”的观察,我们设计了主动学习与标签传播策略以获取准确的PPL。因此,本框架能在标注有限的情况下显著提升模型性能,同时展现出快速收敛的特性。在四个数据集的图像分类任务上,我们的方法平均比基线方法性能高出5.4%,此外,它仅用约1/3的训练时间就能达到与基线方法相同的准确率。

英文摘要:

Training deep models with limited annotations poses a significant challenge when applied to diverse practical domains. Employing semi-supervised learning alongside the self-supervised model offers the potential to enhance label efficiency. However, this approach faces a bottleneck in reducing the need for labels. We observed that the semi-supervised model disrupts valuable information from self-supervised learning when only limited labels are available. To address this issue, this paper proposes a simple yet effective framework, active self-semi-supervised learning (AS3L). AS3L bootstraps semi-supervised models with prior pseudo-labels (PPL). These PPLs are obtained by label propagation over self-supervised features. Based on the observations the accuracy of PPL is not only affected by the quality of features but also by the selection of the labeled samples. We develop active learning and label propagation strategies to obtain accurate PPL. Consequently, our framework can significantly improve the performance of models in the case of limited annotations while demonstrating fast convergence. On the image classification tasks across four datasets, our method outperforms the baseline by an average of 5.4\%. Additionally, it achieves the same accuracy as the baseline method in about 1/3 of the training time.

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

↑