arXiv:2505.21978cs.LG2025-05IJCAI被引 1

用Transformer和强化学习自动生成高质量特征,提升模型表现。

Two-Stage Feature Generation with Transformer and Reinforcement Learning

  • 分两阶段生成特征:先用Transformer编码数据,再用PPO优化生成策略。
  • 在多个数据集上显著提升模型预测性能,优于现有方法。
  • 适合需要自动化特征工程的机器学习场景,尤其对非专家友好。

特征生成是机器学习中的关键步骤,旨在通过捕捉数据中的复杂关系来生成有意义的新特征以提升模型性能。传统方法高度依赖领域知识和人工干预,过程繁琐且难以适应不同场景。尽管自动化特征生成技术缓解了部分问题,但仍面临特征冗余、特征空间探索效率低及对多样化数据集和任务适应性差等挑战。为此,我们提出两阶段特征生成(TSFG)框架,结合基于Transformer的编码器-解码器架构与近端策略优化(PPO)。TSFG中的编码器-解码器利用Transformer的自注意力机制高效表示和转换特征,捕捉数据内部复杂依赖关系;PPO则根据任务反馈动态调整生成策略,优化过程以提升性能与适应性。实验结果表明,TSFG能动态生成高质量特征集,在特征质量和适应性方面均优于现有最先进方法。

原文摘要 · Abstract (English)

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 and manual intervention, making the process labor-intensive and challenging to adapt to different scenarios. Although automated feature generation techniques address these issues to some extent, they often face challenges such as feature redundancy, inefficiency in feature space exploration, and limited adaptability to diverse datasets and tasks. To address these problems, we propose a Two-Stage Feature Generation (TSFG) framework, which integrates a Transformer-based encoder-decoder architecture with Proximal Policy Optimization (PPO). The encoder-decoder model in TSFG leverages the Transformer's self-attention mechanism to efficiently represent and transform features, capturing complex dependencies within the data. PPO further enhances TSFG by dynamically adjusting the feature generation strategy based on task-specific feedback, optimizing the process for improved performance and adaptability. TSFG dynamically generates high-quality feature sets, significantly improving the predictive performance of machine learning models. Experimental results demonstrate that TSFG outperforms existing state-of-the-art methods in terms of feature quality and adaptability.

特征生成Transformer强化学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。