LLM-FS-Agent: A Deliberative Role-based Large Language Model Architecture for Transparent Feature Selection
cs.LG, cs.AI
Submitted: 2025-10-07
Updated: 2025-10-07
DOI: 10.11591/ijai.v15.i5.pp4026-4038
License: http://creativecommons.org/licenses/by/4.0/
The gist: High-dimensional data remains a pervasive challenge in machine learning, often undermining model interpretability and computational efficiency.
Terminology
Abstract
High-dimensional data remains a pervasive challenge in machine learning, often undermining model interpretability and computational efficiency. While Large Language Models (LLMs) have shown promise for dimensionality reduction through feature selection, existing LLM-based approaches frequently lack structured reasoning and transparent justification for their decisions. This paper introduces LLM-FS-Agent, a novel multi-agent architecture designed for interpretable and robust feature selection. The system orchestrates a deliberative "debate" among multiple LLM agents, each assigned a specific role, enabling collective evaluation of feature relevance and generation of detailed justifications. We evaluate LLM-FS-Agent in the cybersecurity domain using the CIC-DIAD 2024 IoT intrusion detection dataset and compare its performance against strong baselines, including LLM-Select and traditional methods such as PCA. Experimental results demonstrate that LLM-FS-Agent consistently achieves superior or comparable classification performance while reducing downstream training time by an average of 46% (statistically significant improvement, p = 0.028 for XGBoost). These findings highlight that the proposed deliberative architecture enhances both decision transparency and computational efficiency, establishing LLM-FS-Agent as a practical and reliable solution for real-world applications.
Sources
- Tabular Data Augmentation for Machine Learning: Progress and Prospects of Embracing Generative AI
- BAMBOO: A Comprehensive Benchmark for Evaluating Long Text Modeling Capacities of Large Language Models
- ICE-SEARCH: A Language Model-Driven Feature Selection Approach
- The Rise and Potential of Large Language Model Based Agents: A Survey
- Data Interpreter: An LLM Agent For Data Science
- Wide-range resistivity characterization of semiconductors with terahertz time-domain spectroscopy
- Mixture-of-Agents Enhances Large Language Model Capabilities
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks