MIGA: A Unified Multi-task Generation Framework for Conversational Text-to-SQL
This work addresses conversational text-to-SQL, a domain-specific task in natural language processing, with incremental improvements over existing methods.
The paper tackles the problem of translating multi-turn natural language questions into SQL queries by proposing MIGA, a two-stage multi-task generation framework that leverages pre-trained language models, achieving state-of-the-art performance on SparC and CoSQL benchmarks.
Conversational text-to-SQL is designed to translate multi-turn natural language questions into their corresponding SQL queries. Most state-of-the-art conversational text- to-SQL methods are incompatible with generative pre-trained language models (PLMs), such as T5. In this paper, we present a two-stage unified MultI-task Generation frAmework (MIGA) that leverages PLMs' ability to tackle conversational text-to-SQL. In the pre-training stage, MIGA first decomposes the main task into several related sub-tasks and then unifies them into the same sequence-to-sequence (Seq2Seq) paradigm with task-specific natural language prompts to boost the main task from multi-task training. Later in the fine-tuning stage, we propose four SQL perturbations to alleviate the error propagation problem. MIGA tends to achieve state-of-the-art performance on two benchmarks (SparC and CoSQL). We also provide extensive analyses and discussions to shed light on some new perspectives for conversational text-to-SQL.