Residential College | false |
Status | 已發表Published |
Toward Efficient Automated Feature Engineering | |
Kafeng Wang1![]() ![]() ![]() ![]() | |
2023-04 | |
Conference Name | The 39th IEEE International Conference on Data Engineering |
Source Publication | Proceedings - 2023 IEEE 39th International Conference on Data Engineering (ICDE)
![]() |
Volume | 2023-April |
Pages | 1625-1637 |
Conference Date | 3-7 April 2023 |
Conference Place | Anaheim, CA |
Country | USA |
Publisher | IEEE Computer Society |
Abstract | Automated Feature Engineering (AFE) refers to automatically generate and select optimal feature sets for downstream tasks, which has achieved great success in real-world applications. Current AFE methods mainly focus on improving the effectiveness of the produced features, but ignoring the lowefficiency issue for large-scale deployment. Therefore, in this work, we propose a generic framework to improve the efficiency of AFE. Specifically, we construct the AFE pipeline based on reinforcement learning setting, where each feature is assigned an agent to perform feature transformation and selection, and the evaluation score of the produced features in downstream tasks serve as the reward to update the policy. We improve the efficiency of AFE in two perspectives. On the one hand, we develop a Feature Pre-Evaluation (FPE) Model to reduce the sample size and feature size that are two main factors on undermining the efficiency of feature evaluation. On the other hand, we devise a two-stage policy training strategy by running FPE on the pre-evaluation task as the initialization of the policy to avoid training policy from scratch. We conduct comprehensive experiments on 36 datasets in terms of both classification and regression tasks. The results show 2.9% higher performance in average and 2x higher computational efficiency comparing to state-of-the-art AFE methods. |
Keyword | Approximate Hashing Automated Feature Engineering Minhash Off-policy Reinforcement Learning |
DOI | 10.1109/ICDE55515.2023.00128 |
URL | View the original |
Language | 英語English |
Scopus ID | 2-s2.0-85167688238 |
Fulltext Access | |
Citation statistics | |
Document Type | Conference paper |
Collection | DEPARTMENT OF COMPUTER AND INFORMATION SCIENCE Faculty of Science and Technology THE STATE KEY LABORATORY OF INTERNET OF THINGS FOR SMART CITY (UNIVERSITY OF MACAU) |
Corresponding Author | Kafeng Wang; Chengzhong Xu |
Affiliation | 1.Shenzhen Institute of Advanced Technology, Chinese Academy of Sciences, and University of Chinese Academy of Sciences, and University of Macau, Macau, China, and Guangdong-Hong Kong-Macao Joint Laboratory of Human-Machine Intelligence-Synergy Systems, She 2.State Key Laboratory of Internet of Things for Smart City, and Department of Computer and Information Science, University of Macau, Macau, China |
First Author Affilication | University of Macau |
Corresponding Author Affilication | University of Macau |
Recommended Citation GB/T 7714 | Kafeng Wang,Pengyang Wang,Chengzhong Xu. Toward Efficient Automated Feature Engineering[C]:IEEE Computer Society, 2023, 1625-1637. |
APA | Kafeng Wang., Pengyang Wang., & Chengzhong Xu (2023). Toward Efficient Automated Feature Engineering. Proceedings - 2023 IEEE 39th International Conference on Data Engineering (ICDE), 2023-April, 1625-1637. |
Files in This Item: | There are no files associated with this item. |
Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.
Edit Comment