arXiv:2508.20824cs.LG2025-08中稿 · APWeb-WAIM 2025被引 1

用改进的GPT实现高效自动特征转换,提升模型性能。

GPT-FT: An Efficient Automated Feature Transformation Using GPT for Sequence Reconstruction and Performance Enhancement

  • 用改进GPT构建嵌入空间,实现特征序列重建与性能预测
  • 在多个基准数据集上达到或超过基线性能,计算效率显著提升
  • 适合需要自动化特征工程的机器学习应用

特征转换对提升机器学习模型性能至关重要,通过优化数据表示实现。现有先进方法将此任务视为连续嵌入优化问题,将离散搜索转化为可学习过程。然而,这些方法常依赖序列编码器-解码器结构,导致高计算成本和参数需求,限制了可扩展性与效率。为此,我们提出一种新框架,通过四步完成自动化特征转换:转换记录收集、基于改进生成式预训练变压器(GPT)模型的嵌入空间构建、梯度上升搜索以及自回归重建。该改进GPT模型承担双重功能:(a) 特征转换序列重建;(b) 通过构建嵌入空间估计并提升下游任务性能。该多目标优化框架减少了参数量并加速了转换过程。在基准数据集上的实验结果表明,所提框架在性能上匹配或超越基线,且计算效率大幅提升。本工作凸显了基于Transformer架构在可扩展、高性能自动化特征转换中的潜力。

原文摘要 · Abstract (English)

Feature transformation plays a critical role in enhancing machine learning model performance by optimizing data representations. Recent state-of-the-art approaches address this task as a continuous embedding optimization problem, converting discrete search into a learnable process. Although effective, these methods often rely on sequential encoder-decoder structures that cause high computational costs and parameter requirements, limiting scalability and efficiency. To address these limitations, we propose a novel framework that accomplishes automated feature transformation through four steps: transformation records collection, embedding space construction with a revised Generative Pre-trained Transformer (GPT) model, gradient-ascent search, and autoregressive reconstruction. In our approach, the revised GPT model serves two primary functions: (a) feature transformation sequence reconstruction and (b) model performance estimation and enhancement for downstream tasks by constructing the embedding space. Such a multi-objective optimization framework reduces parameter size and accelerates transformation processes. Experimental results on benchmark datasets show that the proposed framework matches or exceeds baseline performance, with significant gains in computational efficiency. This work highlights the potential of transformer-based architectures for scalable, high-performance automated feature transformation.

特征转换GPT自动化效率提升

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