D

Dense Encoder Distilbert Frozen Emb

由 vocab-transformers 开发
基于DistilBERT架构的密集检索模型,在MS MARCO数据集上训练,词嵌入层保持冻结状态
下载量 26
发布时间 : 4/5/2022

模型简介

该模型是DistilBERT的变体,专门针对信息检索任务优化,采用MarginMSE损失函数训练,适用于文档和查询的密集向量表示生成

模型特点

冻结词嵌入训练
训练过程中保持预训练的词嵌入层参数不变,可能提高模型稳定性
MarginMSE优化
采用MarginMSE损失函数进行训练,专门优化检索任务的排序性能
轻量级架构
基于DistilBERT架构,比原始BERT模型更小更快,同时保持较好性能

模型能力

文本向量表示
语义相似度计算
信息检索
文档排序

使用案例

搜索引擎
网页搜索结果排序
为搜索引擎生成查询和文档的密集向量表示,用于相关性排序
在TREC-DL等标准检索评测中表现良好
问答系统
答案段落检索
从大量文档中快速检索与问题相关的段落
在FiQA等金融问答数据集上表现稳定
AIbase
智启未来,您的人工智能解决方案智库
简体中文