#175 · 主分类: 知识库与企业搜索

RocketQA

dense-retrieval information-retrieval nlp question-answering

🚀 RocketQA,用于信息检索和问答的密集检索,包含中英文最先进的模型。

项目最后更新:12/19/23

GitHub Stars

784

Forks数量

123

贡献者数量

12

许可证

Apache-2.0

收录理由

开源社区里大多数稠密检索模型都以英文为主,能同时提供中英文预训练模型的工具包并不多见,RocketQA 正好补上了这个缺口。它采用基于 ERNIE 的双编码器做候选召回,再用交叉编码器精排,这种两阶段流程正是生产环境检索系统常用的架构,而不是只给一个单一的嵌入模型。接入现有技术栈也比较省事:加载模型、编码查询和段落、用 Faiss 或 JINA 建索引,官方还提供了两种方案的完整示例。仓库里同时保留了论文对应的训练代码和基线模型,想在自己的领域数据上微调,或者复现论文结果,都有现成材料可用。对于搭建中文知识库、企业搜索或检索增强生成系统的团队来说,这个项目尤其有价值,因为高质量的开源中文稠密检索模型相对稀缺。

相关文章

暂无针对该项目的文章。

如需补充新闻资讯、使用教程、应用测评等内容,欢迎 联系我们推荐选题或投稿。

同类热门项目