English

LLM4FS: Leveraging Large Language Models for Feature Selection

Machine Learning 2025-12-12 v4

Abstract

Recent advances in large language models (LLMs) have provided new opportunities for decision-making, particularly in the task of automated feature selection. In this paper, we first comprehensively evaluate LLM-based feature selection methods, covering the state-of-the-art DeepSeek-R1, GPT-o3-mini, and GPT-4.5. Then, we propose a new hybrid strategy called LLM4FS that integrates LLMs with traditional data-driven methods. Specifically, input data samples into LLMs, and directly call traditional data-driven techniques such as random forest and forward sequential selection. Notably, our analysis reveals that the hybrid strategy leverages the contextual understanding of LLMs and the high statistical reliability of traditional data-driven methods to achieve excellent feature selection performance, even surpassing LLMs and traditional data-driven methods. Finally, we point out the limitations of its application in decision-making. Our code is available at https://github.com/xianchaoxiu/LLM4FS.

Keywords

Cite

@article{arxiv.2503.24157,
  title  = {LLM4FS: Leveraging Large Language Models for Feature Selection},
  author = {Jianhao Li and Xianchao Xiu},
  journal= {arXiv preprint arXiv:2503.24157},
  year   = {2025}
}

Comments

The experimental section should be expanded

R2 v1 2026-06-28T22:40:41.678Z