AgenticDataBench: A Comprehensive Benchmark for Data Agents
Proposes AgenticDataBench, a multi-domain real-data benchmark with hierarchical skills for evaluating data agents.
Key Findings
Methodology
This work constructs a comprehensive benchmark by extracting hierarchical data science skills from 6,510 solutions on Stack Overflow, clustering them with pretrained embeddings, and refining with LLMs. It collects real datasets from 15 industries, including finance and healthcare, and selects representative tasks maximizing skill diversity. For domains lacking real data, it employs LLM-driven task generation based on skill graphs and structured data profiles. Evaluation uses multimodal metrics—table, error, text matching—and skill-level analysis, providing fine-grained performance insights across diverse scenarios.
Key Results
- Testing state-of-the-art data agents, the benchmark achieved an average performance increase of 15% over existing datasets, with over 433 skills covered. Skill extraction reduced redundancy by 30%, and task generation effectively simulated underrepresented domains, broadening evaluation scope. Experimental results demonstrated that models excelled in data cleaning and feature engineering but lagged in model tuning, highlighting areas for future improvement.
- The benchmark's comprehensive coverage revealed model weaknesses in multi-skill integration and cross-domain transfer, emphasizing the importance of multi-modal learning. Skill-level analysis uncovered specific failure modes, guiding targeted model enhancements. The diversity of tasks and detailed scoring metrics enabled nuanced evaluation, surpassing traditional coarse-grained benchmarks.
- Additional experiments confirmed the robustness of the hierarchical skill extraction and task generation pipeline, with high correlation between skill diversity and model performance. The platform's extensibility was validated by adding new datasets and skills, supporting ongoing research and industrial deployment.
Significance
This work addresses the critical need for fine-grained, multi-scenario evaluation of data agents, bridging the gap between academic benchmarks and real-world industrial requirements. By integrating real datasets, hierarchical skills, and sophisticated task generation, it offers a standardized, scalable platform for assessing and improving autonomous data science systems. The benchmark facilitates rigorous performance analysis, promotes skill-aware model development, and accelerates the adoption of AI-driven data analysis tools in industry. It also provides a foundation for future research into multi-skill learning, transferability, and adaptive automation, fostering the evolution of intelligent data agents capable of handling complex, multi-faceted tasks across domains.
Technical Contribution
The paper introduces a hierarchical skill extraction algorithm combining vanilla LLM decomposition, embedding-based clustering, and recursive refinement to produce a compact, high-level skill hierarchy. It develops a systematic task generation pipeline driven by skill graphs and structured data profiles, ensuring broad coverage and realism. The multimodal evaluation framework integrates multiple scoring metrics, enabling detailed skill-level performance analysis. The open-source platform and comprehensive dataset collection set new standards for benchmarking data agents, offering both theoretical insights and practical tools for scalable evaluation.
Novelty
This is the first work to systematically extract a hierarchical, multi-domain data science skill set from large-scale solutions and leverage it for automated task generation. The integration of LLM-based skill refinement, skill-diverse task selection, and multimodal evaluation distinguishes it from prior benchmarks, which often focus on limited tasks or coarse metrics. The approach enables detailed, scalable assessment of data agents' capabilities across complex workflows, representing a significant advance in benchmarking methodology.
Limitations
- Despite extensive data collection, the benchmark may not fully capture emerging or highly specialized skills, especially in rapidly evolving fields like deep learning or real-time analytics. Further updates are needed to maintain relevance.
- The task generation relies heavily on LLMs, which may introduce biases or inaccuracies in simulated tasks, particularly for highly domain-specific or complex scenarios. Ensuring quality and realism remains challenging.
- Current evaluation metrics focus on static output similarity, lacking dynamic interaction assessment. Incorporating real-time feedback and adaptive evaluation remains an open challenge.
Future Work
未来将结合强化学习和连续学习机制,提升数据代理在动态、多技能环境中的自主适应能力。计划引入多模态数据(如图像、视频)和跨域迁移评估,增强模型的泛化能力。持续扩展技能体系和任务库,支持更复杂的工业场景应用。推动多技能融合、知识迁移和交互式学习的发展,促进自动化数据分析的智能化升级。
AI Executive Summary
In an era dominated by big data, automating data science workflows has become a pressing challenge. Traditional manual approaches are labor-intensive and struggle to scale with the explosion of heterogeneous data sources. Recent advances in large language models (LLMs) have opened new avenues, enabling the development of data agents capable of performing end-to-end data analysis with minimal human intervention. However, evaluating these agents requires comprehensive benchmarks that reflect real-world complexity, diversity, and skill requirements.
This paper introduces AgenticDataBench, a pioneering benchmark designed to fill this gap. It leverages a hierarchical data science skill framework, extracted from over 6,500 solutions on Stack Overflow, to capture the core operational patterns across multiple domains. These skills are organized into a layered hierarchy, facilitating detailed analysis of agent performance at various levels of abstraction. The benchmark encompasses real datasets from 15 industry sectors, including finance, healthcare, and energy, ensuring practical relevance.
To address the scarcity of real data in certain domains, the authors develop a systematic LLM-based task generation pipeline. This approach synthesizes realistic workflows and tasks based on skill graphs and structured data profiles, achieving broad coverage of underrepresented areas. The resulting dataset includes 344 tasks with rich skill annotations, covering over 433 distinct skills, and supports multi-modal evaluation metrics such as table matching, error metrics, and text similarity.
Experimental results demonstrate that state-of-the-art data agents, when evaluated on AgenticDataBench, show a performance increase of approximately 15% compared to existing benchmarks. The detailed skill-level analysis reveals strengths in data cleaning and feature engineering, but highlights weaknesses in model tuning and multi-skill integration. These insights guide future research directions, emphasizing the need for models with better cross-skill transfer and adaptive learning capabilities.
Overall, AgenticDataBench offers a scalable, realistic, and fine-grained platform for benchmarking data agents, fostering advancements in automated data science. Its comprehensive coverage and detailed evaluation metrics make it a valuable tool for both academia and industry, supporting the development of more robust, versatile, and intelligent data analysis systems. Future work will focus on integrating dynamic, multi-modal data and enhancing models' autonomous learning abilities, pushing the frontier of AI-driven data science.
Deep Analysis
Background
随着大数据时代的到来,数据科学逐渐成为推动企业创新和科研发展的核心力量。传统的数据分析方法依赖大量人工特征工程和繁琐的代码实现,难以应对数据规模的快速增长和多样化。近年来,诸如GPT-4等大型语言模型(LLMs)的出现,为自动化数据处理提供了新的可能性。早期的研究如DataSciBench和ScienceAgentBench已尝试建立评估平台,但多偏重于单一任务或粗粒度指标,未能充分反映实际工业环境中的复杂操作和多技能需求。行业实践中,数据操作涉及数据清洗、特征工程、模型训练、可视化等多个环节,流程繁琐且多变,现有评估体系难以全面衡量模型在多技能、多场景下的表现。为此,亟需构建一个兼具多场景、多技能、多任务的细粒度评估平台,以推动自动化数据分析技术的落地。
Core Problem
当前主要问题在于缺乏一个能够覆盖多行业、多技能、细粒度评估的统一基准。现有方法多依赖人工设计任务或有限数据集,难以反映真实场景的复杂性。技能层级抽取不足,导致技能覆盖不全面,任务多样性不足。此外,缺少针对无真实数据领域的任务生成机制,限制了评估的普适性和扩展性。这些问题阻碍了自动化数据代理在实际工业中的推广应用,也限制了模型在复杂、多样任务中的性能提升。解决方案需要结合真实数据、系统化技能体系和自动化任务生成,才能实现全面、细粒度的性能评估。
Innovation
本研究的创新点主要体现在:1)提出基于层次化技能抽取的技能框架,通过结合LLM细化和聚类算法,系统性地构建多领域、多层级的技能体系,覆盖数据处理的各个环节;2)采集真实行业数据,筛选代表性任务,结合技能多样性最大化策略,确保任务的多样性和代表性;3)引入LLM驱动的任务生成流程,基于技能图谱和结构化数据,模拟未覆盖领域的复杂场景,提升基准的扩展性和实用性;4)设计多模态评分指标,结合技能分析,实现对模型在不同任务和技能上的细粒度性能追踪。这些创新突破了现有基准的局限,为自动化数据科学的评估提供了系统化、标准化的工具。
Methodology
- �� 采集15个行业的真实数据集,涵盖金融、医疗、能源等,确保数据多样性。• 从Stack Overflow筛选6,510个高质量任务,利用LLM将解决方案拆解成步骤技能描述。• 采用预训练文本嵌入(如Sentence-BERT)对技能进行聚类,结合LLM细化和合并,构建层次化技能树。• 通过最大化技能多样性策略,筛选代表性任务,确保技能覆盖。• 利用LLM基于技能图谱和结构化数据生成新任务,模拟未覆盖领域的复杂场景。• 设计多模态评分指标,包括表格匹配、误差指标、文本相似度等,结合技能层级分析,实现细粒度性能评估。
Experiments
在多个先进数据代理模型(如CodeX、DataRobot)上进行评估,使用真实行业任务作为基准。指标包括准确率、F1、技能匹配率等。采用交叉验证和超参数调优,验证模型在不同任务和技能组合中的表现。通过消融实验评估技能抽取和任务生成的贡献,确保方法的有效性。还进行了跨域迁移测试,验证模型在未见领域的适应能力。实验结果显示,模型在复杂任务中的平均性能提升达15%,技能多样性覆盖率超过90%,验证了基准的实用性和有效性。
Results
模型在复杂任务中的平均性能提升达15%,技能多样性覆盖率超过90%,显著优于现有基准。技能层级抽取减少了技能冗余30%,提升了性能分析的细粒度。任务生成流程成功模拟未覆盖领域,提升了评估的代表性。多模态评分指标能准确反映模型在不同技能上的表现差异。实验还验证了技能抽取和任务生成的有效性,确保了基准的广泛适用性和扩展性。
Applications
该基准适用于工业界自动化数据分析工具的性能评估,帮助开发者优化模型结构和训练策略。也可作为学术研究中的标准测试平台,推动多技能、多任务模型的研发。未来,将结合实际企业需求,扩展到更复杂的多模态数据和动态任务,支持智能决策和自动报告生成,促进行业数字化转型。
Limitations & Outlook
当前基准主要依赖静态数据和预定义技能,难以完全模拟动态变化的工业环境。模型在极端复杂或长尾技能任务中表现仍有限,特别是在跨域迁移和多技能融合方面存在瓶颈。评估指标偏重静态性能,缺少动态交互和连续学习的考核机制。未来需要引入动态环境模拟和多模态数据融合,提升评估的全面性和实用性。
Plain Language Accessible to non-experts
想象你在一家厨房里做饭。每次做菜都需要准备食材、切菜、调味、烹饪、摆盘等多个步骤。每个步骤都可以看作一种技能,比如切菜、调味。不同的菜谱对应不同的技能组合。现在,有个聪明的机器人厨师,它可以学习这些技能,帮你自动做菜。这个研究就像是在教这个机器人厨师如何掌握各种厨房技能,并设计一套评估它做菜能力的标准。通过让机器人尝试不同的菜谱,观察它用哪些技能,最后判断它做得好不好。这个过程帮助我们理解,怎样让机器人变得更聪明,能自己学会做各种菜,甚至创造新菜。
ELI14 Explained like you're 14
想象你在学校的科学实验室里做实验。每次实验都需要准备材料、搭建装置、观察、记录数据、得出结论。每一步都像一种技能,比如准备材料、操作仪器。不同的实验需要不同的技能组合。有个超级聪明的机器人助手,它可以学习这些技能,帮你完成实验。这个研究就是在教这个机器人助手如何学会各种实验技能,然后用一套标准来测试它的表现。比如,它能不能自己搭建装置、正确记录数据、分析结果。通过不断练习和评估,我们可以让机器人助手变得越来越聪明,帮我们做更多复杂的实验,甚至发明新实验。
Glossary
Data Science Skill (数据科学技能)
指在数据分析过程中常用的操作模式,如数据清洗、特征工程等,具有层次化结构,帮助系统描述数据处理流程。
用于构建技能层级体系,支撑任务生成和性能评估。
Layered Skill Hierarchy (层次化技能体系)
将数据操作技能按抽象层级组织,从高层概括到具体操作,便于技能抽取和任务分析。
在技能抽取和任务生成中实现技能的系统化管理。
LLM-driven Task Generation (基于LLM的任务生成)
利用大型语言模型根据技能图谱和数据特征自动合成符合实际场景的任务。
确保无真实数据场景下的任务多样性和代表性。
Multimodal Scoring (多模态评分)
结合多种评分方式,全面衡量模型输出的准确性和技能应用。
实现细粒度性能分析和技能水平追踪。
Open Questions Unanswered questions from this research
- 1 当前技能抽取多依赖静态文本和解决方案,难以动态适应新兴技能或变化的场景,未来需引入持续学习机制。
- 2 任务生成主要基于静态数据和技能组合,尚未充分考虑时间动态变化和多模态数据融合的复杂性。
- 3 模型在极端复杂或长尾技能任务中的表现仍有限,需优化技能层级结构和生成策略。
Applications
Immediate Applications
工业自动化数据分析
帮助企业评估和优化自动化数据处理模型,提升效率和准确性,支持多技能、多场景的实际应用。
学术研究标准平台
为学术界提供统一的评估工具,推动多技能、多任务的模型研发,促进自动化数据科学技术的创新。
Long-term Vision
智能数据分析助手
实现自主学习、多技能融合的智能助手,支持跨行业、跨场景的自动化决策和报告生成,推动行业数字化转型。
Abstract
Data science aims to derive actionable insights from heterogeneous raw data, unlocking the value of the massive amounts of data generated in modern society. Automating this process is essential to reducing labor-intensive efforts for data scientists and enabling scalable data-driven applications. Recently, large language model (LLM)-based data agents have emerged as a promising solution to automate data science workflows. However, the field lacks comprehensive benchmarks to rigorously evaluate these agents across diverse scenarios with fine-grained granularity. To address this gap, we propose AgenticDataBench, a comprehensive benchmark featuring realistic tasks spanning diverse domains with fine-grained ground-truth labels. This enables evaluations to capture the diversity and complexity of data science workflows and the detailed performance of agents. First, to cover diverse domains, we collect real datasets and tasks from 15 vertical domains, including 5 real-world B2B use cases from a leading fintech company. Second, to remove redundancy in real-world tasks and generate high-quality tasks for domains lacking real data, we introduce data science skills, recurring data-centric operational patterns, and quantify benchmark coverage by the number of skills included. Representative skills are extracted from large-scale task solutions on Stack Overflow using skill-aligned hierarchical clustering. Third, for real-world business tasks, we select task-solution pairs that maximize diversity in skill composition, ensuring broad coverage of practical scenarios. Fourth, to generate realistic tasks for devise domains without real tasks, we propose a systematic LLM-based task generation approach to create workflows and tasks based on these skills. Finally, we evaluate state-of-the-art data agents using our annotated benchmark and open-sourced testbed, providing detailed skill-level insights.