MAG-SQL:基于多智能体生成方法、软架构链接与迭代子SQL细化的文本到SQL转换
💡
原文中文,约500字,阅读约需2分钟。
📝
内容提要
研究人员提出了基于大型语言模型的多智能体协作文本到SQL框架MAC-SQL,用于处理庞大的数据库和复杂的用户查询。该框架由三个智能体组成,能够在Text-to-SQL数据集上取得令人鼓舞的结果。他们还开源了指导微调模型SQL-Llama和代理指令数据集。
🎯
关键要点
- 研究人员提出了基于大型语言模型的多智能体协作文本到SQL框架MAC-SQL。
- MAC-SQL框架由三个智能体组成:Selector、Decomposer和Refiner。
- Selector负责压缩庞大的数据库并保留用户问题的相关表模式。
- Decomposer将复杂的用户问题分解为更简单的子问题并逐步解决。
- Refiner负责验证和完善有缺陷的SQL查询。
- 在BIRD和Spider两个Text-to-SQL数据集上进行了大量实验,BIRD测试集上实现了59.59%的执行准确率。
- 开源了基于Code Llama 7B的指导微调模型SQL-Llama。
- SQL-Llama在BIRD和Spider的开发集上展示了良好的结果,但与GPT-4相比仍有提升潜力。
- 相关代码和数据可在GitHub上公开访问。
➡️