arXiv:2502.00329cs.DBcs.AI2025-02被引 12

CoddLLM让大模型更懂数据分析,能精准理解自然语言并生成复杂SQL。

CoddLLM: Empowering Large Language Models for Data Analytics

  • 用合成数据训练模型,强化对表格与文本间转换的理解。
  • 在8个数据集上表现最佳,Text-to-SQL平均提升24.9%。
  • 适合需要智能数据查询和分析的开发者与研究者。

大型语言模型(LLMs)有望通过自然语言交互简化数据发现与SQL生成等任务,推动数据智能发展。本文提出首个专为数据智能设计的基座模型训练方案,构建可扩展的合成数据生成方法,覆盖多样化的数据表示与操作主题,并引入两项新任务以连接表格与文本。基于此,我们对Mistral-NeMo-12B进行后训练,构建新模型CoddLLM。为评估模型在数据领域的理解与推理能力,我们构建AnalyticsMMLU基准,包含数千道数据库、数据分析与机器学习多选题。此外,我们贡献三个涵盖数据库与数据湖场景的综合评测集。实验表明,CoddLLM在八项数据集上达到最高平均准确率,优于GPT-3.5-Turbo,表选择任务超过GPT-4o 12.1%,Text-to-SQL平均提升24.9%。

原文摘要 · Abstract (English)

Large Language Models (LLMs) have the potential to revolutionize data analytics by simplifying tasks such as data discovery and SQL query synthesis through natural language interactions. This work serves as a pivotal first step toward the development of foundation models explicitly designed for data analytics applications. To propel this vision forward, we unveil a new data recipe for post-training LLMs, enhancing their comprehension of data management and empowering them to tackle complex real-world analytics tasks. Specifically, our innovative approach includes a scalable synthetic data generation method that enables the creation of a broad spectrum of topics centered on data representation and manipulation. Furthermore, we introduce two new tasks that seamlessly bridge tables and text. We show that such tasks can enhance models' understanding of schema creation and the nuanced translation between natural language and tabular data. Leveraging this data recipe, we post-train a new foundation model, named CoddLLM, based on Mistral-NeMo-12B. To assess the language understanding and reasoning capabilities of LLMs in the realm of data analytics, we contribute AnalyticsMMLU, a benchmark containing thousands of multiple-choice questions on databases, data analysis, and machine learning. Our focus on data discovery, has resulted in the contribution of three comprehensive benchmarks that address both database and data lake scenarios. CoddLLM not only excels in performance but also sets a new standard, achieving the highest average accuracy across eight datasets. It outperforms GPT-3.5-Turbo on AnalyticsMMLU, exceeding GPT-4o by 12.1% in table selection and showing an average improvement of 24.9% in Text-to-SQL compared to the base model.

大模型数据分析Text-to-SQL基准测试

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。