针对标签数据极少的场景,提出两阶段提示调优框架,显著提升模型性能。
FATE: A Prompt-Tuning-Based Semi-Supervised Learning Framework for Extremely Limited Labeled Data
- 先用无标签数据无监督适应特征分布,再进行分类任务
- 在7个基准上平均性能提升33.74%,优于现有方法
- 适用于视觉与视觉-语言预训练模型,适合标签极稀疏场景
半监督学习(SSL)通过结合有标签和无标签数据取得了显著进展。然而,现有方法忽略了真实世界中标签数据极度稀缺的常见情况,例如数据集中仅有一个标签样本。通用的SSL方法在如此严苛条件下难以从零开始有效训练,而依赖预训练模型的方法又难以在有限标签数据与大量无标签数据间取得平衡。为此,我们提出「首先适应,再分类」(FATE)——一种专为标签极度稀少场景设计的新型SSL框架。核心是两阶段提示调优:首先利用大量无标签样本,以无监督方式将预训练模型适配至下游数据的特征分布;随后应用专为预训练模型设计的SSL方法完成最终分类任务。FATE兼容视觉与视觉-语言预训练模型。大量实验表明,相比最先进方法,FATE在7个基准上平均性能提升33.74%,有效缓解了标签稀缺带来的挑战。代码已公开于https://anonymous.4open.science/r/Semi-supervised-learning-BA72。
原文摘要 · Abstract (English)
Semi-supervised learning (SSL) has achieved significant progress by leveraging both labeled data and unlabeled data. Existing SSL methods overlook a common real-world scenario when labeled data is extremely scarce, potentially as limited as a single labeled sample in the dataset. General SSL approaches struggle to train effectively from scratch under such constraints, while methods utilizing pre-trained models often fail to find an optimal balance between leveraging limited labeled data and abundant unlabeled data. To address this challenge, we propose Firstly Adapt, Then catEgorize (FATE), a novel SSL framework tailored for scenarios with extremely limited labeled data. At its core, the two-stage prompt tuning paradigm FATE exploits unlabeled data to compensate for scarce supervision signals, then transfers to downstream tasks. Concretely, FATE first adapts a pre-trained model to the feature distribution of downstream data using volumes of unlabeled samples in an unsupervised manner. It then applies an SSL method specifically designed for pre-trained models to complete the final classification task. FATE is designed to be compatible with both vision and vision-language pre-trained models. Extensive experiments demonstrate that FATE effectively mitigates challenges arising from the scarcity of labeled samples in SSL, achieving an average performance improvement of 33.74% across seven benchmarks compared to state-of-the-art SSL methods. Code is available at https://anonymous.4open.science/r/Semi-supervised-learning-BA72.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。