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

相关推荐
ksueh6 分钟前
AI写小说工具怎么选?2026年最新实测数据展示
人工智能·ai写作·ai助手
自动驾驶之心14 分钟前
ResWorld:端到端自动驾驶的时序残差世界模型(北航&中关村实验室)
人工智能·机器学习·自动驾驶
墨染天姬17 分钟前
【AI】Hermes Agent 提示词优化与闭环机制
人工智能
蓝田~29 分钟前
大模型本地部署与远程调用 — 从 API 到 Agent
java·人工智能·claude·claude code
猿的天空33 分钟前
AI视觉原生统一!商汤开源视觉任务大统一模型SenseNova-Vision
人工智能·计算机·ai·程序员·大模型·编程·智能体
EAIReport39 分钟前
2026开源AI应用平台深度对比:Dify vs AIFlowy 企业落地选型指南
人工智能·开源
神奇小汤圆40 分钟前
Loop Engineering 实战:/goal 命令让 AI 自己写完整项目
人工智能
Anova.YJ40 分钟前
AI Notebook
人工智能·python·机器学习
秦先生在广东43 分钟前
Agent Island:把 Claude Code / Codex 的会话状态塞进刘海儿屏
人工智能
阿里云大数据AI技术1 小时前
EMR Serverless Daft 如何简化多模态数据处理:视频抽帧、清洗、标注全流程与具身智能实践
人工智能·spark