arXiv:2503.23371cs.CLcs.AI2025-03NAACL被引 15

用大模型自动做表格数据特征工程,省人力还更高效。

FeRG-LLM : Feature Engineering by Reason Generation Large Language Models

  • 用对话式思维链让大模型主动发现新特征
  • 80亿参数模型在多数数据集上超越700亿参数模型
  • 可本地部署,避免云端费用与数据泄露风险

表格数据的特征工程是机器学习中的关键任务,但依赖大量人工经验和领域知识,成本高昂。为此,我们提出FeRG-LLM框架,利用80亿参数的大语言模型实现自动化特征工程。通过构建两阶段对话流程,使模型能分析任务并生成新特征,展现其思维链(CoT)能力。我们基于Llama 3.1 8B模型进行微调,并结合直接偏好优化(DPO)以提升特征质量与模型性能。实验表明,FeRG-LLM在多数数据集上表现优于或媲美Llama 3.1 70B,资源消耗更低、推理速度更快;分类任务中领先于现有方法,回归任务亦表现良好。由于无需调用GPT-4等云服务,可本地部署,有效解决安全与成本问题。

原文摘要 · Abstract (English)

One of the key tasks in machine learning for tabular data is feature engineering. Although it is vital for improving the performance of models, it demands considerable human expertise and deep domain knowledge, making it labor-intensive endeavor. To address this issue, we propose a novel framework, \textbf{FeRG-LLM} (\textbf{Fe}ature engineering by \textbf{R}eason \textbf{G}eneration \textbf{L}arge \textbf{L}anguage \textbf{M}odels), a large language model designed to automatically perform feature engineering at an 8-billion-parameter scale. We have constructed two-stage conversational dialogues that enable language models to analyze machine learning tasks and discovering new features, exhibiting their Chain-of-Thought (CoT) capabilities. We use these dialogues to fine-tune Llama 3.1 8B model and integrate Direct Preference Optimization (DPO) to receive feedback improving quality of new features and the model's performance. Our experiments show that FeRG-LLM performs comparably to or better than Llama 3.1 70B on most datasets, while using fewer resources and achieving reduced inference time. It outperforms other studies in classification tasks and performs well in regression tasks. Moreover, since it does not rely on cloud-hosted LLMs like GPT-4 with extra API costs when generating features, it can be deployed locally, addressing security concerns.

特征工程大模型应用本地部署表格数据

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