arXiv:2506.10974cs.CLcs.AI2025-06被引 21

AutoMind让AI自动完成数据科学任务,更智能、更灵活。

AutoMind: Adaptive Knowledgeable Agent for Automated Data Science

  • 构建专家知识库+树状搜索策略,让AI像人一样思考。
  • 在两个基准测试中表现超越现有最佳模型。
  • 适合需要复杂推理的自动化数据科学场景。

大语言模型(LLM)代理在解决实际数据科学问题方面展现出巨大潜力。基于LLM的数据科学代理有望自动化整个机器学习流程,但其真实效果仍受限。现有框架依赖僵化的预定义工作流和固定的编码策略,仅在相对简单、经典的问题上表现良好,难以捕捉人类从业者在复杂、创新任务中的经验知识。本文提出AutoMind,一种自适应、具备专业知识的LLM代理框架,通过三大改进克服上述缺陷:(1) 构建经过筛选的专家知识库,使代理具备领域专业知识;(2) 设计代理式知识树搜索算法,战略性探索解决方案;(3) 提出自适应编码策略,根据任务复杂度动态调整代码生成。在两个自动化数据科学基准上的评估表明,AutoMind性能优于现有最先进基线。额外分析证实其在有效性、效率和解的质量方面表现优异,标志着向完全自动化数据科学迈出了高效且稳健的一步。代码已公开于 https://github.com/innovatingAI/AutoMind。

原文摘要 · Abstract (English)

Large Language Model (LLM) agents have shown great potential in addressing real-world data science problems. LLM-driven data science agents promise to automate the entire machine learning pipeline, yet their real-world effectiveness remains limited. Existing frameworks depend on rigid, pre-defined workflows and inflexible coding strategies; consequently, they excel only on relatively simple, classical problems and fail to capture the empirical expertise that human practitioners bring to complex, innovative tasks. In this work, we introduce AutoMind, an adaptive, knowledgeable LLM-agent framework that overcomes these deficiencies through three key advances: (1) a curated expert knowledge base that grounds the agent in domain expert knowledge, (2) an agentic knowledgeable tree search algorithm that strategically explores possible solutions, and (3) a self-adaptive coding strategy that dynamically tailors code generation to task complexity. Evaluations on two automated data science benchmarks demonstrate that AutoMind delivers superior performance versus state-of-the-art baselines. Additional analyses confirm favorable effectiveness, efficiency, and qualitative solution quality, highlighting AutoMind as an efficient and robust step toward fully automated data science. Code is at https://github.com/innovatingAI/AutoMind.

自动化数据科学LLM代理知识增强

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