POUF: Prompt-oriented unsupervised fine-tuning for large pre-trained models
POUF: Prompt-oriented unsupervised fine-tuning for large pre-trained models
复制标题
DOI:
10.48550/arxiv.2305.00350
复制
发表时间:
2023-04
期刊:
影响因子:
--
通讯作者:
Korawat Tanwisuth;Shujian Zhang;Huangjie Zheng;Pengcheng He;Mingyuan Zhou
中科院分区:
文献类型:
--
作者:
Korawat Tanwisuth;Shujian Zhang;Huangjie Zheng;Pengcheng He;Mingyuan Zhou
Through prompting, large-scale pre-trained models have become more expressive and powerful, gaining significant attention in recent years. Though these big models have zero-shot capabilities, in general, labeled data are still required to adapt them to downstream tasks. To overcome this critical limitation, we propose an unsupervised fine-tuning framework to directly fine-tune the model or prompt on the unlabeled target data. We demonstrate how to apply our method to both language-augmented vision and masked-language models by aligning the discrete distributions extracted from the prompts and target data. To verify our approach's applicability, we conduct extensive experiments on image classification, sentiment analysis, and natural language inference tasks. Across 13 image-related tasks and 15 language-related ones, the proposed approach achieves consistent improvements over the baselines.