arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2607.08093cs.AIcs.CLcs.LG

因果DS:数据科学智能体中的因果推理基准测试

CausalDS: Benchmarking Causal Reasoning in Data-Science Agents

Department of Statistics University of Michigan(统计学系密歇根大学)

Andrej Leban, Theo Saarinen, Yuekai Sun

AI总结:

介绍用于评估数据科学智能体因果推理的CausalDS基准,由含观测数据的结构因果模型和合成自然语言故事构成场景,导出跨越Pearl三个层级的任务,联合评估符号因果推理、数据科学等多方面能力。

摘要:

大语言模型越来越多地充当集成数据科学智能体,将抽象推理与先进工具使用相结合。然而,相关基准测试主要分为无现实数据分析的符号因果推理基准或无原则因果数据生成结构的数据分析基准。现有因果评估数据集常限于现有来源的精选示例。我们引入CausalDS,用于评估智能体数据科学工作流程中的因果推理。每个基准实例是一个场景,包括采样结构因果模型及生成的观测数据和基于现实领域的合成自然语言故事。我们可选地将基准组件的构成基于从真实世界数据集获得的经验分布。然后从每个场景导出涵盖Pearl三个层级的任务。该基准联合评估符号因果推理、数据科学、不确定性量化、弃权及工具使用/编码。

英文摘要:

Large language models (LLMs) increasingly act as data-science agents, combining abstract reasoning with advanced tool use. Yet the relevant benchmark landscape largely divides into symbolic causal reasoning benchmarks without realistic data analysis or data analysis benchmarks without causal data-generating structure or a range of causal data science tasks. Existing causal evaluation datasets are often restricted to curated examples from existing sources, with diversity coming from limited templatized variations rather than from systematic generation of novel synthetic causal structures. We introduce CausalDS, a benchmark for evaluating causal reasoning in agentic data-science workflows. Each benchmark instance is a scene consisting of a sampled structural causal model (SCM) with generated observational data and an accompanying synthetic natural-language story grounded in a real domain. We ground the composition of the benchmark components in empirical distributions obtained from real-world datasets, thus retaining empirical structure while reducing the memorization risk through completely synthetic generation. From each scene, we then derive tasks spanning all three of Pearl's rungs. Most tasks include a data science coding component, where the model typically needs to use several tools to arrive at the final answer due to imperfect observations. Abstention is tested directly: for targets the graph does not identify, the graded answer is to abstain. The benchmark thus jointly evaluates symbolic causal reasoning, data science, uncertainty quantification, abstention, and tool use/coding. Because the generator is fully synthetic, the exam is not a fixed dataset: composition, difficulty, and size are parameters, and fresh exams can be drawn at will.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑