用ChatGPT生成新领域文本转SQL数据,提升模型泛化能力
REFORMER: A ChatGPT-Driven Data Synthesis Framework Elevating Text-to-SQL Models
- 基于检索-编辑框架,用ChatGPT补全掩码问题生成新配对
- 在多个新领域上超越已有数据增强方法,提升准确率
- 适合需快速适配新数据库的开发者和研究者
现有文本转SQL模型受限于训练数据不足,难以在新领域应用。本文提出REFORMER框架,利用ChatGPT无需额外训练即可生成适配新领域的(问题, SQL)配对。方法基于“检索-编辑”策略,通过ChatGPT解释SQL查询来填充掩码问题。实验表明,该方法持续优于以往数据增强技术。此外,我们还通过改写原始问题及自动生成的SQL描述进行数据扩充,结果证实改写生成问题可有效增强数据多样性。周期一致性验证在适当条件下仍具价值。
原文摘要 · Abstract (English)
The existing Text-to-SQL models suffer from a shortage of training data, inhibiting their ability to fully facilitate the applications of SQL queries in new domains. To address this challenge, various data synthesis techniques have been employed to generate more diverse and higher quality data. In this paper, we propose REFORMER, a framework that leverages ChatGPT's prowess without the need for additional training, to facilitate the synthesis of (question, SQL query) pairs tailored to new domains. Our data augmentation approach is based on a "retrieve-and-edit" method, where we generate new questions by filling masked question using explanation of SQL queries with the help of ChatGPT. Furthermore, we demonstrate that cycle consistency remains a valuable method of validation when applied appropriately. Our experimental results show that REFORMER consistently outperforms previous data augmentation methods. To further investigate the power of ChatGPT and create a general data augmentation method, we also generate the new data by paraphrasing the question in the dataset and by paraphrasing the description of a new SQL query that is generated by ChatGPT as well. Our results affirm that paraphrasing questions generated by ChatGPT help augment the original data.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。