A Double-Layer Reinforcement Learning Feature Optimization Framework for Evolutionary Computation Based Feature Selection Algorithms

Hong WANG, Yaofa SU, Xiaolong OU, Jinxin ZHANG, Ben NIU*

*Corresponding author for this work

Research output: Book Chapters | Papers in Conference ProceedingsConference paper (refereed)Referred Conference Paperpeer-review

Abstract

Recently, Evolution Computing (EC) has gained widespread use in Feature Selection due to its powerful search capabilities. However, many EC algorithms fail to fully utilize historical combination information between features. Moreover, when faced with ultra-high dimensional data, they often lack the necessary decision-making ability to select a suitable optimization direction. In this paper, we propose a double layer-reinforcement learning framework feature optimizes framework. The framework aids the EC algorithm by continuously obtaining and utilizing combined feedback from features during the iteration process. We leverage the adaptability and decision-making abilities of reinforcement learning to overcome the EC algorithm’s limitations. We conducted experiments on 8 datasets from UCI to evaluate the effectiveness of our framework. The experimental results demonstrated that the EC algorithms, optimized by our framework, achieves lower error rates and requires fewer features. Consequently, we posit that reinforcement learning can offer novel methods and ideas for the application of evolutionary computing in feature selection.

Original languageEnglish
Title of host publicationData Mining and Big Data : 8th International Conference, DMBD 2023, Proceedings
EditorsYing TAN, Yuhui SHI
PublisherSpringer Science and Business Media Deutschland GmbH
Pages225-240
Number of pages16
ISBN (Electronic)9789819708376
ISBN (Print)9789819708369
DOIs
Publication statusPublished - 2024
Event8th International Conference on Data Mining and Big Data, DMBD 2023 - Sanya, China
Duration: 9 Dec 202312 Dec 2023

Publication series

NameCommunications in Computer and Information Science
PublisherSpringer
Volume2017
ISSN (Print)1865-0929
ISSN (Electronic)1865-0937

Conference

Conference8th International Conference on Data Mining and Big Data, DMBD 2023
Country/TerritoryChina
CitySanya
Period9/12/2312/12/23

Bibliographical note

Publisher Copyright:
© The Author(s), under exclusive license to Springer Nature Singapore Pte Ltd. 2024.

Keywords

  • Evolution Computing
  • Feature Selection
  • Reinforcement Learning

Fingerprint

Dive into the research topics of 'A Double-Layer Reinforcement Learning Feature Optimization Framework for Evolutionary Computation Based Feature Selection Algorithms'. Together they form a unique fingerprint.

Cite this