MTRAG-UN
收藏资源简介:
MTRAG-UN是由IBM研究院构建的多轮对话基准数据集,聚焦检索增强生成技术中的未回答、未明确、非独立问题及模糊响应等挑战场景。该数据集包含666项任务,涵盖6个领域超过2800轮对话,数据来源于人工标注和合成生成的混合方法,涉及银行、电信等企业级语料。其核心目标是推动大语言模型在复杂多轮交互中的意图理解与响应生成能力,尤其适用于客服对话系统等实际应用场景的优化研究。
MTRAG-UN is a multi-turn dialogue benchmark dataset developed by IBM Research, targeting challenging scenarios in retrieval-augmented generation (RAG) technology, including unanswered, ambiguous, non-independent questions and vague responses. This dataset consists of 666 tasks, covering over 2800 dialogue turns across 6 domains. Its data is collected via a hybrid approach of manual annotation and synthetic generation, involving enterprise-level corpora from sectors such as banking and telecommunications. Its core objective is to advance the intent understanding and response generation capabilities of large language models (LLMs) in complex multi-turn interactions, and it is particularly suitable for optimization research in practical application scenarios such as customer service dialogue systems.
MTRAG: Multi-Turn RAG Benchmark 数据集概述
数据集简介
MTRAG 是一个全面且多样化的人工生成的多轮检索增强生成(RAG)数据集,并附带四个文档语料库。据我们所知,MTRAG 是第一个端到端的人工生成多轮 RAG 基准测试,反映了多轮对话的真实世界特性。
核心组成部分
1. 文档语料库
基准测试建立在来自 4 个领域的文档语料库上:ClapNQ、Cloud、FiQA 和 Govt。ClapNQ 和 FiQA 是来自 QA/IR 数据集的现有语料库,而 Govt 和 Cloud 是专门为此基准测试组装的新的语料库。
| 语料库 | 领域 | 数据文件 | 文档数量 | 段落数量 |
|---|---|---|---|---|
| ClapNQ | 维基百科 | https://github.com/IBM/mt-rag-benchmark/blob/main/corpora/passage_level/clapnq.jsonl.zip | 4,293 | 183,408 |
| Cloud | 技术文档 | https://github.com/IBM/mt-rag-benchmark/blob/main/corpora/passage_level/cloud.json.zip | 57,638 | 61,022 |
| FiQA | 金融 | https://github.com/IBM/mt-rag-benchmark/blob/main/corpora/passage_level/fiqa.jsonl.zip | 7,661 | 49,607 |
| Govt | 政府 | https://github.com/IBM/mt-rag-benchmark/blob/main/corpora/passage_level/govt.jsonl.zip | 8,578 | 72,422 |
2. 人工生成数据
MTRAG 包含 110 个多轮对话,这些对话被转换为 842 个评估任务。
主要特征
- 多样化的问题类型
- 可回答、不可回答、部分回答和会话式问题
- 多轮:后续问题和澄清
- 四个领域
- 相关和不相关的段落
对话数据
提供 110 个对话的基准测试,格式为对话格式,平均每个对话 7.7 轮。每个对话都基于单个语料库领域,并包含各种问题类型、可回答性和多轮维度。
检索任务
每个领域的检索任务采用 BEIR 格式,仅针对可回答和部分回答任务。
| 名称 | 语料库 | 查询文件 |
|---|---|---|
| ClapNQ | https://github.com/IBM/mt-rag-benchmark/blob/main/corpora/passage_level/clapnq.jsonl.zip | https://github.com/IBM/mt-rag-benchmark/tree/main/human/retrieval_tasks/clapnq/ |
| Cloud | https://github.com/IBM/mt-rag-benchmark/blob/main/corpora/passage_level/cloud.json.zip | https://github.com/IBM/mt-rag-benchmark/tree/main/human/retrieval_tasks/cloud/ |
| FiQA | https://github.com/IBM/mt-rag-benchmark/blob/main/corpora/passage_level/fiqa.jsonl.zip | https://github.com/IBM/mt-rag-benchmark/tree/main/human/retrieval_tasks/fiqa/ |
| Govt | https://github.com/IBM/mt-rag-benchmark/blob/main/corpora/passage_level/govt.jsonl.zip | https://github.com/IBM/mt-rag-benchmark/tree/main/human/retrieval_tasks/govt/ |
生成任务
对话被转换为 842 个任务。任务是一个包含所有先前轮次以及最后一个用户问题的对话轮次。生成任务在三种检索设置下测量性能。
| 设置 | 描述 | 文件 |
|---|---|---|
| Reference | 使用参考段落生成 | https://github.com/IBM/mt-rag-benchmark/blob/main/human/generation_tasks/reference.jsonl |
| Reference + RAG | 检索后生成,但参考段落保留在前 5 个段落中 | https://github.com/IBM/mt-rag-benchmark/blob/main/human/generation_tasks/reference+RAG.jsonl |
| Full RAG | 检索后生成,检索结果由前 5 个段落组成 | https://github.com/IBM/mt-rag-benchmark/blob/main/human/generation_tasks/RAG.jsonl |
实验结果
提供论文中实验的生成结果分析文件。
| 设置 | 描述 | 文件 |
|---|---|---|
| Reference | 使用参考段落生成 | https://github.com/IBM/mt-rag-benchmark/blob/main/human/evaluations/reference.json |
| Reference + RAG | 检索后生成,但参考段落保留在前 5 个段落中 | https://github.com/IBM/mt-rag-benchmark/blob/main/human/evaluations/reference+RAG.json |
| Full RAG | 检索后生成,检索结果由前 5 个段落组成 | https://github.com/IBM/mt-rag-benchmark/blob/main/human/evaluations/RAG.json |
| Human Evaluation Reference | 在具有人工评估的子集上使用参考段落生成 | https://github.com/IBM/mt-rag-benchmark/blob/main/human/evaluations/reference_subset_with_human_evaluations.json |
3. 合成数据
提供 200 个遵循人工数据属性的合成生成对话。
生成任务
| 设置 | 描述 | 文件 |
|---|---|---|
| Reference | 使用参考段落生成 | https://github.com/IBM/mt-rag-benchmark/blob/main/synthetic/generation_tasks/synthetic.jsonl |
使用指南
- 运行检索:可以使用 BEIR 代码库运行检索实验。
- 运行生成:可以使用任何所需模型(例如 HuggingFace 上的模型)和设置运行生成实验。
- 评估检索和生成:可以使用提供的评估脚本评估检索和生成实验。
- 查看评估结果:提供 InspectorRAGet 格式的分析文件,可用于检查评估结果并进行进一步分析。
相关论文
描述基准测试和实验的论文可在 Arxiv 获取:
- 标题:MTRAG: A Multi-Turn Conversational Benchmark for Evaluating Retrieval-Augmented Generation Systems
- 链接:https://arxiv.org/abs/2501.03468
联系方式
- Sara Rosenthal: sjrosenthal@us.ibm.com
- Yannis Katsis: yannis.katsis@ibm.com
- Marina Danilevsky: mdanile@us.ibm.com



