基于MiniLM架构优化的轻量级句子嵌入模型,专为高效句子相似度计算设计
下载量 78
发布时间 : 10/13/2023
模型简介
该模型通过对比学习微调,能够将句子编码为高维向量空间中的表示,用于计算句子间语义相似度。在保持较高性能的同时显著减小模型体积并提升推理速度。
模型特点
高效轻量
仅80MB体积,推理速度达14,200句/秒,适合资源受限环境部署
多领域适应
在17个不同领域数据集上微调,包括学术论文、问答社区和技术文档等
对比学习优化
采用批次内负采样策略和余弦相似度对比损失进行微调
模型能力
句子向量化
语义相似度计算
语义搜索支持
文本特征提取
使用案例
信息检索
问答系统匹配
将用户问题与知识库问题编码后匹配最相似结果
在MS MARCO等检索基准上表现良好
内容去重
社区问答去重
识别StackExchange等平台上的重复问题
基于WikiAnswers数据集优化
精选推荐AI模型
Llama 3 Typhoon V1.5x 8b Instruct
专为泰语设计的80亿参数指令模型,性能媲美GPT-3.5-turbo,优化了应用场景、检索增强生成、受限生成和推理任务
大型语言模型
Transformers

支持多种语言
L
scb10x
3,269
16
Cadet Tiny
Openrail
Cadet-Tiny是一个基于SODA数据集训练的超小型对话模型,专为边缘设备推理设计,体积仅为Cosmo-3B模型的2%左右。
对话系统
Transformers

英语
C
ToddGoldfarb
2,691
6
Roberta Base Chinese Extractive Qa
基于RoBERTa架构的中文抽取式问答模型,适用于从给定文本中提取答案的任务。
问答系统
中文
R
uer
2,694
98
AIbase是一个专注于MCP服务的平台,为AI开发者提供高质量的模型上下文协议服务,助力AI应用开发。
简体中文