M

Multi Qa MiniLM BERT Tiny Distill L 2 H 128 A Cos V1

由 rawsh 开发
这是一个基于BERT-Tiny的轻量级句子嵌入模型,专为语义搜索和句子相似度任务设计,模型大小仅为5MB。
下载量 43
发布时间 : 6/5/2023

模型简介

该模型能将句子和段落映射到128维的密集向量空间,适用于聚类或语义搜索等任务。基于nreimers/BERT-Tiny_L-2_H-128_A-2模型,通过知识蒸馏从multi-qa-MiniLM-L6-cos-v1教师模型学习。

模型特点

轻量级设计
模型大小仅为5MB,适合资源受限的环境
知识蒸馏
从性能更强的multi-qa-MiniLM-L6-cos-v1教师模型学习
高效语义表示
将文本映射到128维向量空间,保留语义信息

模型能力

句子相似度计算
文本特征提取
语义搜索
文本聚类

使用案例

信息检索
问答系统
用于匹配用户问题与知识库中的候选答案
在STS-dev数据集上达到0.7336的余弦相似度Pearson系数
内容推荐
相似内容推荐
基于语义相似度推荐相关内容
AIbase
智启未来,您的人工智能解决方案智库
简体中文