SParC数据集介绍

导语

SParC是Text-to-SQL领域的一个多轮查询数据集。本篇博客将对该数据集论文和数据格式进行简要介绍。

SParC数据集概述

SParC是一个跨领域的多轮Text-to-SQL数据集。它包含有4298个问题轮次,大约有12k+的自然语言问句到SQL标注的Question-SQL对。这些问题来自于138个不同领域的200个复杂数据库。

SParC对研究者带来的挑战主要有以下三点:

  1. 这是一个多轮对话形式的Text-to-SQL解析,模型需要考虑复杂的上下文依赖关系;
  2. 由于对话的引入,使得数据集具有更大的语义多样性;
  3. 具有跨域特性,即验证和测试是在与训练集完全不同的数据库schema上进行,因而模型需要有足够强的泛化性能。

如下是一次完整的多轮对话示例。

如上图所示,数据库的简介记为 D 1 D_1 D1,简要介绍了这个数据库的信息,即是一个关于学生宿舍的数据库,包含有5个数据表。而这次交互的目的是 C 1 C_1 C1,即找到住在拥有电视机休息室的宿舍的学生的姓名。为了实现这个目标,一共进行了四次对话。用户的自然语言问句记为 Q Q Q,系统解析后的SQL语句记为 S S S。

下面将详细介绍一下SParC数据集的收集过程。

数据收集

SParC数据集的创建包括四个阶段:

  1. 选择交互目标;
  2. 提出问题;
  3. 进行SQL标注;
  4. 检查回顾。

选择交互目标

为了确保问题序列的主题相关性,SParC使用了Spider数据集(之前的一个单轮Text-to-SQL数据集)中的一些问题作为参考而制定了交互目标。每一系列问题都是围绕着这个最终的交互目标进行。这里SParC使用了Spider数据集中的中等、困难和非常困难的问题作为参考。并在交互过程中为了确保问题的多样性而添加了一些简单的问题。经过选择后,一共在200个数据库上制定了4437个交互目标。

提出问题

SParC共邀请了15个SQL经验的大学生来进行提问和标注。每个系列中的相邻问题之间有着如下四种主题关联:

  1. 约束更加细化:当前的问句和上一个问句相比,问了一个相同类型的实体但是有着不同的约束。比如,前一个问题问:"哪一个专业的学生人最少?;而当前的问题问:"那最受欢迎的专业呢?"
  2. 主题扩展:当前的问句询问上一个问句的同一个实体的另外的属性。例如,前一个问句问:"Anonymous Donor Hall的容量是多少?";而当前的问句问:"列出它所有的设施。"
  3. 属性转移:当前的问句询问另一个实体的相同属性。例如,前一个问句问:"告诉我那部叫做Double Down的电视剧的评分。";而当前的问题则是:"那Keepers(这部电视剧)呢?"
  4. 答案更加细化:当前的问句徐闻上一个问题的答案给出的实体中的某些子集的其他属性。例如,前一个问题是:"请列出所有不同的部门名称。";而现在的问句是"在统计部门的所有员工的平均薪资是多少?"

SQL标注

在进行完提问后,每个标注者都将对自己的问题进行SQL转化,并确保在相应数据库上执行得到正确的答案。

检查

最后,由英语为母语的标注者将问题进行检查,确保没有语法错误。

数据集统计和分析

SParC数据集的统计特性如下图,相比于ATIS(一个古老的用于航班订票的单数据库Text-to-SQL数据集)。SParC拥有更多的数据库和数据表,同时规模也更大。

同时,SParC对于SQL中的各种高级语法结构的使用也更加丰富。如下图所示,对于像ORDER,HAVING,SET,GROUP等SQL结构,ATIS几乎很少或者并不存在这样的结构。

数据集划分

SParC数据集的划分与SPider数据集保持了一致,即140个数据库训练,20个数据库验证,40个数据库进行测试。其数据规模见下图。

数据集格式介绍

该数据集的下载链接为Link

原始文件为zip文件,解压后可以看到由以下几个数据文件和数据库文件夹组成:

其中database文件夹中包含了这200个数据库的内容和schema信息。tables.json则包含了json格式的数据库schema信息。

这里以train.json中的第一个交互为例,分析样例数据。

json 复制代码
    {
        "database_id": "hospital_1", 
        "interaction": [
            {
                "query": "SELECT count(departmentID) FROM department GROUP BY departmentID", 
                "utterance_toks": [
                    "What", 
                    "is", 
                    "the", 
                    "number", 
                    "of", 
                    "employees", 
                    "in", 
                    "each", 
                    "department", 
                    "?"
                ], 
                "utterance": "What is the number of employees in each department?", 
                "sql": {
                    "orderBy": [], 
                    "from": {
                        "table_units": [
                            [
                                "table_unit", 
                                1
                            ]
                        ], 
                        "conds": []
                    }, 
                    "union": null, 
                    "except": null, 
                    "groupBy": [
                        [
                            0, 
                            5, 
                            false
                        ]
                    ], 
                    "limit": null, 
                    "intersect": null, 
                    "where": [], 
                    "having": [], 
                    "select": [
                        false, 
                        [
                            [
                                3, 
                                [
                                    0, 
                                    [
                                        0, 
                                        5, 
                                        false
                                    ], 
                                    null
                                ]
                            ]
                        ]
                    ]
                }
            }, 
            {
                "query": "SELECT name FROM department GROUP BY departmentID ORDER BY count(departmentID) DESC LIMIT 1;", 
                "utterance_toks": [
                    "Which", 
                    "department", 
                    "has", 
                    "the", 
                    "most", 
                    "employees", 
                    "?", 
                    "Give", 
                    "me", 
                    "the", 
                    "department", 
                    "name", 
                    "."
                ], 
                "utterance": "Which department has the most employees? Give me the department name.", 
                "sql": {
                    "orderBy": [
                        "desc", 
                        [
                            [
                                0, 
                                [
                                    3, 
                                    5, 
                                    false
                                ], 
                                null
                            ]
                        ]
                    ], 
                    "from": {
                        "table_units": [
                            [
                                "table_unit", 
                                1
                            ]
                        ], 
                        "conds": []
                    }, 
                    "union": null, 
                    "except": null, 
                    "groupBy": [
                        [
                            0, 
                            5, 
                            false
                        ]
                    ], 
                    "limit": 1, 
                    "intersect": null, 
                    "where": [], 
                    "having": [], 
                    "select": [
                        false, 
                        [
                            [
                                0, 
                                [
                                    0, 
                                    [
                                        0, 
                                        6, 
                                        false
                                    ], 
                                    null
                                ]
                            ]
                        ]
                    ]
                }
            }
        ], 
        "final": {
            "query": "SELECT name FROM department GROUP BY departmentID ORDER BY count(departmentID) DESC LIMIT 1;", 
            "utterance": "Find the department with the most employees."
        }
    }, 

这里首先声明了它所依赖的数据库是"hospital_1",本次交互一共有2轮对话,第一轮对话为:"What is the number of employees in each department?",然后第二轮对话为:"Find the department with the most employees."。最后的"final"字段声明了最后一轮也就是整个交互的最终目标。其中关于SQL语句的token和各种成分的关系则与Spider相同,详细可以参考这篇博客Spider数据集格式介绍

dev.json则是和train.json同样的格式。

另一个dev_gold.txt则是用于评价最终结果时所使用。

相关推荐
小陈phd1 小时前
OpenCV从入门到精通实战(九)——基于dlib的疲劳监测 ear计算
人工智能·opencv·计算机视觉
Guofu_Liao2 小时前
大语言模型---LoRA简介;LoRA的优势;LoRA训练步骤;总结
人工智能·语言模型·自然语言处理·矩阵·llama
ZHOU_WUYI6 小时前
3.langchain中的prompt模板 (few shot examples in chat models)
人工智能·langchain·prompt
如若1236 小时前
主要用于图像的颜色提取、替换以及区域修改
人工智能·opencv·计算机视觉
老艾的AI世界7 小时前
AI翻唱神器,一键用你喜欢的歌手翻唱他人的曲目(附下载链接)
人工智能·深度学习·神经网络·机器学习·ai·ai翻唱·ai唱歌·ai歌曲
DK221517 小时前
机器学习系列----关联分析
人工智能·机器学习
Robot2517 小时前
Figure 02迎重大升级!!人形机器人独角兽[Figure AI]商业化加速
人工智能·机器人·微信公众平台
浊酒南街7 小时前
Statsmodels之OLS回归
人工智能·数据挖掘·回归
畅联云平台8 小时前
美畅物联丨智能分析,安全管控:视频汇聚平台助力智慧工地建设
人工智能·物联网
加密新世界8 小时前
优化 Solana 程序
人工智能·算法·计算机视觉