arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2207.07087cs.CLcs.IRcs.LG
收藏

参数有效的提示微调实现广义和校准的神经文本检索器

Parameter-Efficient Prompt Tuning Makes Generalized and Calibrated Neural Text Retrievers

Tsinghua University(清华大学);Meituan-Dianping Group(美团点评集团)

Weng Lam Tam, Xiao Liu, Kaixuan Ji, Lilong Xue, Xingjian Zhang, Yuxiao Dong, Jiahua Liu, Maodi Hu, Jie Tang

摘要:

提示微调尝试更新预训练模型中的一些特定于任务的参数。它在语言理解和生成任务上都取得了与微调完整参数集相当的性能。在这项工作中,我们研究了神经文本检索器的提示微调问题。我们引入参数高效的提示微调跨域,跨域和跨主题设置的文本检索。通过广泛的分析,我们表明,该策略可以减轻两个问题-参数效率低下和弱的泛化能力-所面临的微调为基础的检索方法。值得注意的是,它可以显着提高域外zero-shot泛化的检索模型。通过仅更新0.1%的模型参数,提示微调策略可以帮助检索模型实现更好的泛化性能比传统的方法,其中所有参数都更新。最后,为了促进检索者跨主题泛化能力的研究,我们策划并发布了一个学术检索数据集,其中包含87个主题的18 K查询结果对,使其成为迄今为止最大的特定主题检索数据集。

英文摘要:

Prompt tuning attempts to update few task-specific parameters in pre-trained models. It has achieved comparable performance to fine-tuning of the full parameter set on both language understanding and generation tasks. In this work, we study the problem of prompt tuning for neural text retrievers. We introduce parameter-efficient prompt tuning for text retrieval across in-domain, cross-domain, and cross-topic settings. Through an extensive analysis, we show that the strategy can mitigate the two issues -- parameter-inefficiency and weak generalizability -- faced by fine-tuning based retrieval methods. Notably, it can significantly improve the out-of-domain zero-shot generalization of the retrieval models. By updating only 0.1% of the model parameters, the prompt tuning strategy can help retrieval models achieve better generalization performance than traditional methods in which all parameters are updated. Finally, to facilitate research on retrievers' cross-topic generalizability, we curate and release an academic retrieval dataset with 18K query-results pairs in 87 topics, making it the largest topic-specific one to date.

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

↑