arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2010.08784cs.LGcs.NE
收藏

DIFER:可区分的自动化特征工程

DIFER: Differentiable Automated Feature Engineering

Nanjing University(南京大学)

Guanghui Zhu, Zhuoer Xu, Xu Guo, Chunfeng Yuan, Yihua Huang

摘要:

特征工程是机器学习的关键步骤,其目的是从原始数据中提取有用的特征,以提高数据质量。近年来,人们一直致力于自动特征工程(AutoFE),以取代昂贵的人工劳动。然而,现有的方法是计算要求,由于治疗AutoFE作为一个粗粒度的黑盒优化问题在离散空间。在这项工作中,我们提出了一个有效的基于梯度的方法称为DIFER进行微分自动特征工程在连续向量空间。DIFER基于进化算法选择潜在特征,并利用编码器-预测器-解码器控制器来优化现有特征。我们通过编码器将特征映射到连续的向量空间中,沿着预测分数引起的梯度方向优化嵌入,并通过解码器从优化的嵌入中恢复更好的特征。在分类和回归数据集上的大量实验表明,DIFER可以显着提高各种机器学习算法的性能,并在效率和性能方面优于当前最先进的AutoFE方法。

英文摘要:

Feature engineering, a crucial step of machine learning, aims to extract useful features from raw data to improve data quality. In recent years, great efforts have been devoted to Automated Feature Engineering (AutoFE) to replace expensive human labor. However, existing methods are computationally demanding due to treating AutoFE as a coarse-grained black-box optimization problem over a discrete space. In this work, we propose an efficient gradient-based method called DIFER to perform differentiable automated feature engineering in a continuous vector space. DIFER selects potential features based on evolutionary algorithm and leverages an encoder-predictor-decoder controller to optimize existing features. We map features into the continuous vector space via the encoder, optimize the embedding along the gradient direction induced by the predicted score, and recover better features from the optimized embedding by the decoder. Extensive experiments on classification and regression datasets demonstrate that DIFER can significantly improve the performance of various machine learning algorithms and outperform current state-of-the-art AutoFE methods in terms of both efficiency and performance.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑