SIGMA:基于SHAP的无元数据LLM自动特征工程

SIGMA: SHAP-Guided Implicit-Trajectory Generation for Metadata-Free LLM-Based AutoFE

精选理由

研究人员提出SIGMA方法,解决了LLM在特征工程中的元数据缺失问题,大幅降低特征重复率,提升效率。

AI 摘要

SIGMA提出了一种SHAP增强的隐式轨迹生成方法,用于无元数据的自动特征工程。该方法利用SHAP值提供任务感知信号,替代不可用的语义信息,并采用EXposed-feature Implicit Trajectory (EXIT)方法。实验显示,SIGMA将生成特征的重复率从37.2%降至6.8%,仅用平均5.4个特征就达到传统SOTA性能。

原文 · arXiv cs.LG

SIGMA: SHAP-Guided Implicit-Trajectory Generation for Metadata-Free LLM-Based AutoFE

Recent research has leveraged Large Language Models (LLMs) to enhance Automated Feature Engineering (AutoFE) through semantic descriptions and trajectory-based prompting. However, there exist two challenges that limit their applicability and scalability in long-horizon optimization: (1) semantic metadata is unavailable in many practical settings, and (2) trajectory accumulation increases the risk of exceeding the context window, while without it, the generation process can become unstable, leading to becoming stuck in the local optima and a high duplicate rate of generated features. To this end, we propose a SHAP-enhanced Implicit-trajectory Generation for Metadata-free AutoFE (SIGMA), a scalable constant-context optimization framework. SIGMA leverages SHAP values to provide task-aware signals for guiding group feature generation instead of semantic information. In addition, we adopt an EXposed-feature Implicit Trajectory (EXIT) approach, where the exposed features in the prompt implicitly represent the trajectory. Empirical results demonstrate that SIGMA achieves performance comparable to the state-of-the-art (SOTA) LLM baselines with a nearly constant prompt length. Notably, EXIT significantly reduces the duplicate ratio of generated features from 37.2% to 6.8%. At the same time, SIGMA matches traditional SOTA performance with only 5.4 features on average, demonstrating substantial efficiency gains in feature utilization.