arXiv:2602.17641cs.LGcs.AI2026-02被引 2

用AI代理自动发现最优特征,提升表格数据建模效果。

FAMOSE: A ReAct Approach to Automated Feature Discovery

  • 基于ReAct框架构建智能代理,迭代生成与评估特征。
  • 分类任务平均提升0.23% ROC-AUC,回归任务平均降低2.0% RMSE。
  • 适合需要创新特征工程的机器学习研究者与实践者。

特征工程仍是机器学习中的关键瓶颈,尤其在表格数据中,从指数级增长的特征空间中识别最优特征传统上需大量领域知识。为此,我们提出FAMOSE(Feature AugMentation and Optimal Selection agEnt),一种基于代理式ReAct范式的新型框架,可自主探索、生成并优化特征,同时将特征选择与评估工具集成于代理架构中。据我们所知,FAMOSE是首个将代理式ReAct框架应用于自动化特征工程的工作,适用于回归与分类任务。大量实验表明,其在分类任务中达到或接近当前最优水平(尤其在实例数超过1万的任务中,平均提升0.23% ROC-AUC);在回归任务中实现最优表现,平均降低2.0% RMSE,且对错误更具鲁棒性。我们推测,其优异性能源于ReAct机制使大型语言模型(LLM)上下文窗口能通过迭代特征发现与评估步骤记录有效/无效特征,类似少样本提示,引导模型发明更优、更具创新性的特征。本工作表明,AI代理在需要高度创造性解决方案的问题中表现卓越。

原文摘要 · Abstract (English)

Feature engineering remains a critical yet challenging bottleneck in machine learning, particularly for tabular data, as identifying optimal features from an exponentially large feature space traditionally demands substantial domain expertise. To address this challenge, we introduce FAMOSE (Feature AugMentation and Optimal Selection agEnt), a novel framework that leverages the ReAct paradigm to autonomously explore, generate, and refine features while integrating feature selection and evaluation tools within an agent architecture. To our knowledge, FAMOSE represents the first application of an agentic ReAct framework to automated feature engineering, especially for both regression and classification tasks. Extensive experiments demonstrate that FAMOSE is at or near the state-of-the-art on classification tasks (especially tasks with more than 10K instances, where ROC-AUC increases 0.23% on average), and achieves the state-of-the-art for regression tasks by reducing RMSE by 2.0% on average, while remaining more robust to errors than other algorithms. We hypothesize that FAMOSE's strong performance is because ReAct allows the LLM context window to record (via iterative feature discovery and evaluation steps) what features did or did not work. This is similar to a few-shot prompt and guides the LLM to invent better, more innovative features. Our work offers evidence that AI agents are remarkably effective in solving problems that require highly inventive solutions, such as feature engineering.

特征工程AI代理自动化表格数据

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