arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.22223cs.IRcs.CLcs.LG
收藏

SQaLe:基于真实模式的大型文本到SQL语料库

SQaLe: A Large Text-to-SQL Corpus Grounded in Real Schemas

Centrum Wiskunde & Informatica(荷兰数学与信息学研究中心);University of Amsterdam(阿姆斯特丹大学)

Cornelius Wolff, Daniel Gomm, Madelon Hulsebos

AI总结:

SQaLe是一个基于真实数据库模式的大型半合成文本到SQL数据集,通过结合模式采样、问题合成和SQL构建,生成了高质量的(问题,模式,查询)三元组,用于提升文本到SQL模型的泛化能力。

摘要:

大型语言模型的发展加速了文本到SQL的进步,即将自然语言查询转换为有效的SQL查询。开发通用文本到SQL模型的关键瓶颈是缺乏具有足够模式和查询复杂性、领域覆盖和任务多样性的大规模数据集。我们介绍了SQaLe:一个基于135,875个关系数据库模式扩展自真实世界模式集合SchemaPile的大型半合成文本到SQL数据集。我们建立了一个原则性的生成流程,结合模式采样、问题合成和SQL构建,并生成了517,676个高质量的(问题,模式,查询)三元组。SQaLe数据集捕捉了现实的模式大小变化、多样的查询模式和自然语言模糊性,同时保持执行有效性。我们分析了其内容和特征,并发现与现有基准和数据集相比,SQaLe是迄今为止最现实的大型文本到SQL数据集。我们讨论了SQaLe如何使我们的数据扩展和文本到SQL研究中的模型泛化愿景成为可能。该数据集可通过:https://huggingface.co/datasets/trl-lab/SQaLe-text-to-SQL-dataset访问。

英文摘要:

Advances in large language models have accelerated progress in text-to-SQL, methods for converting natural language queries into valid SQL queries. A key bottleneck for developing generalizable text-to-SQL models is the lack of large-scale datasets with sufficient schema and query complexity, domain coverage, and task diversity. We introduce SQaLe: a large-scale semi-synthetic text-to-SQL dataset built on 135,875 relational database schemas expanded from a collection of real-world schemas, SchemaPile. We establish a principled generation pipeline which combines schema sampling, question synthesis, and SQL construction, and produce 517,676 high-quality (question, schema, query) triples. The SQaLe dataset captures realistic schema size variability, diverse query patterns, and natural language ambiguity while maintaining execution validity. We provide an analysis of its contents and characteristics, and find that SQaLe introduces the most realistic large-scale text-to-SQL dataset to date in comparison with existing benchmarks and datasets. We discuss how SQaLe enables our vision for data scaling and model generalization in text-to-SQL research. The dataset is accessible at: https://huggingface.co/datasets/trl-lab/SQaLe-text-to-SQL-dataset.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑