SGPT: GPT Sentence Embeddings for Semantic Search

简介

语义搜索分为两个部分:

1.搜索和query 相关的topk文档。

2.理解文档和query后面隐藏的语义信息,而不是字面含义。

这篇论文提出了SGPT模型,只用decoder-only的transformer来进行语义搜索和sentence向量的提取。

1.SGPT-BE:来对文档和query进行粗略的相关度计算,由于可以对文档的向量进行缓存,所以计算量和文档的数量线性相关,SGPT使用了BitFit的方式只对模型bias等少部分参数进行微调,大部分模型参数在微调的过程中是被冻结的,所以能够大大提升模型的训练效率。

2.SGPT-CE:对文档和query进行concat拼接,拼接后输入到gpt模型中去,对模型输出的query token的概率进行sum pooling的方式,作为文档的得分。由于CE的方式每一个query都需要重复计算很多次,所以计算量比较大,所以一般是在BE之后,对top的文档进行encoder概率计算。

SGPT Cross-Encoder


SGPT Bi-Encoder

相关推荐
stonewl25995 分钟前
2026工业标识国产化替代深度解读:从“能用“到“好用“,让中国制造不再依赖进口标签软件
人工智能
imbackneverdie20 分钟前
撰写系统性综述/叙述性综述,如何搭建清晰的领域发展脉络?
大数据·人工智能·aigc·论文·科研·ai写作·学术
CCYe、25 分钟前
企业内训知识助手:怎么搭才不踩坑
人工智能
2601_9632827726 分钟前
寒地专网通信实战:对讲机技术选型、组网优化与东北多行业落地全指南
大数据·数据库·人工智能
广凌股份(广凌科技)33 分钟前
广凌智慧大内控一体化平台:告别纸上内控,把握真实运行状态
大数据·人工智能
Zldaisy3d33 分钟前
船舶与海工增材制造市场迎来规范化新阶段!中国船级社新指南9月1日生效
人工智能
ZhouDevin34 分钟前
算法论文/模型微调1——CNN架构做lora微调训练
人工智能·深度学习·算法·架构·cnn
ZGi.ai35 分钟前
ZGI Runtime:切换模型后 Agent 为什么失效?
人工智能·aiagent·企业ai·模型切换·zgi·agentruntime
禾高网络37 分钟前
开启环保新潮流:回收小程序系统功能介绍
java·大数据·小程序
YH552698441 分钟前
有什么AI工具能将网课和视频播客的内容转化成文档?
人工智能·音视频