Automatic Feature Engineering Through Monte Carlo Tree Search
Publikation: Beitrag in Buch/Konferenzbericht/Sammelband/Gutachten › Beitrag in Konferenzband › Beigetragen › Begutachtung
Beitragende
Abstract
The performance of machine learning models depends heavily on the feature space and feature engineering. Although neural networks have made significant progress in learning latent feature spaces from data, compositional feature engineering through nested feature transformations can reduce model complexity and can be particularly desirable for interpretability. To find suitable transformations automatically, state-of-the-art methods model the feature transformation space by graph structures and use heuristics such as ϵ -greedy to search for them. Such search strategies tend to become less efficient over time because they do not consider the sequential information of the candidate sequences and cannot dynamically adjust the heuristic strategy. To address these shortcomings, we propose a reinforcement learning-based automatic feature engineering method, which we call Monte Carlo tree search Automatic Feature Engineering (mCAFE). We employ a surrogate model that can capture the sequential information contained in the transformation sequence and thus can dynamically adjust the exploration strategy. It balances exploration and exploitation by Thompson sampling and uses a Long Short Term Memory (LSTM) based surrogate model to estimate sequences of promising transformations. In our experiments, mCAFE outperformed state-of-the-art automatic feature engineering methods on most common benchmark datasets.
Details
Originalsprache | Englisch |
---|---|
Titel | Machine Learning and Knowledge Discovery in Databases |
Redakteure/-innen | Massih-Reza Amini, Stéphane Canu, Asja Fischer, Tias Guns, Petra Kralj Novak, Grigorios Tsoumakas |
Herausgeber (Verlag) | Springer, Cham |
Seiten | 581–598 |
Seitenumfang | 18 |
ISBN (elektronisch) | 978-3-031-26409-2 |
ISBN (Print) | 978-3-031-26408-5 |
Publikationsstatus | Veröffentlicht - 2023 |
Peer-Review-Status | Ja |
Extern publiziert | Ja |
Publikationsreihe
Reihe | Lecture Notes in Computer Science, Volume 13715 |
---|---|
ISSN | 0302-9743 |
Externe IDs
Scopus | 85151051582 |
---|
Schlagworte
ASJC Scopus Sachgebiete
Schlagwörter
- Data mining, Feature engineering, Monte Carlo tree search, Reinforce learning