Project Icon

miracl

跨18语言的多语言信息检索数据集

MIRACL项目提供了一个跨18种语言的多语言信息检索数据集,涵盖全球超过30亿母语使用者。该数据集包含16种已知语言的Wikipedia语料库,并提供相应的训练和开发数据。项目还包括基于BM25和mDPR的基线系统,以及使用Pyserini复现结果的指南。这一资源为多语言信息检索研究提供了有价值的支持。

image

Build License Downloads

论文 | 基准 | HuggingFace | 排行榜

🙌 MIRACL

MIRACL 🌍🙌🌏(多语言信息检索跨语言连续体)是WSDM 2023杯挑战赛,专注于18种不同语言的搜索,这些语言覆盖了全球超过30亿的母语使用者。 活动网站可在miracl.ai找到。 本仓库提供了访问实际数据集的指引。

欲了解更多详情,请查看我们的arXiv论文:创造奇迹:跨语言连续体的多语言信息检索

与我们联系!

🙌 语料库

MIRACL使用的维基百科语料库可作为HuggingFace数据集获取。 目前,我们已发布了16种"已知语言"的语料库;剩余2种"惊喜语言"将在稍后公布!

  • 🤗 = 直接链接到HuggingFace数据集
  • 🌏 = 链接到原始维基转储 | 语言 | 段落数 | 文章数 | 链接 | |:----------------|------------:|------------:|:------| | 阿拉伯语 (ar) | 2,061,414 | 656,982 | 🤗 🌏 | 孟加拉语 (bn) | 297,265 | 63,762 | 🤗 🌏 | 英语 (en) | 32,893,221 | 5,758,285 | 🤗 🌏 | 西班牙语 (es) | 10,373,953 | 1,669,181 | 🤗 🌏 | 波斯语 (fa) | 2,207,172 | 857,827 | 🤗 🌏 | 芬兰语 (fi) | 1,883,509 | 447,815 | 🤗 🌏 | 法语 (fr) | 14,636,953 | 2,325,608 | 🤗 🌏 | 印地语 (hi) | 506,264 | 148,107 | 🤗 🌏 | 印度尼西亚语 (id) | 1,446,315 | 446,330 | 🤗 🌏 | 日语 (ja) | 6,953,614 | 1,133,444 | 🤗 🌏 | 韩语 (ko) | 1,486,752 | 437,373 | 🤗 🌏 | 俄语 (ru) | 9,543,918 | 1,476,045 | 🤗 🌏 | 斯瓦希里语 (sw) | 131,924 | 47,793 | 🤗 🌏 | 泰卢固语 (te) | 518,079 | 66,353 | 🤗 🌏 | 泰语 (th) | 542,166 | 128,179 | 🤗 🌏 | 中文 (zh) | 4,934,368 | 1,246,389 | 🤗 🌏 每种语言的语料库都是从维基百科转储中准备的,我们只保留纯文本,丢弃图片、表格等内容。 每篇文章都使用WikiExtractor根据自然话语单位(例如维基标记中的\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,49525,3822,89629,197🤗
孟加拉语 (bn)1,63116,7544114,206🤗
英语 (en)2,86329,4167998,350🤗
西班牙语 (es)2,16221,5316486,443🤗
波斯语 (fa)2,10721,8446326,571🤗
芬兰语 (fi)2,89720,3501,27112,008🤗
法语 (fr)1,14311,4263433,429🤗
印地语 (hi)1,16911,6683503,494🤗
印度尼西亚语 (id)4,07141,3589609,668🤗
日语 (ja)3,47734,3878608,354🤗
韩语 (ko)86812,7672133,057🤗
俄语 (ru)4,68333,9211,25213,100🤗
斯瓦希里语 (sw)1,9019,3594825,092🤗
泰卢固语 (te)3,45218,6088281,606🤗
泰语 (th)2,97221,2937337,573🤗
中文 (zh)1,31213,1133933,928🤗
总计40,203343,17713,071126,076

上表显示了每种(语言,数据集划分)组合中的查询数量(#Q)和判断数量(#J),其中判断包括正面和负面标签。

主题以TSV格式呈现,每行按以下方式组织:

qid\t查询

相关性判断采用标准TREC qrels格式,如下所示:

qid Q0 文档id 相关性

🙌 基线模型

使用Pyserini复现结果:

我们已发布了使用BM25、mDPR以及两者混合的基线模型,详见我们的arXiv论文。 BM25和mDPR的结果可以使用Pyserini复现。

复现我们的基线模型步骤如下:

  1. 按照这些说明安装Pyserini的开发版本。(要在surprise语言上运行基线,您需要重新构建Anserini和Pyserini)
  2. 手动将所有主题和qrels文件放在tools/topics-and-qrels下。主题和qrels文件可在HuggingFace数据集miracl-v1.0-${lang}/topicsmiracl-v1.0-${lang}/qrels下找到。
    git clone https://huggingface.co/datasets/miracl/miracl
    mv miracl/*/*/* $PYSERINI_PATH/tools/topics-and-qrels/
    
  3. 按照我们的两步复现(2CR)网站上的命令操作。

请注意,上述2CR仅用于复现搜索阶段,其中索引是预先计算好并由Pyserini自动加载的。如果您想复现索引构建阶段,请参考此文档

密集模型的检查点:

  • mDPR (未在MIRACL上微调): castorini/mdpr-tied-pft-msmarco
  • mContriever (未在MIRACL上微调): facebook/mcontriever-msmarco
  • mDPR (在MIRACL上微调): castorini/mdpr-tied-pft-msmarco-ft-miracl-{lang},其中{lang}为两字母ISO代码(如arbn等)

🙌 引用

如果您发现这个数据集和仓库有帮助,请按如下方式引用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问题。

项目侧边栏1项目侧边栏2
推荐项目
Project Cover

豆包MarsCode

豆包 MarsCode 是一款革命性的编程助手,通过AI技术提供代码补全、单测生成、代码解释和智能问答等功能,支持100+编程语言,与主流编辑器无缝集成,显著提升开发效率和代码质量。

Project Cover

AI写歌

Suno AI是一个革命性的AI音乐创作平台,能在短短30秒内帮助用户创作出一首完整的歌曲。无论是寻找创作灵感还是需要快速制作音乐,Suno AI都是音乐爱好者和专业人士的理想选择。

Project Cover

有言AI

有言平台提供一站式AIGC视频创作解决方案,通过智能技术简化视频制作流程。无论是企业宣传还是个人分享,有言都能帮助用户快速、轻松地制作出专业级别的视频内容。

Project Cover

Kimi

Kimi AI助手提供多语言对话支持,能够阅读和理解用户上传的文件内容,解析网页信息,并结合搜索结果为用户提供详尽的答案。无论是日常咨询还是专业问题,Kimi都能以友好、专业的方式提供帮助。

Project Cover

阿里绘蛙

绘蛙是阿里巴巴集团推出的革命性AI电商营销平台。利用尖端人工智能技术,为商家提供一键生成商品图和营销文案的服务,显著提升内容创作效率和营销效果。适用于淘宝、天猫等电商平台,让商品第一时间被种草。

Project Cover

吐司

探索Tensor.Art平台的独特AI模型,免费访问各种图像生成与AI训练工具,从Stable Diffusion等基础模型开始,轻松实现创新图像生成。体验前沿的AI技术,推动个人和企业的创新发展。

Project Cover

SubCat字幕猫

SubCat字幕猫APP是一款创新的视频播放器,它将改变您观看视频的方式!SubCat结合了先进的人工智能技术,为您提供即时视频字幕翻译,无论是本地视频还是网络流媒体,让您轻松享受各种语言的内容。

Project Cover

美间AI

美间AI创意设计平台,利用前沿AI技术,为设计师和营销人员提供一站式设计解决方案。从智能海报到3D效果图,再到文案生成,美间让创意设计更简单、更高效。

Project Cover

稿定AI

稿定设计 是一个多功能的在线设计和创意平台,提供广泛的设计工具和资源,以满足不同用户的需求。从专业的图形设计师到普通用户,无论是进行图片处理、智能抠图、H5页面制作还是视频剪辑,稿定设计都能提供简单、高效的解决方案。该平台以其用户友好的界面和强大的功能集合,帮助用户轻松实现创意设计。

投诉举报邮箱: service@vectorlightyear.com
@2024 懂AI·鲁ICP备2024100362号-6·鲁公网安备37021002001498号