期刊
KNOWLEDGE-BASED SYSTEMS
卷 238, 期 -, 页码 -出版社
ELSEVIER
DOI: 10.1016/j.knosys.2021.107835
关键词
Automatic feature generation; Categorical features; Graph Structure Learning; Reinforcement learning
资金
- National Key R&D Program of China [2019YFC1711000]
- Collaborative Innovation Center of Novel Software Technology and Industrialization
Feature engineering relies on domain knowledge and human intervention. To automate this process, this paper proposes a novel Automatic Feature Generation model based on Graph Structure Learning (AFGSL). AFGSL utilizes the adjacency matrix to model feature relationships and employs Q-learning to train the stacking interaction layers, achieving better performance.
Feature engineering relies on domain knowledge and human intervention. To automate the process of feature engineering, automated feature construction methods use deep neural networks to capture feature interactions and attention coefficients to quantify the relationship between features. However, these methods ignore the influence of insignificant features that introduce noise and degrade the performance of the model. In this paper, we study the problem of feature interactions from the perspective of graph and propose a novel Automatic Feature Generation model based on Graph Structure Learning, called AFGSL. In this model, the adjacency matrix reflects the relationships between features, so that it can be used to filter out insignificant features. Furthermore, Q-learning is used to train the policy of stacking interaction layers, which enables it to make full use of both local and global information in the process of feature generation. The results of experiments on four real-world datasets show that AFGSL outperforms the state-of-the-art methods. (c) 2021 Elsevier B.V. All rights reserved.
作者
我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。
推荐
暂无数据