中文

You Only Read Once (YORO):用于文本到 SQL 的数据库知识内化学习

计算与语言 2024-09-19 v1

摘要

尽管在文本到 SQL 任务上取得了显著进展,最近的解决方案在每个问题上反复编码相同的 database schema,导致推理成本高企,常常忽视关键 database 知识。为此,我们提出 You Only Read Once (YORO),一种新型范式,直接在训练期间将 database 知识内化到文本到 SQL 模型的可参数知识中,从而在推理时消除 schema 编码的需求。YORO 将输入 token 长度降低 66%-98%。尽管输入更短,我们的实验结果表明,YORO 在三个基准数据集上与传统系统保持竞争性能,并在大型 database 上显著优于传统方法。此外,YORO 在处理诸如缩写等具备挑战性值检索问题方面表现突出。

关键词

引用

@article{arxiv.2409.12172,
  title  = {You Only Read Once (YORO): Learning to Internalize Database Knowledge for Text-to-SQL},
  author = {Hideo Kobayashi and Wuwei Lan and Peng Shi and Shuaichen Chang and Jiang Guo and Henghui Zhu and Zhiguo Wang and Patrick Ng},
  journal= {arXiv preprint arXiv:2409.12172},
  year   = {2024}
}