跨18语言的多语言信息检索数据集
MIRACL项目提供了一个跨18种语言的多语言信息检索数据集,涵盖全球超过30亿母语使用者。该数据集包含16种已知语言的Wikipedia语料库,并提供相应的训练和开发数据。项目还包括基于BM25和mDPR的基线系统,以及使用Pyserini复现结果的指南。这一资源为多语言信息检索研究提供了有价值的支持。
MIRACL 🌍🙌🌏(多语言信息检索跨语言连续体)是WSDM 2023杯挑战赛,专注于18种不同语言的搜索,这些语言覆盖了全球超过30亿的母语使用者。
活动网站可在miracl.ai
找到。
本仓库提供了访问实际数据集的指引。
欲了解更多详情,请查看我们的arXiv论文:创造奇迹:跨语言连续体的多语言信息检索。
与我们联系!
MIRACL使用的维基百科语料库可作为HuggingFace数据集获取。 目前,我们已发布了16种"已知语言"的语料库;剩余2种"惊喜语言"将在稍后公布!
\n\n
)分割成多个段落。
这些段落中的每一个都构成一个"文档"或检索单位。
我们保留每个段落的维基百科文章标题。语料库数据文件采用JSON lines格式,并用gzip
压缩。
文件中的每一行对应一个段落。
以英语语料库中的一个例子为例:
{
"docid": "39#0",
"title": "Albedo",
"text": "Albedo(意为"白度")是天体(如地球等行星)接收的总太阳辐射中漫反射太阳辐射的度量。它是无量纲的,以0到1的比例尺度测量(0对应于吸收所有入射辐射的黑体,1对应于反射所有入射辐射的物体)。"
}
docid
的格式为X#Y
,其中所有具有相同X
的段落来自同一篇维基百科文章,而Y
表示该文章中的段落,按顺序编号。
text
字段包含段落的文本。
title
字段包含该段落所属文章的名称。
MIRACL训练集和开发集中16种已知语言的主题(= 查询)和相关性判断(= 相关性标签)可在HuggingFace Dataset上获得!
🤗 = 直接链接到HuggingFace Dataset
训练集 | 开发集 | ||||
---|---|---|---|---|---|
语言 | 问题数 | 判断数 | 问题数 | 判断数 | 链接 |
阿拉伯语 (ar) | 3,495 | 25,382 | 2,896 | 29,197 | 🤗 |
孟加拉语 (bn) | 1,631 | 16,754 | 411 | 4,206 | 🤗 |
英语 (en) | 2,863 | 29,416 | 799 | 8,350 | 🤗 |
西班牙语 (es) | 2,162 | 21,531 | 648 | 6,443 | 🤗 |
波斯语 (fa) | 2,107 | 21,844 | 632 | 6,571 | 🤗 |
芬兰语 (fi) | 2,897 | 20,350 | 1,271 | 12,008 | 🤗 |
法语 (fr) | 1,143 | 11,426 | 343 | 3,429 | 🤗 |
印地语 (hi) | 1,169 | 11,668 | 350 | 3,494 | 🤗 |
印度尼西亚语 (id) | 4,071 | 41,358 | 960 | 9,668 | 🤗 |
日语 (ja) | 3,477 | 34,387 | 860 | 8,354 | 🤗 |
韩语 (ko) | 868 | 12,767 | 213 | 3,057 | 🤗 |
俄语 (ru) | 4,683 | 33,921 | 1,252 | 13,100 | 🤗 |
斯瓦希里语 (sw) | 1,901 | 9,359 | 482 | 5,092 | 🤗 |
泰卢固语 (te) | 3,452 | 18,608 | 828 | 1,606 | 🤗 |
泰语 (th) | 2,972 | 21,293 | 733 | 7,573 | 🤗 |
中文 (zh) | 1,312 | 13,113 | 393 | 3,928 | 🤗 |
总计 | 40,203 | 343,177 | 13,071 | 126,076 |
上表显示了每种(语言,数据集划分)组合中的查询数量(#Q
)和判断数量(#J
),其中判断包括正面和负面标签。
主题以TSV格式呈现,每行按以下方式组织:
qid\t查询
相关性判断采用标准TREC qrels格式,如下所示:
qid Q0 文档id 相关性
我们已发布了使用BM25、mDPR以及两者混合的基线模型,详见我们的arXiv论文。 BM25和mDPR的结果可以使用Pyserini复现。
复现我们的基线模型步骤如下:
tools/topics-and-qrels
下。主题和qrels文件可在HuggingFace数据集的miracl-v1.0-${lang}/topics
和miracl-v1.0-${lang}/qrels
下找到。
git clone https://huggingface.co/datasets/miracl/miracl
mv miracl/*/*/* $PYSERINI_PATH/tools/topics-and-qrels/
请注意,上述2CR仅用于复现搜索阶段,其中索引是预先计算好并由Pyserini自动加载的。如果您想复现索引构建阶段,请参考此文档。
castorini/mdpr-tied-pft-msmarco
facebook/mcontriever-msmarco
castorini/mdpr-tied-pft-msmarco-ft-miracl-{lang}
,其中{lang}
为两字母ISO代码(如ar
、bn
等)如果您发现这个数据集和仓库有帮助,请按如下方式引用MIRACL:
@article{10.1162/tacl_a_00595,
author = {Zhang, Xinyu and Thakur, Nandan and Ogundepo, Odunayo and Kamalloo, Ehsan and Alfonso-Hermelo, David and Li, Xiaoguang and Liu, Qun and Rezagholizadeh, Mehdi and Lin, Jimmy},
title = "{MIRACL: A Multilingual Retrieval Dataset Covering 18 Diverse Languages}",
journal = {Transactions of the Association for Computational Linguistics},
volume = {11},
pages = {1114-1131},
year = {2023},
month = {09},
issn = {2307-387X},
doi = {10.1162/tacl_a_00595},
url = {https://doi.org/10.1162/tacl\_a\_00595},
eprint = {https://direct.mit.edu/tacl/article-pdf/doi/10.1162/tacl\_a\_00595/2157340/tacl\_a\_00595.pdf},
}
如果您有任何问题,欢迎发送电子邮件至project.miracl [at] gmail.com或在此仓库下提出Github问题。
AI辅助编程,代码自动修复
Trae是一种自适应的集成开发环境(IDE),通过自动化和多元协作改变开发流程。利用Trae,团队能够更快速、精确地编写和部署代码,从而提高编程效率和项目交付速度。Trae具备上下文感知和代码自动完成功能,是提升 开发效率的理想工具。
最强AI数据分析助手
小浣熊家族Raccoon,您的AI智能助手,致力于通过先进的人工智能技术,为用户提供高效、便捷的智能服务。无论是日常咨询还是专业问题解答,小浣熊都能以快速、准确的响应满足您的需求,让您的生活更加智能便捷。
像人一样思考的AI智能体
imini 是一款超级AI智能体,能根据人类指令,自主思考、自主完成、并且交付结果的AI智能体。
AI数字人视频创作平台
Keevx 一款开箱即用的AI数字人视频创作平台,广泛适用于电商广告、企业培训与社媒宣传,让全球企业与个人创作者无需拍摄剪辑,就能快速生成多语言、高质量的专业视频。
一站式AI创作平台