Return
Feature Selection Method Using Multi-Agent Reinforcement Learning Based on Guide Agents
DOI:10.3390/s23010098.png)
Abstract
En 中文
In this study, we propose a method to automatically find features from a dataset that are effective for classification or prediction, using a new method called multi-agent reinforcement learning and a guide agent. Each feature of the dataset has one of the main and guide agents, and these agents decide whether to select a feature. Main agents select the optimal features, and guide agents present the criteria for judging the main agents' actions. After obtaining the main and guide rewards for the features selected by the agents, the main agent that behaves differently from the guide agent updates their Q-values by calculating the learning reward delivered to the main agents. The behavior comparison helps the main agent decide whether its own behavior is correct, without using other algorithms. After performing this process for each episode, the features are finally selected. The feature selection method proposed in this study uses multiple agents, reducing the number of actions each agent can perform and finding optimal features effectively and quickly. Finally, comparative experimental results on multiple datasets show that the proposed method can select effective features for classification and increase classification accuracy.
Keywords:
feature selection
guide agents
main agents
multi-agent
reinforcement learning (RL)
rewards
AI Summary
Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.
Journal
IF:
3.5
Papers:
7.2W
Citations:
20.9W
Organization
Cited Papers
Two stage forecast engine with feature selection technique and improved meta-heuristic algorithm for electricity load forecasting
ENERGY
IF9.4
Application of high-dimensional feature selection: evaluation for genomic prediction in man
SCIENTIFIC REPORTS
IF3.9

