When Do LLMs Help with Feature Selection? A Systematic Empirical Study
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.
ProofPaper ↗
Key points
- 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.
- 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.
Sources (1)
- [1]When Do LLMs Help with Feature Selection? A Systematic Empirical StudyarXiv (AI, ML, NLP, CV, robotics, multi-agent) · Oct 20, 05:18 PM
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.
Extractive summary: sentences quoted from the sources.
Before this
- Oct 20, 2026FairStress: A deeper evaluation of fairness in LLM decision-making
- Oct 20, 2026RANGER: Retrieval Agent for Navigating Graphs with Semantic Evaluation Rewards
- Oct 19, 2026Evaluating Agents on Highly Subjective Tasks
- Oct 19, 2026Safety Mid-Training: Internalizing LLM Safety as a Foundational Capability
- Oct 19, 2026Introducing Clairic, a new System One class of extraction models
- Oct 19, 2026Push-based KV transfer to improve TTFT and TPOT for disaggregated inference in vLLM
