arXiv cs.AI· Fateme Memar, Dongjie Wang, Arvin Agah·· 12 小时前AI 评分27
LLM 何时有助于特征选择?一项系统性实证研究
When Do LLMs Help with Feature Selection? A Systematic Empirical Study
AI 导读
一项覆盖 18 个数据集的系统性实证研究发现,在表格特征选择任务上,经典统计选择器整体预测性能优于测试的 7B LLM/RL 选择器,包括在约 33% 特征预算下。
正文 · 原文
Abstract:Large language models (LLMs) can leverage semantic feature names alongside summary statistics for tabular feature selection, yet it remains unclear when such information improves subset selection relative to established statistical methods. We investigate this question within an 18-dataset framework comprising 10 datasets for Group Relative Policy Optimization (GRPO) training, 2 for checkpoint validation, and 6 held-out evaluation datasets. We compare five classical baselines, frozen Qwen2.5-7B and Llama-3.3-70B selectors, and two GRPO-adapted Qwen2.5-7B models, reporting classification ROC-AUC and regression R^2 separately and using dataset-wise ranks for cross-task comparison. Across five seeded evaluation partitions, classical selectors generally achieve higher predictive performance than the tested 7B LLM/RL selectors, including under a pre-specified ~33% feature budget with strict train-only classical fitting, while the 7B selectors retain compact subsets. A cardinality-matched semantic sensitivity experiment shows that, for Zero-shot Qwen-7B, informative feature names improve regression performance over anonymized and shuffled alternatives (+0.1964 and +0.1606 mean R^2), whereas aggregate classification differences are smaller (+0.0159 and +0.0084 ROC-AUC) and dataset-dependent. These results characterize when semantic information benefits LLM-based feature selection and show that classical selectors remain strong baselines for predictive tabular feature selection.
| Comments: | 8 pages, 6 figures, 2 tables |
| Subjects: | Machine Learning (cs.LG); Artificial Intelligence (cs.AI) |
| Cite as: | arXiv:2610.28556 [cs.LG] |
| (or arXiv:2610.28556v1 [cs.LG] for this version) | |
| https://doi.org/10.48550/arXiv.2610.28556 arXiv-issued DOI via DataCite (pending registration) |
Submission history
From: Fateme Memar [view email]
[v1]
Tue, 20 Oct 2026 17:18:55 UTC (1,220 KB)
来源:arXiv cs.AI · arxiv.org