AgenticDataBench: A Comprehensive Benchmark for Data Agents


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 \(\textit{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 (e.g., “Handling Missing Data”), 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.

1 Introduction↩︎

Data science aims to extract actionable insights from heterogeneous raw data, which plays a central role in realizing the value of massive data generated in modern IT and business [1]. Traditionally, data scientists expend substantial effort on understanding and processing poorly organized data, incorporating implicit domain knowledge, and iteratively implementing complex codes.

Recent advancements of large language models (LLMs) have demonstrated superiority in data science–related tasks such as planing [2][5], reasoning [6][8], database operations [9][19], and code generation [20], [21], leading to the emergence of data agents that automate end-to-end insight extraction from raw data with minimal human intervention [22][30].

Here we present a simplified workflow of data agents (see 1a). \((i)\) Planning. Given a complex data science task (e.g., predicting loan delinquency from monthly user statistics with AUC evaluation), the agent interprets user instructions and grounds them in relevant data sources. The challenge lies in instruction ambiguity (e.g., whether missing values should be filled with -1 instead of being pre-filled), heterogeneous data schemas, and large-scale datasets (e.g., “input.csv”) that necessitate iterative exploration. \((ii)\) Iterative Execution. The agent iteratively plans actions, generates executable code, and interacts with execution environments (e.g., Python, databases). It progressively constructs an executable reasoning chain from intermediate results, such as adjusting feature processing or selecting more efficient algorithms under time constraints. \((iii)\) Termination. The process terminates upon either successful completion or reaching predefined step or time limits.

Figure 1: Agentic Data Science Benchmark Example.

While numerous data agents have been proposed, a comprehensive evaluation framework for their systematic comparison is still lacking. Drawing from real-world data science practices, we identify that an effective benchmark should feature realistic tasks spanning diverse domains, accompanied by fine-grained ground-truth labels, enabling the evaluation to capture both the diversity and complexity of data science workflows as well as the detailed performance of agents. However, as shown in 1, existing benchmarks fall short of meeting these criteria. They often rely on a limited set of manually selected task types, overlook the complexities of real-world business applications, and provide only coarse-grained task categories and aggregate scores, which obscure step-level behaviors.

Table 1: vs Existing Data Agent Benchmarks (– indicates no solution code).
Benchmark # Skills Covered # Tags of Task # Lines of Code Data Source Data Modality Data (MB) Per Task
DSBench [31] 2 Public Competition (Semi-)Structured, Text 11.1
BLADE [32] 51 2 16.3 Public Study Structured 2.4
DA-Code [33] 10 85 [33] Public Dataset (Semi-)Structured, Markup, Text, Binary, Database 23.0
DataSciBench [34] 281 6 33.4 Public Dataset (Semi-)Structured, Text, Binary 0.8
ScienceAgentBench [35] 220 11 40.1 Public Study (Semi-)Structured, Markup, Text, Binary 54.3
KramaBench [36] 194 9 34.4 Public Study (Semi-)Structured, Markup, Text, Binary 15.8
433 433 113.6
Public Dataset
Markup, Text, Binary,
Script, Database 493.4

To address this gap, we propose a systematic pipeline for building a comprehensive data agent benchmark (see 1b). Our approach begins by collecting real datasets and tasks from 15 vertical domains, including 5 real-world B2B practices from a leading fintech company [37]. These tasks involve complex scenarios with large-scale noisy data and long code implementations. However, these raw tasks are not directly suitable for benchmarking due to \((i)\) redundancy caused by repeated patterns with minor variations (e.g., consistently filling missing values with -1), and \((ii)\) the lack of high-quality tasks for certain domain datasets. To address this, we abstract recurring data-processing patterns shared across tasks as data science skills (e.g., “Handling Missing Data” in 1), and quantify benchmark coverage based on the number of skills included. From extensive task solutions, we derive a representative skill set (see 2) and select tasks that maximize skill diversity. For generating tasks in uncovered domain datasets, we ensure benchmark quality by \((i)\) sampling realistic skill compositions, \((ii)\) promoting skill diversity across tasks, and \((iii)\) achieving comprehensive coverage of the extracted skills. Additionally, skill annotations provide the foundation for fine-grained analysis of data agent performance.

There are three main challenges. C1: Discovery of Highly Representative Skills. It is non-trivial to extract data science skills from large task collections with ensured diverse representation [38], i.e., a relatively small set of skills that represent data science operations in solving these tasks. C2: Task Selection for Collected Real-world Tasks. For the collected real-world tasks, we aim to select a highly representative set of tasks that capture diverse workload patterns and scenarios while minimizing redundancy to ensure benchmarking efficiency. C3: Realistic Task Generation for Public Datasets. For public datasets that lack predefined tasks, it is essential to systematically generate realistic tasks while ensuring comprehensive coverage of underrepresented skills.

To tackle these challenges, we introduce \(\textit{AgenticDataBench}\), a comprehensive data agent benchmark built on the foundation of data-driven, discovered data science skills. First, we extract representative skills from large-scale task solutions from Stack Overflow [39] through skill-aligned hierarchical clustering. Specifically, we leverage LLMs to break down task solutions into stepwise skill descriptions. To eliminate redundancy, we cluster semantically similar skills using pretrained text embeddings and refine each cluster through LLM-based splitting to identify distinct higher-level skills. This cluster-and-refine process is applied recursively, producing a representative set of high-level skills (addressing C1). Next, for real-world business tasks within each domain, we select task-solution pairs that maximize diversity in skill compositions, ensuring coverage across a wide range of practical scenarios (addressing C2). Finally, to ensure the benchmark comprehensively represents the extracted skills, we propose a systematic LLM-based task generation approach. This method samples frequency-aware skill compositions, uses structured dataset profiles, and generates corresponding workflows and tasks based on these skills (addressing C3).

In summary, we make the following contributions:

(1) We propose a data science skill framework (see Section 2) and subsequently develop a comprehensive data agent benchmark, \(\textit{AgenticDataBench}\), characterized by fine-grained skill composition and real-world complexity (see Section 3). We open-source the testbed at https://github.com/AgenticDataBench/AgenticDataBench.

(2) We propose a hierarchical skill extraction algorithm, which performs agglomerative clustering aligned with skill boundaries using LLM-based semantic refinement (see Section 4).

(3) We propose task selection and generation modules with controlled skill coverage, including selecting skill-diverse real-world tasks and generating realistic tasks that simulate practical skill compositions (see Section 5).

(4) We have conducted an in-depth fine-grained empirical study of state-of-the-art data agents, uncovering four key insights (see Section 6).

Figure 2: 433 Skills Generated by \textit{AgenticDataBench}.

2 Preliminaries↩︎

2.1 Data Science Benchmark↩︎

Solving a data science task typically involves a sequence of data-related operations. We identify and summarize recurring operation patterns into skills, which represent higher-level capabilities characterized by similar application stages, technology stacks, or systematic objectives. For example, data preprocessing can be viewed as a high-level skill that encompasses several fine-grained skills such as missing data handling and feature engineering. Together, these skills provide a multi-faceted characterization of real-world data science workflows.

Definition 1 (Data Science Skill). A data science skill, \(s\), is defined as a data-centric operational pattern commonly used to solve data science tasks. Formally, skills are structured in a hierarchical tree, where each skill node \(s\) includes a textual description \(\delta_s\) and is linked to its child skills, which represent more fine-grained capabilities. The parent-child relationships within the tree reflect abstraction and specialization among skills, with higher-level nodes representing broader, more general operations, while leaf nodes correspond to specific, actionable skill patterns.

For instance, 1a presents some examples of data science skills. “Handling Missing Data” involves identifying NULL values (e.g., incomplete records or artifacts from prior processing), and applying appropriate strategies such as imputation, removal, or transformation to ensure data consistency. We will discuss the scope of data science skills in detail in Section 2.2.

Recently, data agents powered by LLMs have been introduced to automate the entire pipeline of data science tasks, from organization to execution. To comprehensively evaluate such agents, we propose a benchmark designed to ensure broad coverage of data science skills. Each benchmark instance consists of a data science task necessitating specific skills for its resolution, along with a ground-truth solution and an evaluation function.

Definition 2 (Data Science Benchmark Instance). A data science benchmark instance is represented as a quintuple \((\delta_t,D,y,S,eval)\), where \(\delta_t\) is a textual task objective description, \(D\) is the dataset required to solve the task, \(y\) is the executable task solution, \(S\) is the set of skills required to solve the task as reflected in the solution \(y\), and \(eval\) is an evaluation function that maps the output of the data agent to a scalar score in \([0,1]\), quantifying its performance.

For instance, 1a presents a representative data science benchmark. The task description specifies the required statistical computations, including rules for handling missing data and the output format (e.g., CSV). The dataset consists of user loan behavior records in a wide-format CSV file. The solution is a complete implementation that satisfies both correctness and efficiency requirements. The associated skills capture key competencies (e.g., “Handling Missing Data”, “Efficient Data Structures and Algorithms”) and enable fine-grained analysis of agent failures. The evaluation function compares monthly, user-level metrics between agent outputs and the ground truth using a normalized mean squared error. This yields a final score for ranking agents and supporting skill-level analysis (Section 3.3).

2.2 Data Science Skill↩︎

In this section, we discuss the categorical scope of data science skills, and clarify how they differ from the notion of Agent Skills.

Data science skills that underpin task solutions can be categorized into seven exclusive categories according to stages of the data science workflow (see 2): \((i)\) Data Format Handling, including data parsing and file handling; \((ii)\) Data Preprocessing, including data cleaning, transformation, validation, and feature engineering; \((iii)\) Data Manipulation, including restructuring, indexing, filtering, modifying, and merging data; \((iv)\) Data Analysis, including pattern exploration, statistical computation, and aggregation for data insights; \((v)\) Data Modeling, including design, training, evaluation, and deployment of statistical and machine learning models; \((vi)\) Data Visualization, including creation of charts and other visual data representations; \((vii)\) Cross-Stage Skills are general-purpose skills applicable across multiple stages, such as environment management, error handling, SQL & database, code optimization.

Some works implement agent skills as reusable modules that extend LLM capabilities [40], [41], such as guidance, knowledge, scripts, and examples, dynamically incorporated to enhance scenario-specific actions. Others theoretically conceptualize skills as atomic units underlying LLM performance, analyzing outcomes at the skill level [42][48]. Unlike these, we focus on data science scenarios, proposing a representative, data-driven skill set as a quantifiable foundation for benchmarking data agents, as elaborated in Section 4.

Figure 3: The Workflow of Constructing \textit{AgenticDataBench}.

3 Benchmark Overview↩︎

3.1 Design Goals↩︎

We design \(\textit{AgenticDataBench}\) by following the four benchmark design criteria proposed by Jim Gray  [49].

The benchmark covers a wide range of real datasets and data science task patterns. First, we collect 97 real datasets spanning 15 domains from various sources, including 46 Kaggle datasets [50], 2 UCI ML datasets [51], 2 Mendeley datasets [52], and 8 academic and government datasets (UCSD [53], BIRD [54], NatEarth [55], 2 from NYC TLC [56], U.S. BTS [57], NCI GDC [58], OWID [59]), and 39 real business applications at Ant Group. Next, we abstract recurring data-centric operational patterns as data science skills, extracting 433 representative skills from 6,510 high-quality Stack Overflow data science task solutions. Based on these skills, we generate 344 benchmark tasks that collectively cover all identified skills while simulating realistic skill compositions and usage patterns.

The benchmark is designed to reduce task redundancy and enhance clarity. We introduce data science skills to capture core operational patterns in task solutions, and construct the benchmark by \((i)\) selecting 102 real-world tasks from Ant Group with maximal skill diversity and \((ii)\) generating 242 additional tasks with controlled skill coverage. Each task and its corresponding solution are independently annotated by data science experts with the skills required, allowing for a detailed analysis of the data agent’s strengths and weaknesses at the skill level.

The benchmark includes large-scale datasets and diverse tasks of realistic complexity, requiring over 1,560 person-hours of careful construction and labeling. First, the datasets span 15 domains, totaling over 27.3 GB of data across 18 file formats, with 123.1M rows and 35.0K attributes (real business data: 5 domains, 20.1 GB, 7 file formats, 59.4M rows, 4.4K attributes). Next, as shown in 1, the benchmark contains 344 realistically complex data science tasks covering 433 skills, with an average of 23.5 skills per task, 113.6 lines of solution code, and 493.4 MB of data per task.

The benchmark is compatible with a wide range of data agent systems that accept natural language task descriptions and support data file input or manipulation.

3.2 Benchmark Methodology Overview↩︎

Based on these design goals, we construct the data agent benchmark using a systematic creation methodology. First, we introduce a data science skill framework to guide benchmark development, capturing recurring task patterns and enabling skill-level diversity and coverage measurement (see Section 2). Next, we hierarchically extract representative skills from large-scale task solutions (see Section 4). To evaluate the practical efficacy of data agents in industrial-grade scenarios, we collect real-world business datasets and tasks from a leading fintech company and reduce redundancy by selecting skill-diverse representative tasks (see Section 5.1). Finally, to generate realistic tasks for other domains without predefined tasks and ensure comprehensive skill coverage, we generate tasks with realistic skill compositions that cover underrepresented areas (see Section 5.2).

A vanilla approach extracts data science skills from task solutions using LLMs by decomposing step-by-step rationales and summarizing stepwise skills [48]. However, such approaches often produce a large number of loosely defined skills with redundancy and entanglement. To address this issue, we propose a hierarchical skill extraction method that further clusters related skills and abstracts higher-level skills, yielding a more compact and representative skill hierarchy. Specifically, we first decompose task solutions into stepwise skill usage descriptions using vanilla LLM-based approach. Next, we cluster stepwise skills using pretrained text embeddings. As embeddings may capture semantic details irrelevant to skill abstraction, we further prompt LLM to refine each cluster by splitting it into subclusters aligned with higher-level skill boundaries and merging redundant skills. Then, to enhance representativeness, we recursively apply this cluster-and-refine procedure to the resulting skills to derive higher-level skills. The process continues until the number of skills falls below a predefined threshold. Finally, we manually refine the skill hierarchy to ensure quality.

This stage builds realistic benchmark instances across 15 domains of public datasets and real business applications, selecting skill-diverse tasks from collected data and generating realistic tasks for public datasets, collectively covering all extracted skills. Each instance comprises a task description, dataset, ground-truth solution, expert-annotated skill usage, and a task-specific evaluation method.

(1) Skill-Diverse Task Selection. We select real-world task-solution pairs as benchmark instances by maximizing skill diversity to ensure diverse task patterns. Specifically, given the extracted representative skills, we first prompt LLM to annotate each task with its relevant skills. Since selecting a subset of tasks under a fixed budget to maximize skill coverage is an NP-hard problem [60], we adopt a greedy approximation strategy that iteratively selects the task covering the largest number of previously uncovered skills. Finally, we manually refine the selected tasks and solutions to form complete and suitable benchmark instances.

(2) Skill Coverage-Driven Task Generation. To ensure comprehensive coverage of representative skills, we propose an LLM-based pipeline to generate practical tasks under controlled skill compositions. Specifically, we first construct a skill graph by merging skill application traces extracted from task solutions extracted from Stack Overflow and real practices, with frequency-based weights. Next, we sample skill compositions and few-shot task–solution examples. Conditioned on them, we generate structured data profiles that model data formats and cross-dataset relationships, synthesize a skill-based workflow, and produce the corresponding task description. To encourage diversity, we apply penalties to previously covered skills and few-shot examples during the sampling process. Finally, we manually refine the task, and annotate its solution code and evaluation method to form a complete benchmark instance.

In the remainder of this section, we present the details about target test systems and evaluation pipeline of \(\textit{AgenticDataBench}\), and leave the details about hierarchical skill extraction and skill-based benchmark creation in Sections 4 and 5, respectively.

Figure 4: The Overview of \textit{AgenticDataBench} Pipeline.

3.3 \(\textit{AgenticDataBench}\) Pipeline↩︎

As shown in 4, \(\textit{AgenticDataBench}\) tests data agents through four components: Benchmark Instances, Data Agent, Executor, and Leaderboard. First, we load Benchmark Instances, each of which includes a data science task, dataset, the ground-truth solution and answer, required skills, and an evaluation function for scoring task performance. Second, the Executor is prepared within a Docker image, supporting Bash, Python, and database operations, and the benchmark dataset is loaded for exploration and execution. Third, the task is passed to the LLM-driven Data Agent, which iteratively generates and executes code in the Executor based on previous execution responses, and produces a final solution and output. The Leaderboard evaluates the solutions generated by the data agent, compares them against the ground truth, and ranks the data agent on the leaderboard based on its evaluation score. The evaluation process is conducted in two steps. (1) The evaluation function produces a performance score. \(\textit{AgenticDataBench}\) supports five scoring modes: \((i)\) table matching, which checks equality between the predicted and ground-truth tables, allowing tolerance thresholds for numerical columns and exact matching for others; \((ii)\) modeling-based scoring, e.g., mean squared error, normalized to \([0,1]\); \((iii)\) JSON matching, i.e., the proportion of keys with matched values (approximate matching for numerical fields within thresholds, exact matching otherwise); \((iv)\) chart matching, which compares both underlying numerical data and plot configurations; \((v)\) exact and fuzzy text matching. (2) We conduct skill-level analysis by prompting LLMs to compare the solution with the ground truth, using annotated skills as candidates and scoring information to identify incorrectly applied skills as root causes of performance gaps.

4 Hierarchical Skill Extraction↩︎

We discuss how to extract a representative set of data science skills from a large corpus of data science task solutions. These skills comprehensively capture recurring data-related operational patterns across the solutions. Building on this skill framework, we construct benchmark data science tasks with controlled coverage and divergence among selected and newly generated tasks.

We first collect 6,510 data science tasks and solutions from Stack Overflow [39], filtering by \((i)\) relevant tags such as “data-science” and “data-analysis,” and \((ii)\) quality indicators, including accepted answers or scores higher than 3. Next, since many solutions involve complex pipelines that require multiple data science skills, we employ LLM to decompose each solution into stepwise rationales of skill usage [48]. Specifically, we prompt LLM to ensure that each step corresponds to a distinct data science skill while preserving actionable details, and that the collection of steps collectively reconstructs the original solution. In total, this process yields 29,602 stepwise skill descriptions.

However, these skills are unsuitable as a basis for benchmark creation due to three main drawbacks. \((i)\) Scalability. Tens of thousands of loosely defined skills hinder the construction of an efficient and effective benchmark at a manageable scale (see 1). \((ii)\) Redundancy. A common issue is that many skill descriptions refer to the same underlying skill, reducing the diversity of the skill set. \((iii)\) Entanglement. We also observe a prevalent entanglement phenomenon among extracted skills, where one skill represents a high-level abstraction that subsumes another. Such skills should not be simultaneously retained as representative.

The above drawbacks can be mitigated by adopting hierarchical clustering, where each cluster corresponds to a smaller set of higher-level skill abstractions. Specifically, during the agglomerative clustering process [61], \((i)\) the number of top-level skills naturally decreases as clusters are progressively merged, and \((ii)\) redundant or entangled skills, which often share similar semantics, are prone to being merged.

First, we adopt a state-of-the-art text embedding model Qwen3-Embedding [62] to encode skill descriptions into vectors. We also apply UMAP [63] to reduce the embedding dimension while preserving local data manifold structures. Next, we apply GMM [64] for soft clustering, allowing skills to be associated with multiple higher-level skills. Since embedding vector similarities may fail to capture shared high-level skill abstractions due to irrelevant details (e.g., formats or topics), we further use LLMs to split each cluster and align it with coherent higher-level skills (see Step 3). To fit LLM context, we constrain the number of skill descriptions in each cluster below a predefined threshold. If a cluster exceeds this threshold, we recursively apply GMM to split it into smaller clusters.

We split each cluster into sub-clusters representing higher-level skill abstractions, yielding a more compact hierarchy than the original low-level skill annotations. Specifically, for each cluster of semantically related skills, we prompt LLMs to derive higher-level skills and group the low-level skills accordingly. We also preserve the lineage between low-level skills and their parent skills.

Since many skills correspond to synonymous skills, we apply DBSCAN [65] to detect and merge them. Specifically, we embed the LLM-generated skill descriptions using Qwen3-Embedding model, and perform clustering with a strict distance threshold to separate semantically divergent skills. We then merge the synonymous skills, and select the shortest skill description as their representative.

If the resulting top-level skills remain too numerous for a manageable benchmark scale, we further repeat the cluster-and-refine process to derive fewer higher-level skills. Specifically, we generate a summary for each skill by augmenting the LLM-generated description with representative solution steps that exhibit the largest average cosine similarity to the other steps using the skill. We recursively cluster these skill summaries until the number of top-level skills falls below a predefined threshold.

We further address entanglement in the skill hierarchy, where an LLM-generated skill is often overly general if it subsumes another skill at the same or a shallower level. Specifically, we extract syntactic tokens from skill descriptions and identify entanglement via token-set subset relations, assuming such containment indicates semantic subsumption. We then replace overly general skills with their more specific children, and update the hierarchy accordingly.

Finally, we engage data science experts to review the top-level skills to ensure they are appropriately scoped, diverse, representative of common data-related operations in practice, and aligned with realistic evaluation scenarios. Through this process, we obtain 433 top-level skills.

5 Skill-based Benchmark Creation↩︎

We describe the construction of \(\textit{AgenticDataBench}\) based on the extracted data science skill set. Each benchmark instance consists of a task description grounded in real-world datasets, a ground-truth solution, annotated skills required to solve the task, and a task-specific evaluation method. To reflect practical data science challenges, we incorporate real-world datasets and tasks across real business domains within a leading fintech company (see Section 5.1). Then, to ensure cross-domain coverage and comprehensive skill coverage, we generate additional tasks over real-world datasets from diverse domains with specific skill compositions (see Section 5.2).

5.1 Skill-Diverse Task Selection↩︎

Given massive corpus of anonymized production data from Ant Group’s B2B ecosystem, 30 domain experts from 5 business units spent over 600 person-hours curating 600 representative and complex tasks from real-world practice. These tasks span diverse industries, including commercial banking, consumer finance, internet finance, insurance, automotive, aviation, mobile manufacturing, and retail. They also cover a wide range of scenarios (e.g., exploratory analysis, modeling, operations), and preserve realistic challenges such as noise, long-tailed distributions, and feature leakage.

Since many tasks exhibit similar operational structures with variations only in parameters or datasets, we propose a skill-diverse task selection method that adopts the skill framework to maximize coverage across diverse task patterns while reducing redundancy. We implement the method in three steps:

For each task, we use LLMs to annotate relevant skills. Specifically, we evaluate the presence of each candidate skill in the task solution independently using the asynchronous batch inference mode of the Bailian platform [66]. Next, we input the solution and the identified skills into LLMs and prompt LLMs to infer skill dependencies and generate a skill usage trace, which is later used to simulate skill composition during task generation (see Section 5.2).

To enhance benchmark efficiency, we select a representative subset of tasks under a predefined budget while maximizing skill coverage. This can be formulated as an NP-hard problem that maximizes a submodular set function, i.e., the number of skills covered by the selected task set. It admits a \(1-1/e\) approximation guarantee via a greedy algorithm [60]. Specifically, we iteratively select tasks, each time choosing the task that covers the largest number of previously uncovered skills, until the selected task set covers all candidate skills.

To curate tasks suitable for benchmarking, human experts design task-specific evaluation functions, review datasets to ensure the absence of privacy concerns, and refine task descriptions and skill annotations. Though this process, we obtain 102 benchmark instances from real-world business practices.

5.2 Skill Coverage-Driven Task Generation↩︎

To enhance benchmark coverage, we further incorporate 58 datasets from popular open repositories spanning 10 previously uncovered domains, including 46 Kaggle datasets, 2 UCI ML datasets, 2 Mendeley datasets, and 8 academic and government datasets (UCSD, BIRD, NatEarth, 2 from NYC TLC, U.S. BTS, NCI GDC, OWID). We select these repositories based on three criteria: \((i)\) real-world relevance to prevalent data science domains; \((ii)\) inherent complexity, including large-scale data, complex file structures, noisy content, and heterogeneous formats; and \((iii)\) flexible cross-file associations, such as overlapping semantic topics or joinable attributes (e.g., time, users, countries).

We generate tasks with realistic skill compositions absent from the collected tasks. Specifically, we design a skill coverage-driven task generation method. First, we construct a skill graph by merging skill application traces from task solutions collected from Stack Overflow and real practices. The node and edge weights reflect real-world frequencies of skills and their dependencies. Next, to generate a practical task, we prepare key ingredients including real-world datasets, sampled paths from the skill graph, and skill-relevant tasks and solutions as references. Then, we employ a systematic LLM-based pipeline to generate the task, including structured data profiling, workflow synthesis using sampled skills, and task construction grounded in the workflow with quality verification. To promote benchmark diversity, we also dynamically reduce sampling weights of previously covered skills and reference examples. Finally, human experts refine the generated tasks to ensure alignment with the skills, and curate corresponding solutions and evaluation methods to produce complete benchmark instances.

To enhance realism, we simulate real skill compositions by building a skill graph based on aggregated skill usage traces from task solutions. Specifically, for Stack Overflow tasks, we trace the extracted skills back to their original solution steps, and derive skill traces from the step sequences within each solution (see Section 4). For real business tasks from Ant Group, we directly utilize the skill traces obtained in Step 1 of Section 5.1. Based on these skill-annotated task solutions, we construct a skill graph where nodes denote skills and edges represent consecutive skill usage in task solutions. Node and edge weights are frequencies of individual skills and ordered skill pairs, respectively.

Before creating a new task, we prepare three necessary ingredients:

\((1)\) Dataset. We load datasets from the target domain.

\((2)\) Skills. We sample a skill composition by drawing a random path from the skill graph. Specifically, the starting node is sampled from skills that appear within the first 10% of steps in some task solutions, with probabilities proportional to node weights. Each subsequent node is sampled from the neighbors of the current node, with probability proportional to a weighted combination of the corresponding edge weight and neighbor node weight. Sampling continues until the path reaches the predefined length.

\((3)\) Examples. We retrieve representative task-solution pairs relevant to the sampled skills to guide task generation. Specifically, first, for each skill, we assign a relevance score to its annotated steps, defined as the average cosine similarity between the step and other steps annotated with the same skill, plus one to ensure non-negativity. Steps not associated with the skill receive a relevance score of zero. Then, given the sampled skills, we compute the relevance score of each task–solution pair by summing the step-wise relevance scores for each skill and aggregating them across all sampled skills. Finally, we sample a predefined number of tasks with probabilities proportional to their aggregated relevance scores, and retain them as examples of skill application.

Leveraging the prepared dataset, sampled skills, and skill-related examples, we use LLM to generate new tasks through three stages:

\((1)\) Structured Data Profiling. We create a data profile for each dataset file to provide structured information to LLMs, consisting of three components: \((i)\) Basic Information, which applies to all data files and includes the file path, number of rows, and sampled initial rows; \((ii)\) Data Format-Specific Structure, represented as a structured dictionary where each key describes a key attribute of the data file. For example, for tabular data, it includes columns, column types, numerical and categorical columns, missing values, delimiters, and detected header rows determined by textual value ratios or LLM-based distinction between metadata and tabular content; \((iii)\) Relationship, which captures potential join relationships among attributes across data files. Specifically, we first programmatically identify attribute names shared across files within the same subfolder. To capture subtler semantic relationships, we prompt LLMs with the data format-specific structure of each file to generate cross-source relationships including fuzzy attribute matches, thematic parallels, and suggested joins. All discovered joins are manually reviewed to ensure the correctness.

\((2)\) Skill-based Workflow Generation. Since the generated task should require the sampled skills for solution, we first synthesize a solution workflow based on these skills and then generate the task accordingly. \((i)\) Initialization. We first sample one skill and its examples, and prompt LLM to generate an actionable step that potentially involves multiple correlated data files conditioned on data profiles. If there are too many data files, we cluster files with similar name patterns (e.g., differing only by indices) or tables within the same directory that share schema. We then provide the clustered file paths and a predefined number of representative data profiles to LLM. \((ii)\) Skill Iteration. For each remaining sampled skill, we iteratively insert it into the workflow by prompting the LLM to generate a step using the skill, determine its position in the workflow, and update step dependencies accordingly. After insertion, we require LLM to verify step actionability and dependency coherence. If verification fails, we retry until a predefined failure threshold is reached. If the threshold is exceeded, the skill is discarded. \((iii)\) Termination. We repeat this process until all skills are either integrated into or excluded from the workflow.

\((3)\) Workflow-based Task Generation. Based on the workflow steps annotated with used skills and data files, we prompt LLM to generate a task description with verification to ensure six quality criteria, including solvability by the workflow, necessity of the skills, conciseness, clarity, actionability, and a verifiable answer. We retain only tasks that pass these verifications.

\((4)\) Dynamic Sampling Penalty. To enhance diversity among the generated tasks, we first penalize repeated skills by dividing the weights of previously covered nodes and edges by one plus their sampling count. We also apply the same penalization to the relevance scores of previously used task–solution pairs.

we establish a systematic annotation pipeline to ensure the quality of generated benchmark instances, including: \((i)\) validating pipeline correctness; \((ii)\) identifying missing or redundant data sources in each step; \((iii)\) refining questions to better evaluate skill application; \((iv)\) assessing question quality in terms of conciseness, clarity, and domain relevance; \((v)\) designing evaluation functions; and \((vi)\) implementing ground-truth solutions. We further conduct multiple rounds of cross-validation to ensure annotation consistency and reliability. This pipeline engages 8 experts and requires 960 person-hours, resulting in 242 benchmark instances derived from real-world public datasets.

6 Experiments↩︎

6.1 Experimental Setup↩︎

All experiments are conducted on a Linux server with 128 GB RAM and a 3.1 GHz CPU. We execute data agents in a Docker environment to ensure safety and consistent evaluation.

Table 2: Statistics Across 15 Domains.
Domain # Files Data (GB) Per Task
Files Data (MB) # Skills
Real Business Financial 6 0.1 6.0 89.4 14.3
Loan Model 121 0.03 33.7 24.6 20.8
Loan Risk 39 16.1 1.1 397.5 16.4
Marketing 4 3.3 4.0 3304.1 18.1
Strategy 4 0.5 1.0 207.7 12.9
Public Dataset Agriculture 14 0.2 3.7 58.1 27.2
Ecommerce 12 3.1 7.2 2602.4 27.7
Energy 9 0.1 5.3 73.8 26.7
Entertainment 14 1.0 6.0 93.7 22.3
Healthcare 15 0.2 5.4 38.9 29.5
Real Estate 48 0.4 7.5 64.1 24.8
Sports 18 0.4 4.0 231.0 28.8
Social Network 8 0.5 3.2 27.3 24.4
Tourism 18 0.04 6.4 37.6 24.5
Transportation 12 1.3 8.9 980.2 27.1
Total 342 27.3 6.4 493.4 23.5
Table 3: Representative (TF-IDF) and Challenging (Score) Skills by Domain (DFH: Data Format Handling, DP: Data Preprocessing, DM: Data Manipulation, DA: Data Analysis, DML: Data Modeling, DV: Data Visualization, CS: Cross-Stage Skills). TF-IDF ranks skills by frequency scaled by \(log(\)total tasks\(/\)skill tasks\()\) (the higher, the more frequent); challenging skills are those with the lowest aggregated LLM-assigned scores across domain-relevant tasks (the lower, the more challenging).
Domain Representative Skill (TF-IDF) Category Challenging Skill (Score) Category
Financial Metadata and Documentation Review (64.48) DFH SQL Optimization and Advanced Usage (0.50) CS
Query Construction and Execution (34.25) CS Data Transformation and Calculation (0.58) DM
Loan Model DataFrame Column Management (6.59) DM Data Comparison and Validation (0.32) DA
Model Training and Customization (4.40) DML Statistical Testing for Feature-Target Evaluation (0.34) DA
Loan Risk Custom Value Replacement and Correction (6.44) DP Data Preprocessing and Column Management (0.30) DP
Helper Functions and Reusable Code (4.83) CS Normalization and Percentile Calculations (0.34) DA
Marketing Model Training and Customization (6.29) DML Performance Metrics and Optimization (0.27) CS
Performance Benchmarking and Evaluation (4.83) CS Computational Frameworks and Libraries (0.30) CS
Strategy Event Tracking and Funnel Analysis (4.03) DA Data Preprocessing and Column Management (0.13) DP
Data Preprocessing and Segmentation (2.20) DP Data Preprocessing and Segmentation (0.20) DP
Agriculture Entity Mapping and Matching (6.44) DA Probability Modeling and Conversion (0.24) DA
Data Exploration and Comparison (5.33) DA Time Series Analysis and Causality (0.25) DA
Ecommerce Downsampling and Resampling (8.06) DA Statistical Analysis and Testing (0.11) DA
String and Categorical Data Handling (8.05) DM Data Analysis and Visualization (0.23) DV
Energy Mapping and Lookup (3.81) DM Normalization and Percentile Calculations (0.28) DA
Reshaping and Aggregation (3.22) DA Statistical Modeling and Uncertainty (0.33) DA
Entertainment Data Extraction from JSON (3.97) DFH Data Categorization & Mapping (0.27) DM
Data Structure and Dictionary Operations (2.64) DA Regression Modeling and Interpretation (0.27) DML
Healthcare Special Data Handling and Padding (12.09) DM Incremental and Comparative Calculations (0.25) DA
ETL and Data Integration (6.91) DP Time Series and Window Analysis (0.28) DA
Real Estate DataFrame Transformation and Reshaping (9.66) DM Correlation Matrix Generation (0.17) DA
Date Adjustment and Alignment (6.10) DM Data Manipulation and Validation (0.23) DM
Sports ETL and Data Integration (10.69) DP Rolling Statistics and Window-Based Signal Processing (0.20) DA
Data Alignment & Merging (7.15) DM Gradient and Derivative Methods (0.26) DM
Social Network Encoding and Format Identification (8.79) DFH Mathematical Foundations and Algorithm Understanding (0.20) CS
ETL and Data Integration (8.17) DP Validation and Verification of Merge Results (0.27) DA
Tourism Excel File Handling and Automation (45.06) DFH Ranking and Top N Logic (0.20) DM
Command-Line and Shell Operations (9.30) CS Ranking and Normalization (0.27) DM
Transportation Compression and Archiving (62.46) DFH Time Series Analysis and Causality (0.18) DA
Time Series Alignment and Matching (36.27) DM Topic Modeling and Evaluation (0.19) DML
Figure 5: Skill Category Distribution across Domain.
Table 4: Scores (%) over . SA=Smolagents, DA=DA-Agent, CC=Claude Code, CX=CodeX. =Qwen3.5-397B-A17B, =Kimi-K2.5, =Claude Sonnet 4.6.
Domain SA () SA () SA () DA () DA () DA () CC () CC () CC () CX () CX () CX ()
Real Business Financial 58.1 61.5 54.9 54.9 65.4 58.6 55.5 64.3 59.1 60.7 66.9 52.9
Loan Model 41.1 42.3 41.2 43.9 43.4 41.6 41.5 42.9 42.5 33.6 40.6 39.7
Loan Risk 72.0 72.9 69.0 70.9 69.7 71.6 74.0 72.7 72.8 58.9 74.0 52.7
Marketing 37.3 43.5 47.5 36.8 33.4 39.4 31.9 35.4 46.2 29.6 32.2 18.9
Strategy 43.1 45.4 45.4 36.3 47.4 50.8 42.9 39.2 48.0 38.2 44.8 34.1
Public Dataset Agriculture 37.9 32.6 40.3 33.3 35.7 31.3 37.1 38.5 37.1 32.9 40.9 21.2
Ecommerce 31.8 27.9 35.8 29.9 28.0 30.4 31.0 30.6 30.0 26.5 36.3 18.1
Energy 63.1 48.9 61.6 59.0 41.3 58.9 54.7 47.1 59.3 51.2 69.0 46.8
Entertainment 69.3 58.9 51.1 63.3 57.2 60.3 46.4 52.0 57.7 29.0 50.7 42.8
Healthcare 33.3 32.6 29.0 26.5 26.4 25.4 32.7 27.8 31.4 28.3 29.9 14.0
Real Estate 22.4 22.9 22.9 19.3 25.0 17.8 14.1 21.0 21.7 18.1 23.3 8.1
Sports 40.6 34.0 39.4 37.1 39.5 43.9 29.0 37.6 42.0 35.4 44.0 25.7
Social Network 59.7 48.6 67.0 58.8 60.1 68.0 45.1 55.3 60.9 59.7 63.3 33.1
Tourism 56.6 53.8 55.9 57.2 55.4 60.4 49.9 49.2 54.5 46.5 56.8 42.0
Transportation 49.9 37.4 46.2 45.8 39.7 44.8 32.7 42.6 45.1 36.3 53.4 34.5
Total 47.1 43.8 46.7 44.4 44.8 46.1 40.9 44.3 46.6 39.9 48.8 31.6

We evaluate state-of-the-art LLMs, including open-source Qwen3.5-397B-A17B, Kimi-K2.5, and the closed-source Claude Sonnet 4.6. We use default temperatures. We evaluate four representative data-agent harnesses: \((i)\) DA-Agent [33], a data science agent equipped with Bash, Python, and SQL execution tools, reactively invoking tools with execution feedback and a moving memory window. We set 80 maximum steps, and retrain the default 15-step history window and 1-minute step-level timeout; \((ii)\) Smolagents [67], a general-purpose ReAct-style agent that iteratively generates and executes code snippets with execution feedback and periodic planning. We cap the number of coding steps at 40 to limit memory growth, and impose a 5-minute per-step timeout to handle unstable Jupyter Kernel Gateway connections; \((iii)\) Claude Code [68] and CodeX [69], two widely used ReAct-style agent harnesses supporting long-horizon planning, environment interaction (e.g., Bash, files, and coding), concurrent execution, and automatic context management. We cap execution time at 60 minutes per task with an adaptive step-level timeout mechanism. We pair each harness with each LLM in a compositional manner.

We include 15 real-world data science domains. Real-world business domains from Ant Group include: \((i)\) Financial, involving cross-table aggregation of fund holdings, returns, and financial metrics; \((ii)\) Loan Model, covering end-to-end credit risk modeling; \((iii)\) Loan Risk, focusing on post-deployment model monitoring and metric-driven analysis; \((iv)\) Marketing, targeting conversion rate prediction across businesses; and \((v)\) Strategy, supporting business decision-making and multi-faceted strategy analysis. Public domains include: \((vi)\) Agriculture, involving correlated agricultural environments, production, and markets; \((vii)\) E-commerce, covering products and user behaviors across major platforms; \((viii)\) Energy, supporting cross-regional and temporal analysis of industrial consumption and energy indicators; \((ix)\) Entertainment, capturing consumption of multimodal entertainment content; \((x)\) Healthcare, comprising heterogeneous clinical, biomedical, and lifestyle data; \((xi)\) Real Estate, integrating housing properties with socio-economic conditions; \((xii)\) Sports, including records of teams, matches, and athletes across events; \((xiii)\) Social Network, reflecting user behaviors and content across different platforms; \((xiv)\) Tourism, enabling cross-country and travel-related analysis; and \((xv)\) Transportation, modeling spatiotemporal mobility patterns in urban systems.

\(\textit{AgenticDataBench}\) consists of a total of 344 tasks and 342 data files, amounting to 27.3 GB. As shown in 2, our carefully curated data pipeline enables \(\textit{AgenticDataBench}\) to capture the complexity of real-world data science workflows, where each task involves, on average, 6.4 data files, 493.4 MB of data, and 23.5 skill applications.

To illustrate skill-level characteristics across domains, we compute the ratio of skill categories within each domain, as shown in 5. We also identify the most representative and challenging skills per domain, summarized in 3. Representative skills are determined using TF-IDF, calculated as skill frequency scaled by \(\log(\text{domain tasks}/\text{skill tasks})\). Challenging skills are identified by assigning skill application scores via LLM for each task, aggregating them across domain tasks, and selecting the lowest-scoring skills. We find that each domain exhibits distinct skill usage patterns, collectively covering a diverse range of data science task patterns. For example, the Marketing domain emphasizes Data Modeling skills, with prevalent use of “Model Training and Customization” to develop diverse models with rich feature representations for predicting key business indicators (e.g., user payment propensity). 3 further highlights the most challenging skills in this domain, where computing business metrics (e.g., conversion rates) over complex schema and modeling high-dimensional features frequently lead to failures of data agents. In contrast, the Transportation domain emphasizes Data Manipulation skills, focusing on integrating heterogeneous data through transformations (e.g., “Time Series Alignment and Matching”) to support spatiotemporal analysis.

We adopt a two-level evaluation. First, we implement five scoring functions to assess accuracy: table matching, modeling-based scoring, JSON matching, chart matching, and text matching, using Pass@1 by comparing data agent outputs with ground truth, tailored to data format and task type. Second, we perform skill-level analysis by leveraging LLMs to identify misused skills for each task based on annotated skills, revealing failure patterns of data agents. Technical details are provided in Section 3.3.

6.2 Overall Performance Evaluation↩︎

We begin by evaluating each agent’s performance on each dataset, with the overall results summarized in 4.

We start by comparing different agent harnesses, and make two key observations. First, among the evaluated data agents, the top three overall performers are CodeX (Kimi-K2.5), Smolagents (Qwen3.5), and Smolagents (Claude 4.6), suggesting that production-grade agent harnesses currently outperform the data science-specific DA-Agent. This performance gap arises because DA-Agent adopts a lightweight design with limited engineering optimizations (e.g., a fixed memory horizon) and currently lacks specialized components, such as data profiling tools and data science–specific skills. Second, we find that no agent harness achieves the best score across all domains, indicating that different harnesses have distinct domain-specific advantages. For example, Smolagents performs best on the Marketing domain, where many tasks involve large single data files (\(\sim\)​1 GB or more). Smolagents uses a notebook-based implementation that shares loaded data across steps for improved efficiency. In contrast, other data agents often generate separate files at each step, repeatedly reloading the original data during exploration and execution, which can cause timeouts and sub-optimal performance (see also 9b). Besides, DA-Agent can also perform best on the Real Estate domain. This domain is challenging due to multi-source data alignment and complex metric calculations. In this setting, DA-Agent generates large code blocks and achieves relatively high (though still low) scores, whereas other agents iteratively generate small code pieces, leading to inconsistent outputs.

Table 5: Average Metrics per Trajectory by Data Agent. SA=Smolagents, DA=DA-Agent, CC=Claude Code, CX=CodeX. =Qwen3.5-397B-A17B, =Kimi-K2.5, =Claude Sonnet 4.6.
Data Agent # Steps Tokens (K) Cost ($) Success Steps (%) Finish (%)
SA () 18.1 319.4 0.07 94.6 100.0
SA () 21.2 379.4 0.26 88.1 100.0
SA () 24.0 493.5 1.66 97.0 99.7
DA () 20.2 263.0 0.06 95.5 97.4
DA () 17.1 145.4 0.11 94.1 98.8
DA () 13.6 123.7 0.52 91.9 94.8
CC () 28.1 683.4 0.12 93.1 99.7
CC () 23.9 530.5 0.32 93.9 99.7
CC () 16.8 408.3 0.47 96.7 99.7
CX () 26.1 513.2 0.10 91.1 96.8
CX () 40.2 1091.2 0.19 59.5 99.7
CX () 20.0 218.7 0.76 92.2 88.1

Next, we compare different LLMs within the same agent harness. We find that the LLM achieving the best score varies across the four evaluated agent harnesses. \((i)\) Claude 4.6 performs best within DA-Agent and Claude Code. This advantage is mainly due to the superior coding capabilities of Claude 4.6, which result in fewer syntax errors and more efficient, instruction-following code implementations. For example, Claude 4.6 is able to read Parquet files with specified columns, avoiding large-scale data loading and reducing the risk of timeouts. \((ii)\) Qwen3.5 performs best within Smolagents. This is because Kimi-K2.5 and Claude 4.6 are less adaptable to the prompts of Smolagents, causing them to sometimes overlook parts of the instructions (e.g., wrapping code within <code> and </code>), which leads to repeated parsing errors. \((iii)\) Kimi-K2.5 performs best within CodeX. This is because both Qwen3.5 and Claude 4.6 are ill-suited to CodeX: Qwen3.5 tends to generate responses misaligned with CodeX(e.g., invalid function parameter errors, large blocks of inefficient code), while Claude 4.6 frequently stops early without producing a task solution and fails to benefit from auto-compaction of memory (e.g., when the LLM context overflows).

We also record trajectory-level statistics, including token consumption and execution efficiency, as shown in 5. We make two observations. We have two observations. First, although CodeX (Kimi-K2.5) achieves the highest overall score, it exhibits the largest number of execution steps and the lowest successful-step ratio. This is because CodeX aggressively explores multiple solution paths and relies on rapid execution feedback to iteratively refine its trajectory. Interestingly, although CodeX (Kimi-K2.5) uses more tokens than Smolagents (Kimi-K2.5), its cost is lower. This is because CodeX is more input-heavy and output-efficient, while output tokens are substantially more expensive than input tokens for Kimi-K2.5. Second, although Claude 4.6 achieves the best performance within both DA-Agent and Claude Code, it is consistently more expensive than the open-source alternatives. We also observe that Claude Code appears particularly well optimized for Claude models, yielding higher token efficiency and cache utilization. For example, Claude Code (Claude 4.6) achieves a higher score than Claude Code (Kimi-K2.5) (46.6 vs 43.3) while costing only 1.5\(\times\) more, compared to 4-6x in other harnesses.

We also visualize the trade-off between task score and token cost in 6. We make two observations. First, Smolagents (Qwen3.5) and DA-Agent (Qwen3.5) achieve favorable cost-performance trade-offs, attaining moderate task scores at the lowest token costs. This is largely because Qwen3.5 is the least expensive evaluated LLM, while both Smolagents and DA-Agent adopt relatively lightweight orchestrations. Second, Smolagents (Claude 4.6) and CodeX (Claude 4.6) exhibit comparatively poor cost-performance trade-offs, incurring substantially higher token costs without proportional score improvements. This observation is consistent with the LLM–harness mismatch discussed above.

Figure 6: Trade-off between Cost and Score.
Figure 7: Skill Score Comparison across Categories. SA=Smolagents, DA=DA-Agent, CC=Claude Code, CX=CodeX. =Qwen3.5-397B-A17B, =Kimi-K2.5, =Claude Sonnet 4.6.

Finding 1. General-purpose harnesses achieve higher accuracy via mature components (high cost), e.g., CodeX \(>\) Smolagents \(>\) Claude Code \(>\) DA-Agent. Smolagents also benefits from cross-step continuity. Higher accuracy generally comes at higher token costs, whereas lightweight harnesses (Smolagents and DA-Agent) can achieve favorable cost-performance trade-offs.

Finding 2. In terms of token cost, Claude 4.6 incurs much higher token cost than open-source LLMs. In terms of accuracy, the best LLM varies across agent harnesses, emphasizing the adaptivity between LLM and harness (e.g., CodeX (Kimi-K2.5) benefits from more concurrent exploration steps).

Figure 8: Skill-Level Strengths and Weaknesses of the Top-Performing Agent per Harness: CodeX (Kimi-K2.5), Smolagents (Qwen3.5), Claude Code (Claude 4.6), and DA-Agent (Claude 4.6).

6.3 Skill-Level Performance Analysis↩︎

Macro-level aggregate metrics do not reveal fine-grained root causes of data agent performance. To enable deeper analysis of step-wise behaviors, we utilize skill annotations for each \(\textit{AgenticDataBench}\) task. Since these skills represent common data-related operational patterns, they provide a breakdown of key steps in end-to-end tasks, enabling the evaluation of individual skill applications to uncover failure patterns in agent performance. Specifically, by comparing data agent solutions with ground truth and leveraging feedback from evaluation functions, we use LLMs to score each skill application, where inefficient or incorrect executions receive lower scores aligned with the task evaluation score. For each data agent and skill, we compute the skill score as the average score across all applications, and filter out skills with fewer than three applications to ensure reliability.

For each data agent, we aggregate skill scores to compute the average capability for each skill category, as shown in 7. We have two observations. First, the overall skill scores are broadly consistent with task accuracy scores, supporting the validity of our skill-level evaluation. For example, the relative ranking of LLMs is generally preserved within each agent harness. Second, performance varies substantially across skill categories and data agents. For example, Data Format Handling and Cross-Stage Skills generally achieve higher scores, because they are more closely aligned with common coding tasks encountered during pretraining and require less domain-specific data science knowledge. Besides, CodeX (Qwen3.5) performs particularly poorly on Cross-Stage Skills, frequently producing erroneous shell commands (“Command-Line & Shell Operations”; see also 8) or failing to recover from missing dependencies (“Dependency Management”). These weaknesses partially explain its low performance on the Tourism domain (see also 3).

Figure 9: Data Agent Failure Distribution.

To characterize fine-grained skill profiles, we rank skills by their average score across all agents, weighted by \(1-e^{-\text{skill frequency}/40}\) to emphasize frequent skills. The lowest-scoring skills are Data Alignment & Merging, Histogram Creation and Manipulation, and Text Processing and Cleaning, highlighting common limitations in processing heterogeneous and non-relational data. For individual agents, we rank skills by the weighted deviation from the average score of all other agents. Positive and negative values indicate strengths and weaknesses, respectively, as shown in 8. Due to space constraints, we report only the top-performing agent from each of the four harnesses, ranked by overall task accuracy. We find that different data agents exhibit distinct strengths and weaknesses. For example, all agents using Claude 4.6 perform strongly on “Statistical Modeling and Uncertainty”, suggesting that this capability is primarily determined by the underlying LLM. Claude 4.6 more faithfully follows task instructions and prefers installing and leveraging mature Python packages (e.g., using the t-distribution or ARIMA) over ad hoc implementations. In contrast, all DA-Agent-based agents consistently underperform on “Model Training and Inference”, because its static 1-minute per-step timeout can terminate training on high-dimensional feature tables, forcing agents to resort to simpler models.

To differentiate challenging scenarios across domains, we summarize the lowest-scoring skills for each domain in 3, and make two observations. First, the lowest-scoring skills differ from the representative skills of the domain, focusing on challenging task execution (e.g., “Time Series Analysis and Causality”) rather than data characteristics (e.g., “Compression and Archiving”). Second, both the skill scores and specific skills vary across domains, reflecting differences in complexity (e.g., the relatively easy Financial domain with high scores) and task patterns.

Finding 3. Skill-level strength and weakness varies across agents, domains, and skill categories, partially aligning with task accuracy, with consistent weaknesses in processing heterogeneous and non-relational data overlooked by current harness designs.

6.4 Failure Analysis of Data Agents↩︎

Our empirical analysis of data science tasks identifies ten primary categories of errors in data agent execution: \((i)\) Global Limit Exceeded, where the agent exceeds global constraints on maximum steps or total runtime; \((ii)\) Single-Step Timeout, where a single-step execution exceeds its time limit; \((iii)\) Self-Repair Failure, where the agent fails to resolve errors from execution feedback; and \((iv)\) seven skill-specific error categories corresponding to the seven skill categories. Our analysis results are shown in 9.

We compare failure distributions across data agents in 9a and make two observations. First, Data Analysis accounts for the largest proportion of failures, despite not being the most frequently invoked skill category. These failures often arise from data validation, summarization, and statistical calculation, and may further propagate to downstream stages. Second, high rates of Global Limit Exceeded, Single-Step Timeout and Self-Repair Failure suggest poor adaptation between LLMs and agent harnesses, consistent with Section 6.2. For example, CodeX (Qwen3.5) exhibits a high rate of Self-Repair Failure, frequently producing invalid function arguments or Bash commands with minor syntax errors (e.g., missing quotes or whitespace issues).

We compare failure modes across domains in 9b and find that failure distributions vary substantially with data characteristics and task requirements. For example, the Marketing domain exhibits the largest share of Global Limit Exceeded and Single-Step Timeout failures, primarily due to repeated loading of large-scale data files (\(\sim\)​1 GB). The Healthcare domain shows the highest rate of Self-Repair Failure, as its heterogeneous data formats (e.g., ARFF) frequently trigger file parsing errors. In contrast, the Loan Model domain is most affected by Data Modeling failures, owing to its simple file structure (two wide tables) and stronger reliance on complex feature derivation and modeling procedures, which are particularly challenging for current agents.

We re-run the failed tasks due to global limit exceed and single-step timeout constraints, by increasing the global step/runtime limits and the per-step timeout (sampling up to 10 failed tasks), respectively. We find that \((i)\) tasks exceeding the global limits account for less than 0.6% of all tasks, and \((ii)\) neither intervention significantly improves task scores. Instead, increasing the global limits merely prolongs unproductive execution loops, while increasing the per-step timeout can even mislead agents into less effective reasoning trajectories.

Finding 4. Data Analysis contributes the largest share of failures, whereas Global Limit Exceeded, Single-Step Timeout, and Self-Repair Failure reflect LLM–harness misalignment. Failure modes also vary substantially across domains due to differences in data scale, file structure, and task complexity.

7 Conclusion↩︎

We propose a comprehensive benchmark for evaluating data agents, named \(\textit{AgenticDataBench}\), which covers realistic tasks spanning diverse domains with fine-grained labels. We design a hierarchical skill extraction algorithm that leverages LLM-based semantic refinement to perform agglomerative clustering aligned with skill boundaries. We implement task selection and generation modules to ensure controlled skill coverage, enabling the inclusion of skill-diverse real-world tasks and realistic task simulations. Finally, through an in-depth empirical study of state-of-the-art data agents, we uncover key insights and identify important open problems to inspire future research. Our contributions pave the way for advancing the capabilities of autonomous data-science agents and fostering innovation in dynamic, skill-centered data-science systems.

We thank the following contributors for their support in providing and curating the business datasets used in this work: Yuyang Xia, Ziyu Jiang, Yingqi Gao, Xiongfeng Guo, Siyue Liu, Xinyu Li, Fengqin Wei, Xiaochen Liu, Chenlong Li, Haixia Peng, Minzhi Tang, Wenyi Liu, Mengzhen Zhang, Shan Zhang, Jieyuan Chen, Wenyan Liu, Xiuyun Yu, Fan Gou, Linyi Li, Siyu Lv, Shenkang Gu, and Linqi Li.

References↩︎

[1]
Gartner, Inc., Published May 28, 2025Magic Quadrant for Data Science and Machine Learning Platforms,” Gartner, Inc., May 2025. [Online]. Available: https://www.gartner.com/en/documents/6533902.
[2]
J. Wang and J. Feng, “Unify: An unstructured data analytics system,” in ICDE, 2025, pp. 4662–4674.
[3]
S. Shankar, T. Chambers, T. Shah, A. G. Parameswaran, and E. Wu, “DocETL: Agentic query rewriting and evaluation for complex document processing,” VLDB, vol. 18, no. 9, pp. 3035–3048, 2025.
[4]
R. Hu, Y. Luo, G. Li, S. Wu, and Y. Luo, “OpenSQL: Data-efficient text-to-SQL for open-source LLMs via synthesized intermediate supervision,” Proc. VLDB Endow, 2026.
[5]
S. Shankar, S. Zeighami, and A. Parameswaran, “Task cascades for efficient unstructured data processing,” Proceedings of the ACM on Management of Data, vol. 4, no. 1 (SIGMOD, pp. 1–26, 2026.
[6]
A. Yang et al., “Qwen3 technical report,” arXiv preprint arXiv:2505.09388, 2025.
[7]
D. Guo et al., “Deepseek-r1: Incentivizing reasoning capability in llms via reinforcement learning,” arXiv preprint arXiv:2501.12948, 2025.
[8]
Y. Zhang et al., “Reward-SQL: Boosting text-to-SQL via stepwise execution-aware reasoning and process-supervised rewards,” Proceedings of the ACM on Management of Data, vol. 4, no. 3 (SIGMOD, pp. 1–27, 2026.
[9]
X. Zhou et al., “D-bot: Database diagnosis system using large language models,” VLDB, vol. 17, no. 10, pp. 2514–2527, 2024.
[10]
X. Zhou, Z. Sun, and G. Li, “Db-gpt: Large language model meets database,” Data Science and Engineering, vol. 9, no. 1, pp. 102–111, 2024.
[11]
Z. Sun, X. Zhou, G. Li, X. Yu, J. Feng, and Y. Zhang, “R-bot: An LLM-based query rewrite system,” VLDB, vol. 18, no. 12, pp. 5031–5044, 2025.
[12]
Z. Sun, X. Zhou, J. Wu, W. Zhou, and G. Li, “D-bot: An LLM-powered DBA copilot,” in SIGMOD companion, 2025, pp. 235–238.
[13]
W. Zhou, Y. Gao, X. Zhou, and G. Li, “CrackSQL: A hybrid dialect translation system powered by LLM,” in Companion of the international conference on management of data, 2026, pp. 154–157.
[14]
J. Xu, R. Wang, J. Wang, and G. Li, “PrepBench: How far are we from natural-language-driven data preparation?” arXiv preprint arXiv:2605.08687, 2026.
[15]
S. Xu, J. Wang, and G. Li, “Bridging the gap: Cardinality estimation for semantic queries on unstructured data,” Proceedings of the ACM on Management of Data, vol. 4, no. 3 (SIGMOD, pp. 1–26, 2026.
[16]
X. Lyu, C. Lin, Y. Zheng, Z. Bao, Y. Zhang, and G. Li, “GenIA: Generative index advisor for dynamic workloads and data,” IEEE Transactions on Knowledge and Data Engineering, 2026.
[17]
W. Zhou et al., “Can LLMs clean up your mess? A survey of application-ready data preparation with LLMs,” arXiv preprint arXiv:2601.17058, 2026.
[18]
W. Zhou, Y. Gao, X. Zhou, and G. Li, “Cracking SQL barriers: An llm-based dialect translation system,” Proceedings of the ACM on Management of Data, vol. 3, no. 3, pp. 1–26, 2025.
[19]
W. Zhou et al., “Dbaiops: A reasoning llm-enhanced database operation and maintenance system using knowledge graphs,” arXiv preprint arXiv:2508.01136, 2025.
[20]
A. Zeng et al., “GLM-5: From vibe coding to agentic engineering,” arXiv preprint arXiv:2602.15763, 2026.
[21]
W. Zhou et al., “Automating database-native function code synthesis with LLMs,” CoRR, vol. abs/2604.06231, 2026, doi: 10.48550/ARXIV.2604.06231.
[22]
J. Sun, G. Li, P. Zhou, Y. Ma, J. Xu, and Y. Li, “Agenticdata: An agentic data analytics system for heterogeneous data,” arXiv preprint arXiv:2508.05002, 2025.
[23]
Z. Sun, J. Wang, X. Zhao, J. Wang, and G. Li, “Data agent: A holistic architecture for orchestrating data+ ai ecosystems,” arXiv preprint arXiv:2507.01599, 2025.
[24]
Z. Tang et al., “Workspace-bench 1.0: Benchmarking AI agents on workspace tasks with large-scale file dependencies.” 2026, [Online]. Available: https://arxiv.org/abs/2605.03596.
[25]
W. Zhou et al., “Are we ready for an agent-native memory system?” 2026, [Online]. Available: https://arxiv.org/abs/2606.24775.
[26]
J. Qu et al., “ST-raptor: An agentic system for semi-structured table QA,” arXiv preprint arXiv:2602.07034, 2026.
[27]
H. Lan et al., “AgenticScholar: Agentic data management with pipeline orchestration for scholarly corpora,” Proceedings of the ACM on Management of Data, vol. 4, no. 3 (SIGMOD, pp. 1–28, 2026.
[28]
Y. Luo, G. Li, J. Fan, and N. Tang, “Data agents: Levels, state of the art, and open problems,” in Companion of the international conference on management of data, 2026, pp. 571–579.
[29]
K. Wang et al., “Data agents under attack: Vulnerabilities in LLM-driven analytical systems,” arXiv preprint arXiv:2606.08661, 2026.
[30]
S. Liu et al., “Supporting our AI overlords: Redesigning data systems to be agent-first,” in 16th conference on innovative data systems research, CIDR 2026, chaminade, CA, USA, january 18-21, 2026, 2026, [Online]. Available: https://vldb.org/cidrdb/2026/supporting-our-ai-overlords-redesigning-data-systems-to-be-agent-first.html.
[31]
L. Jing et al., “DSBench: How far are data science agents from becoming data science experts?” in ICLR, 2025.
[32]
K. Gu et al., “BLADE: Benchmarking language model agents for data-driven science,” in EMNLP, 2024, pp. 13936–13971.
[33]
Y. Huang et al., “DA-code: Agent data science code generation benchmark for large language models,” in EMNLP, 2024, pp. 13487–13521.
[34]
D. Zhang et al., “Datascibench: An llm agent benchmark for data science,” arXiv preprint arXiv:2502.13897, 2025.
[35]
Z. Chen et al., “ScienceAgentBench: Toward rigorous assessment of language agents for data-driven scientific discovery,” in ICLR, 2025.
[36]
E. Lai et al., “Kramabench: A benchmark for ai systems on data-to-insight pipelines over data lakes,” arXiv preprint arXiv:2506.06541, 2025.
[37]
“AI-driven eKYC & mobile solutions | ant digital technologies,” 2026. https://antdigital.com/en.
[38]
K. H. R. Chan, Y. Yu, C. You, H. Qi, J. Wright, and Y. Ma, “ReduNet: A white-box deep network from the principle of maximizing rate reduction,” J. Mach. Learn. Res., vol. 23, pp. 114:1–114:103, 2022.
[39]
“Stack overflow - where developers learn, share, & build careers,” 2026. https://stackoverflow.com.
[40]
[41]
A. Didolkar et al., “Metacognitive capabilities of LLMs: An exploration in mathematical problem solving,” in NIPS, 2024.
[42]
S. Arora and A. Goyal, “A theory for emergence of complex skills in language models,” CoRR, vol. abs/2307.15936, 2023, [Online]. Available: https://arxiv.org/abs/2307.15936.
[43]
E. J. Michaud, Z. Liu, U. Girit, and M. Tegmark, “The quantization model of neural scaling,” in NIPS, 2023.
[44]
Z. Liu, Y. Liu, E. J. Michaud, J. Gore, and M. Tegmark, “Physics of skill learning,” CoRR, vol. abs/2501.12391, 2025, [Online]. Available: https://arxiv.org/abs/2501.12391.
[45]
M. F. Chen et al., “Skill-it! A data-driven skills framework for understanding and training language models,” in NIPS, 2023.
[46]
S. Kaur, S. Park, A. Goyal, and S. Arora, “Instruct-SkillMix: A powerful pipeline for LLM instruction tuning,” in ICLR, 2025.
[47]
D. Yu, S. Kaur, A. Gupta, J. Brown-Cohen, A. Goyal, and S. Arora, SKILL-MIX: A flexible and expandable family of evaluations for AI models,” in ICLR, 2024.
[48]
M. Moayeri et al., “Unearthing skill-level insights for understanding trade-offs of foundation models,” in ICLR, 2025.
[49]
J. Gray, Benchmark handbook: For database and transaction processing systems. San Francisco, CA, USA: Morgan Kaufmann Publishers Inc., 1992.
[50]
“Kaggle: Your machine learning and data science community,” 2026. https://www.kaggle.com/.
[51]
“Home - UCI machine learning repository,” 2026. https://archive.ics.uci.edu.
[52]
“Mendeley data,” 2026. https://data.mendeley.com/.
[53]
J. Ni, J. Li, and J. McAuley, “Justifying recommendations using distantly-labeled reviews and fine-grained aspects,” in Proceedings of the 2019 conference on empirical methods in natural language processing and the 9th international joint conference on natural language processing (EMNLP-IJCNLP), 2019, pp. 188–197, [Online]. Available: https://cseweb.ucsd.edu/~jmcauley/datasets/amazon_v2/.
[54]
J. Li et al., “Can llm already serve as a database interface? A big bench for large-scale database grounded text-to-sqls,” Advances in Neural Information Processing Systems, vol. 36, 2024, [Online]. Available: https://bird-bench.github.io/.
[55]
“Natural earth - free vector and raster map data,” 2026. https://www.naturalearthdata.com/downloads/50m-cultural-vectors/.
[56]
“TLC trip record data - new york city taxi and limousine commission,” 2026. https://www.nyc.gov/site/tlc/about/tlc-trip-record-data.page.
[57]
“Bureau of transportation statistics - national transportation atlas database,” 2026. https://geodata.bts.gov/datasets/usdot::means-of-transportation-to-work/about.
[58]
“Genomic data commons - TCGA pan-cancer clinical data resource,” 2026. https://gdc.cancer.gov/about-data/publications/PanCan-Clinical-2018.
[59]
“Our world in data - CO2 and greenhouse gas emissions,” 2026. https://github.com/owid/co2-data.
[60]
G. L. Nemhauser, L. A. Wolsey, and M. L. Fisher, “An analysis of approximations for maximizing submodular set functions—i,” Mathematical programming, vol. 14, no. 1, pp. 265–294, 1978.
[61]
F. Murtagh and P. Contreras, “Algorithms for hierarchical clustering: An overview,” Wiley interdisciplinary reviews: data mining and knowledge discovery, vol. 2, no. 1, pp. 86–97, 2012.
[62]
“Qwen3-embedding - a qwen collection,” 2026. https://huggingface.co/collections/Qwen/qwen3-embedding.
[63]
L. McInnes, J. Healy, and J. Melville, “Umap: Uniform manifold approximation and projection for dimension reduction,” arXiv preprint arXiv:1802.03426, 2018.
[64]
P. Sarthi, S. Abdullah, A. Tuli, S. Khanna, A. Goldie, and C. D. Manning, “RAPTOR: Recursive abstractive processing for tree-organized retrieval,” in ICLR, 2024.
[65]
M. Ester, H.-P. Kriegel, J. Sander, X. Xu, et al., “A density-based algorithm for discovering clusters in large spatial databases with noise,” in SIGKDD, 1996, vol. 96, pp. 226–231.
[66]
“Bailian console of the large model service platform,” 2026. https://bailian.console.alibabacloud.com/.
[67]
“Smolagents: A barebones library for agents that think in code.” 2026. https://github.com/huggingface/smolagents.
[68]
“Overview - claude code docs,” 2026. https://code.claude.com/docs/en/overview.
[69]
“Codex | AI coding partner from OpenAI | OpenAI,” 2026. https://openai.com/codex.