Enhancing Classification Performance via Reinforcement Learning for Feature Selection

Younes Ghazagh Jahed, Seyyed Ali Sadat Tavana
3/9/2024
cs.LG

No confident code implementation found for this paper. 2 possible matches can be shown below.

Abstract

Feature selection plays a crucial role in improving predictive accuracy by identifying relevant features while filtering out irrelevant ones. This study investigates the importance of effective feature selection in enhancing the performance of classification models. By employing reinforcement learning (RL) algorithms, specifically Q-learning (QL) and SARSA learning, this paper addresses the feature selection challenge. Using the Breast Cancer Coimbra dataset (BCCDS) and three normalization methods (Min-Max, l1, and l2), the study evaluates the performance of these algorithms. Results show that QL@Min-Max and SARSA@l2 achieve the highest classification accuracies, reaching 87% and 88%, respectively. This highlights the effectiveness of RL-based feature selection methods in optimizing classification tasks, contributing to improved model accuracy and efficiency.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. 2 weaker matches are hidden by default — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Automation and Feature Selection Enhancement with Reinforcement Learning (RL)

84% match2025

Comprehend, Divide, and Conquer: Feature Subspace Exploration via Multi-Agent Hierarchical Reinforcement Learning

80% match2025

A Multi-Component Reward Function with Policy Gradient for Automated Feature Selection with Dynamic Regularization and Bias Mitigation

80% match2025

Knockoff-Guided Feature Selection via A Single Pre-trained Reinforced Agent

79% match2024

An Experiment on Feature Selection using Logistic Regression

79% match2024

Greedy feature selection: Classifier-dependent feature selection via greedy methods

79% match2024

Cite this paper

@article{jahed2024enhancing,
  title  = {Enhancing Classification Performance via Reinforcement Learning for Feature Selection},
  author = {Younes Ghazagh Jahed and Seyyed Ali Sadat Tavana},
  year   = {2024},
  eprint = {2403.05979},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2403.05979v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2403.05979.svg)](https://codeofpaper.com/paper/2403.05979)

Discussion