用真实竞赛数据评估大模型生成特征效果,发现其与专家水平差距明显。
ELF-Gym: Evaluating Large Language Models Generated Features for Tabular Prediction
- 构建新框架ELF-Gym,量化评估大模型生成特征的质量。
- 大模型仅能捕捉约56%的优质特征,复杂任务下准确率降至13%。
- 适合关注自动化特征工程、模型可解释性的研究者参考。
有效特征工程是机器学习流程中关键但耗时且领域依赖的任务。尽管大语言模型(LLM)在自动化数据科学任务方面展现潜力,但现有评估多基于完整模型管道的最终性能,难以揭示大模型在特征工程上与人类专家的真实差距。为此,我们提出ELF-Gym框架,用于评估大模型生成特征的表现。我们从历史Kaggle竞赛中构建了一个新数据集,包含251个由顶尖团队使用的“黄金”特征。ELF-Gym通过下游模型性能提升及语义与功能相似性分析,定量评估大模型生成特征的效果。结果表明,在最优情况下,大模型可语义捕获约56%的黄金特征,但在更严格的实现层面,该重合率下降至13%。此外,在需复杂特征的数据集上,大模型可能完全失效,暴露出显著改进空间。
原文摘要 · Abstract (English)
Crafting effective features is a crucial yet labor-intensive and domain-specific task within machine learning pipelines. Fortunately, recent advancements in Large Language Models (LLMs) have shown promise in automating various data science tasks, including feature engineering. But despite this potential, evaluations thus far are primarily based on the end performance of a complete ML pipeline, providing limited insight into precisely how LLMs behave relative to human experts in feature engineering. To address this gap, we propose ELF-Gym, a framework for Evaluating LLM-generated Features. We curated a new dataset from historical Kaggle competitions, including 251 "golden" features used by top-performing teams. ELF-Gym then quantitatively evaluates LLM-generated features by measuring their impact on downstream model performance as well as their alignment with expert-crafted features through semantic and functional similarity assessments. This approach provides a more comprehensive evaluation of disparities between LLMs and human experts, while offering valuable insights into specific areas where LLMs may have room for improvement. For example, using ELF-Gym we empirically demonstrate that, in the best-case scenario, LLMs can semantically capture approximately 56% of the golden features, but at the more demanding implementation level this overlap drops to 13%. Moreover, in other cases LLMs may fail completely, particularly on datasets that require complex features, indicating broad potential pathways for improvement.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。