千问怎么做文本相似度匹配?找出两段文字之间的语义相似程度

冬雪君_4161

冬雪君_4161

2026-05-20

369人浏览

原创

应使用qwen3-embedding-4b生成句向量并计算余弦相似度,或结合vllm加速批量推理,必要时接入qwen3-reranker-0.6b进行交叉编码精细化打分。

☞☞☞AI 智能聊天, 问答助手, AI 智能搜索, 多模态理解力帮你轻松跨越从0到1的创作门槛☜☜☜

千问怎么做文本相似度匹配?找出两段文字之间的语义相似程度

如果您希望量化两段中文文本在语义层面的接近程度,而非仅依赖字面重复或关键词匹配,则需要借助专为语义建模设计的嵌入模型。Qwen3-Embedding-4B 是通义千问团队开源的高质量文本向量化模型,它能将任意长度文本映射为2560维稠密向量,使语义相近的文本在向量空间中距离更近。以下是实现该目标的具体方法:

一、使用 Qwen3-Embedding-4B 生成句向量并计算余弦相似度

该方法基于双塔结构,对输入文本独立编码,输出高维语义向量;通过计算两个向量间的余弦值反映语义相似程度,取值范围为[-1, 1],越接近1表示语义越一致。此方式支持长文本(最高32768 tokens)、多语言,并具备动态降维能力以适配不同硬件条件。

1、安装依赖库:执行 pip install transformers torch sentence-transformers 命令获取基础运行环境。

2、加载模型与分词器:调用 AutoModel.from_pretrained("Qwen/Qwen3-Embedding-4B") 和 AutoTokenizer.from_pretrained("Qwen/Qwen3-Embedding-4B") 初始化组件。

3、对两段文本分别进行编码:使用 model.encode([text_a, text_b]) 获取对应向量,注意启用 normalize_embeddings=True 参数确保向量单位化。

4、计算余弦相似度:利用 torch.nn.functional.cosine_similarity 对两个输出向量执行运算,结果即为最终相似度分数。

二、结合 vLLM 加速批量向量推理

当需处理大量文本对(如文档去重、知识库检索)时,单次CPU推理效率偏低。vLLM 提供高效的PagedAttention机制与连续批处理能力,可显著提升吞吐量,尤其适合部署在A10/A100等显卡上运行Qwen3-Embedding-4B。

1、安装 vLLM 支持包:运行 pip install vllm 完成扩展依赖安装。

Aliyun Qwen Omni
Aliyun Qwen Omni

适用于阿里云 Model Studio Qwen Omni 模型的全模态(图像+音频)理解与生成任务。

下载

2、启动嵌入服务:执行 python -m vllm.entrypoints.openai.api_server --model Qwen/Qwen3-Embedding-4B --dtype half --tensor-parallel-size 1 启动本地API服务。

3、发送嵌入请求:构造 JSON 请求体,包含 input 字段传入待编码文本列表,调用 /v1/embeddings 接口获取向量数组。

4、本地计算相似度:将返回的两组向量导入 NumPy,使用 numpy.dot(a, b) / (numpy.linalg.norm(a) * numpy.linalg.norm(b)) 得到标量结果。

三、接入通义千问 Reranker 模型进行精细化打分

当仅靠向量距离难以区分歧义场景(例如“苹果手机”与“苹果水果”共现于同一段落),可引入 Qwen3-Reranker-0.6B 模型。它采用交叉编码架构,将查询与候选文本拼接后联合建模,输出0~1之间的细粒度相关性概率,更适合高精度匹配任务。

1、加载重排序模型:调用 AutoModelForSequenceClassification.from_pretrained("Qwen/Qwen3-Reranker-0.6B") 并设置 num_labels=1。

2、构建输入格式:将两段文本按 "[QUERY]"+query_text+"[DOC]"+doc_text 方式拼接,送入分词器获得 input_ids 与 attention_mask。

3、执行前向传播:调用 model(**inputs).logits.squeeze() 获取原始打分值。

4、归一化输出:应用 Sigmoid 函数 torch.sigmoid(score) 将结果压缩至0~1区间,作为最终语义相关性置信度。

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
LLVM自定义Pass怎么写
LLVM自定义Pass怎么写

本专题聚焦LLVM自定义Pass开发,整理Pass类结构、run()方法、PreservedAnalyses、CMake构建、插件注册、-load-pass-plugin加载和测试用例编写流程。

2026.09.30

80

10

LLVM RISC-V参数配置教程
LLVM RISC-V参数配置教程

本专题介绍LLVM对RISC-V基础ISA和扩展的支持方式,涵盖RV32、RV64、标准扩展、实验性扩展、厂商扩展、-menable-experimental-extensions和版本差异。

2026.09.30

80

14

LLVM IR中间表示入门指南
LLVM IR中间表示入门指南

本专题整理LLVM IR的核心概念,包括中间表示作用、模块结构、函数、基本块、SSA形式、类型系统和常见语法,帮助新手理解LLVM编译流程中的关键层。

2026.09.30

40

12

PDF转图片方法
PDF转图片方法

需要把 PDF 页面用于上传、预览、分享或图片归档时,PDF 转图片方法专题整理 JPG/PNG 格式选择、逐页导出、清晰度设置、批量下载和结果检查等流程,帮助用户稳定完成 PDF 图片化处理。

2026.09.30

40

26

PixTV AI视频生成与无限画布创作
PixTV AI视频生成与无限画布创作

PixTV专题整理AI视频与视觉内容创作相关功能使用教程,涵盖AI生图、视频生成、无限画布、多模型创作、素材管理、声音音乐及视频剪辑等功能,帮助用户快速掌握PixTV从创意到成片的完整制作方法。

2026.09.29

60

15

Buffalo框架数据库开发全教程
Buffalo框架数据库开发全教程

本专题围绕Buffalo框架数据库开发,讲解database.yml多环境配置、soda与fizz迁移生成回滚、模型结构体标签、增删改查与条件查询、一对多与多对多关联、数据校验、回调钩子、事务处理及原生SQL执行能力。

2026.09.23

280

15

Buffalo框架路由与请求处理实操指南
Buffalo框架路由与请求处理实操指南

本专题讲解Buffalo框架路由与请求处理机制,涵盖路由注册与分组、资源路由、Handler编写规范、Context上下文方法、参数绑定、中间件编写挂载、Session与Cookie读写、Flash消息及错误页面定制方法。

2026.09.23

160

15

Buffalo框架零基础入门教程
Buffalo框架零基础入门教程

本专题整理Buffalo框架入门内容,涵盖Go环境准备、buffalo CLI安装、新项目生成、目录结构说明、dev热加载启动、数据库连接配置与常见报错排查,帮助新手按约定优于配置的思路跑通第一个Buffalo框架应用。

2026.09.23

120

15

Conan创建软件包配方指南
Conan创建软件包配方指南

本专题介绍通过conanfile.py创建软件包的方法,讲解包名、版本、依赖和构建设置等基础信息,以及source、build、package、package_info等常用方法的作用及编写思路。

2026.09.22

80

12

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
通义千问基本用法
通义千问基本用法

共1课时 | 228人学习

通义千问的提示词工程
通义千问的提示词工程

共1课时 | 258人学习