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

相关推荐
海盗1234几秒前
微软技术日报 2026-10-29:Copilot harness 进 Foundry,云重建能拉镜像
人工智能·microsoft·机器人·aigc·copilot
发量惊人的中年网工4 分钟前
做 AI 应用怎么选大模型 API?选项应该看哪几点?
人工智能·ai·大模型
zhikouai8 分钟前
只读权限与连接器写入:一款AI助手应用的数据通路设计
人工智能
lucas_AI10 分钟前
别让大模型一个字一个字点手机了:GUI-JEPA 用「打分」替代「解码」,快 3 倍只花 6% 参数
人工智能·agent
FOORIR 客流统计13 分钟前
AI 3D客流统计技术拆解:从深度数据到越线计数
人工智能·3d
一拳不是超人14 分钟前
用了两周,我给 DeepSeek Harness 写了 5 个插件
人工智能·ai编程·deepseek
愤怒的代码16 分钟前
什么是Paged Attention?
人工智能·agent
樱木花道長24 分钟前
AMD Ross 深度系列 7:六组对照实验看Ross如何赋能AI完成Vivado 硬核IP配置
人工智能·fpga开发
QYRese24 分钟前
CAGR3.1%!抗辐射相机市场迎替换潮,核设施数字化、水下场景带来增量
大数据·数码相机