Back to Search Start Over

다중 에이전트 강화학습 기반 특징 선택에 대한 연구.

Authors :
김민우
배진희
왕보현
임준식
Source :
Journal of Digital Convergence; 2021, Vol. 19 Issue 12, p347-352, 6p
Publication Year :
2021

Abstract

In this paper, we propose a method for finding feature subsets that are effective for classification in an input dataset by using a multi-agent reinforcement learning method. In the field of machine learning, it is crucial to find features suitable for classification. A dataset may have numerous features; while some features may be effective for classification or prediction, others may have little or rather negative effects on results. In machine learning problems, feature selection for increasing classification or prediction accuracy is a critical problem. To solve this problem, we proposed a feature selection method based on reinforced learning. Each feature has one agent, which determines whether the feature is selected . After obtaining corresponding rewards for each feature that is selected, but not by the agents, the Q-value of each agent is updated by comparing the rewards. The reward comparison of the two subsets helps agents determine whether their actions were right. These processes are performed as many times as the number of episodes, and finally, features are selected. As a result of applying this method to the Wisconsin Breast Cancer, Spambase, Musk, and Colon Cancer datasets, accuracy improvements of 0.0385, 0.0904, 0.1252 and 0.2055 were shown, respectively, and finally, classification accuracies of 0.9789, 0.9311, 0.9691 and 0.9474 were achieved, respectively. It was proved that our proposed method could properly select features that were effective for classification and increase classification accuracy. [ABSTRACT FROM AUTHOR]

Details

Language :
Korean
ISSN :
27136434
Volume :
19
Issue :
12
Database :
Complementary Index
Journal :
Journal of Digital Convergence
Publication Type :
Academic Journal
Accession number :
155590635
Full Text :
https://doi.org/10.14400/JDC.2021.19.12.347