mlsanityData to 7 Oct 2026

Automation and Feature Selection Enhancement with Reinforcement Learning (RL)

arXiv 2503.11991 ↗First posted 15 Mar 2025cs.LGcited by 2

Not on the map. Why?

Abstract (arXiv)

Effective feature selection, representation and transformation are principal steps in machine learning to improve prediction accuracy, model generalization and computational efficiency. Reinforcement learning provides a new perspective towards balanced exploration of optimal feature subset using multi-agent and single-agent models. Interactive reinforcement learning integrated with decision tree improves feature knowledge, state representation and selection efficiency, while diversified teaching strategies improve both selection quality and efficiency. The state representation can further be enhanced by scanning features sequentially along with the usage of convolutional auto-encoder. Monte Carlo-based reinforced feature selection(MCRFS), a single-agent feature selection method reduces computational burden by incorporating early-stopping and reward-level interactive strategies. A dual-agent RL framework is also introduced that collectively selects features and instances, capturing the interactions between them. This enables the agents to navigate through complex data spaces. To outperform the traditional feature engineering, cascading reinforced agents are used to iteratively improve the feature space, which is a self-optimizing framework. The blend of reinforcement learning, multi-agent systems, and bandit-based approaches offers exciting paths for studying scalable and interpretable machine learning solutions to handle high-dimensional data and challenging predictive tasks.

Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 9 of its 15 references that are in the map. First posted on arXiv 15 Mar 2025.

Builds on

9 of 15 references.

References · 9

  1. 2015Deep Reinforcement Learning with Double Q-learningcited by 1,704
  2. 2022Group-wise Reinforcement Feature Generation for Optimal and Explainable Representation Space Reconstructioncited by 35
  3. 2020AutoFS: Automated Feature Selection via Diversity-aware Interactive Reinforcement Learningcited by 24
  4. 2021Efficient Reinforced Feature Selection via Early Stopping Traverse Strategycited by 19
  5. 2020Interactive Reinforcement Learning for Feature Selection with Decision Tree in the Loopcited by 18
  6. 2020Simplifying Reinforced Feature Selection via Restructured Choice Strategy of Single Agentcited by 14
  7. 2023Feature Interaction Aware Automated Data Representation Transformationcited by 6
  8. 2022Self-Optimizing Feature Transformationcited by 4
  9. 2022Feature and Instance Joint Selection: A Reinforcement Learning Perspectivecited by 3

Built on by

Cited by year

Citing papers per year: 2025: 2.

2 papers.