Before applying data analytics or machine learning to a data set, a vital step is usually the construction of an informative set of features from the data. In this paper, we present SMARTFEAT, an efficient automated feature engineering tool to assist data users, even non-experts, in constructing useful features. Leveraging the power of Foundation Models (FMs), our approach enables the creation of new features from the data, based on contextual information and open-world knowledge. Our method incorporates an intelligent operator selector that discerns a subset of operators, effectively avoiding exhaustive combinations of original features, as is typically observed in traditional automated feature engineering tools. Moreover, we address the limitations of performing data tasks through row-level interactions with FMs, which could lead to significant delays and costs due to excessive API calls. We introduce a function generator that facilitates the acquisition of efficient data transformations, such as dataframe built-in methods or lambda functions, ensuring the applicability of SMARTFEAT to generate new features for large datasets. Code repo with prompt details and datasets: (https://github.com/niceIrene/SMARTFEAT).
翻译:在对数据集应用数据分析或机器学习之前,关键步骤通常是从数据中构建一组信息丰富的特征。本文提出SMARTFEAT,一种高效的自动化特征工程工具,旨在帮助数据用户(包括非专家)构建有用的特征。我们的方法利用基础模型(FMs)的能力,基于上下文信息和开放世界知识从数据中创建新特征。该方法包含一个智能操作符选择器,可识别操作符子集,从而避免传统自动化特征工程工具中常见的原始特征穷举组合。此外,我们解决了通过行级交互与FMs执行数据任务的局限性——此类交互可能因过多API调用导致显著延迟和成本。我们引入了一个函数生成器,便于获取高效的数据转换(如数据框内置方法或lambda函数),确保SMARTFEAT能够为大型数据集生成新特征。代码仓库包含提示词详情和数据集:(https://github.com/niceIrene/SMARTFEAT)。