SQLformer: Deep Auto-Regressive Query Graph Generation for Text-to-SQL Translation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Bazaga, Adrián, Liò, Pietro, Micklem, Gos
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909210667646976
author Bazaga, Adrián
Liò, Pietro
Micklem, Gos
author_facet Bazaga, Adrián
Liò, Pietro
Micklem, Gos
contents In recent years, the task of text-to-SQL translation, which converts natural language questions into executable SQL queries, has gained significant attention for its potential to democratize data access. Despite its promise, challenges such as adapting to unseen databases and aligning natural language with SQL syntax have hindered widespread adoption. To overcome these issues, we introduce SQLformer, a novel Transformer architecture specifically crafted to perform text-to-SQL translation tasks. Our model predicts SQL queries as abstract syntax trees (ASTs) in an autoregressive way, incorporating structural inductive bias in the encoder and decoder layers. This bias, guided by database table and column selection, aids the decoder in generating SQL query ASTs represented as graphs in a Breadth-First Search canonical order. Our experiments demonstrate that SQLformer achieves state-of-the-art performance across six prominent text-to-SQL benchmarks.
format Preprint
id arxiv_https___arxiv_org_abs_2310_18376
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle SQLformer: Deep Auto-Regressive Query Graph Generation for Text-to-SQL Translation
Bazaga, Adrián
Liò, Pietro
Micklem, Gos
Computation and Language
Machine Learning
In recent years, the task of text-to-SQL translation, which converts natural language questions into executable SQL queries, has gained significant attention for its potential to democratize data access. Despite its promise, challenges such as adapting to unseen databases and aligning natural language with SQL syntax have hindered widespread adoption. To overcome these issues, we introduce SQLformer, a novel Transformer architecture specifically crafted to perform text-to-SQL translation tasks. Our model predicts SQL queries as abstract syntax trees (ASTs) in an autoregressive way, incorporating structural inductive bias in the encoder and decoder layers. This bias, guided by database table and column selection, aids the decoder in generating SQL query ASTs represented as graphs in a Breadth-First Search canonical order. Our experiments demonstrate that SQLformer achieves state-of-the-art performance across six prominent text-to-SQL benchmarks.
title SQLformer: Deep Auto-Regressive Query Graph Generation for Text-to-SQL Translation
topic Computation and Language
Machine Learning
url https://arxiv.org/abs/2310.18376