arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2103.06402cs.CL
收藏

自监督文本到SQL学习与头部对齐训练

Self-supervised Text-to-SQL Learning with Header Alignment Training

KAIST(韩国科学技术院);Kookmin Bank(国民银行)

Donggyu Kim, Seanie Lee

摘要:

由于我们可以在没有任何人为监督的情况下利用大量未标记的数据来训练模型并将知识转移到目标任务,因此自监督学习是深度学习最近在各个领域取得成功的事实上的组成部分。然而,在许多情况下,在自监督学习目标和任务特定目标之间存在差异。为了解决文本到SQL任务中的这种差异,我们提出了一种新颖的自监督学习框架。我们利用Text-to-SQL任务的特定属性和表内容的底层结构来训练模型,以从未标记的表数据中学习\textit{header-column}对齐任务的有用知识。我们能够将知识转移到带有注释样本的监督文本到SQL训练中,以便模型可以利用知识更好地执行\textit{header-span}对齐任务来预测SQL语句。实验结果表明,我们的自监督学习框架显着提高现有的强BERT为基础的模型,而不使用大型外部语料库的性能。特别是,我们的方法对于训练具有稀缺标记数据的模型是有效的。这项工作的源代码可以在GitHub上找到。

英文摘要:

Since we can leverage a large amount of unlabeled data without any human supervision to train a model and transfer the knowledge to target tasks, self-supervised learning is a de-facto component for the recent success of deep learning in various fields. However, in many cases, there is a discrepancy between a self-supervised learning objective and a task-specific objective. In order to tackle such discrepancy in Text-to-SQL task, we propose a novel self-supervised learning framework. We utilize the task-specific properties of Text-to-SQL task and the underlying structures of table contents to train the models to learn useful knowledge of the \textit{header-column} alignment task from unlabeled table data. We are able to transfer the knowledge to the supervised Text-to-SQL training with annotated samples, so that the model can leverage the knowledge to better perform the \textit{header-span} alignment task to predict SQL statements. Experimental results show that our self-supervised learning framework significantly improves the performance of the existing strong BERT based models without using large external corpora. In particular, our method is effective for training the model with scarce labeled data. The source code of this work is available in GitHub.

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

↑