Saved in:
Bibliographic Details
Main Authors: Li, Dawei, Tan, Zhen, Liu, Huan
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2408.12025
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916449304444928
author Li, Dawei
Tan, Zhen
Liu, Huan
author_facet Li, Dawei
Tan, Zhen
Liu, Huan
contents The rapid advancement of Large Language Models (LLMs) has significantly influenced various domains, leveraging their exceptional few-shot and zero-shot learning capabilities. In this work, we aim to explore and understand the LLMs-based feature selection methods from a data-centric perspective. We begin by categorizing existing feature selection methods with LLMs into two groups: data-driven feature selection which requires numerical values of samples to do statistical inference and text-based feature selection which utilizes prior knowledge of LLMs to do semantical associations using descriptive context. We conduct experiments in both classification and regression tasks with LLMs in various sizes (e.g., GPT-4, ChatGPT and LLaMA-2). Our findings emphasize the effectiveness and robustness of text-based feature selection methods and showcase their potentials using a real-world medical application. We also discuss the challenges and future opportunities in employing LLMs for feature selection, offering insights for further research and development in this emerging field.
format Preprint
id arxiv_https___arxiv_org_abs_2408_12025
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Exploring Large Language Models for Feature Selection: A Data-centric Perspective
Li, Dawei
Tan, Zhen
Liu, Huan
Artificial Intelligence
The rapid advancement of Large Language Models (LLMs) has significantly influenced various domains, leveraging their exceptional few-shot and zero-shot learning capabilities. In this work, we aim to explore and understand the LLMs-based feature selection methods from a data-centric perspective. We begin by categorizing existing feature selection methods with LLMs into two groups: data-driven feature selection which requires numerical values of samples to do statistical inference and text-based feature selection which utilizes prior knowledge of LLMs to do semantical associations using descriptive context. We conduct experiments in both classification and regression tasks with LLMs in various sizes (e.g., GPT-4, ChatGPT and LLaMA-2). Our findings emphasize the effectiveness and robustness of text-based feature selection methods and showcase their potentials using a real-world medical application. We also discuss the challenges and future opportunities in employing LLMs for feature selection, offering insights for further research and development in this emerging field.
title Exploring Large Language Models for Feature Selection: A Data-centric Perspective
topic Artificial Intelligence
url https://arxiv.org/abs/2408.12025