用分层智能体探索特征子空间,提升机器学习性能与效率
Comprehend, Divide, and Conquer: Feature Subspace Exploration via Multi-Agent Hierarchical Reinforcement Learning
- 通过大模型提取特征属性,聚类后构建分层智能体
- 相比单特征单智能体,性能更优且运行更快
- 适合处理复杂数据集的特征选择任务
特征选择旨在预处理数据集,找到最优且最精简的特征子集,以提升下游机器学习任务表现。在滤波、封装和嵌入式方法中,基于强化学习的子空间探索策略提供了目标优化导向的新视角并展现良好性能。然而,当前强化学习方法在处理复杂数据集时仍面临与传统方法类似的挑战,主要源于每个特征对应一个智能体的低效范式及数据本身的固有复杂性。为此,本文提出一种新方法HRLFS:首先利用大语言模型(LLM)构建混合状态提取器,捕捉各特征的数学与语义特性;基于此信息对特征进行聚类,进而为每个簇及子簇构建分层智能体。大量实验表明,该方法在效率、可扩展性和鲁棒性方面均表现优异。相较于现有或单特征单智能体的强化学习方法,HRLFS通过迭代式特征子空间探索,在提升下游机器学习性能的同时,显著减少参与智能体数量,加快整体运行时间。
原文摘要 · Abstract (English)
Feature selection aims to preprocess the target dataset, find an optimal and most streamlined feature subset, and enhance the downstream machine learning task. Among filter, wrapper, and embedded-based approaches, the reinforcement learning (RL)-based subspace exploration strategy provides a novel objective optimization-directed perspective and promising performance. Nevertheless, even with improved performance, current reinforcement learning approaches face challenges similar to conventional methods when dealing with complex datasets. These challenges stem from the inefficient paradigm of using one agent per feature and the inherent complexities present in the datasets. This observation motivates us to investigate and address the above issue and propose a novel approach, namely HRLFS. Our methodology initially employs a Large Language Model (LLM)-based hybrid state extractor to capture each feature's mathematical and semantic characteristics. Based on this information, features are clustered, facilitating the construction of hierarchical agents for each cluster and sub-cluster. Extensive experiments demonstrate the efficiency, scalability, and robustness of our approach. Compared to contemporary or the one-feature-one-agent RL-based approaches, HRLFS improves the downstream ML performance with iterative feature subspace exploration while accelerating total run time by reducing the number of agents involved.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。