Wanfu Gao

Dual-Agent Reinforcement Learning for Automated Feature Generation

2024

Feature generation involves creating new features from raw data to capture complex relationships among the original features, improving model robustness and machine learning performance. Current methods using reinforcement learning for feature generation have made feature …

Two-Stage Feature Generation with Transformer and Reinforcement Learning

2025 · arXiv (Cornell University)

Feature generation is a critical step in machine learning, aiming to enhance model performance by capturing complex relationships within the data and generating meaningful new features. Traditional feature generation methods heavily rely on domain expertise …