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

PromptBERT: Improving BERT Sentence Embeddings with Prompts

Beihang University(北京航空航天大学);Microsoft(微软);Zhongguancun Laboratory(中关村实验室);Institute of Artificial Intelligence, Beihang University(北京航空航天大学人工智能研究院)

Ting Jiang, Jian Jiao, Shaohan Huang, Zihan Zhang, Deqing Wang, Fuzhen Zhuang, Furu Wei, Haizhen Huang, Denvy Deng, Qi Zhang

英文摘要:

We propose PromptBERT, a novel contrastive learning method for learning better sentence representation. We firstly analyze the drawback of current sentence embedding from original BERT and find that it is mainly due to the static token embedding bias and ineffective BERT layers. Then we propose the first prompt-based sentence embeddings method and discuss two prompt representing methods and three prompt searching methods to make BERT achieve better sentence embeddings. Moreover, we propose a novel unsupervised training objective by the technology of template denoising, which substantially shortens the performance gap between the supervised and unsupervised settings. Extensive experiments show the effectiveness of our method. Compared to SimCSE, PromptBert achieves 2.29 and 2.58 points of improvement based on BERT and RoBERTa in the unsupervised setting.

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

↑