Automatic Feature Engineering Through Monte Carlo Tree Search

Publikation: Beitrag in Buch/Konferenzbericht/Sammelband/GutachtenBeitrag in KonferenzbandBeigetragenBegutachtung

Beitragende

  • Yiran Huang - , Karlsruher Institut für Technologie (Autor:in)
  • Yexu Zhou - , Karlsruher Institut für Technologie (Autor:in)
  • Michael Hefenbrock - , Karlsruher Institut für Technologie (Autor:in)
  • Till Riedel - , Karlsruher Institut für Technologie (Autor:in)
  • Likun Fang - , Karlsruher Institut für Technologie (Autor:in)
  • Michael Beigl - , Karlsruher Institut für Technologie (Autor:in)

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

OriginalspracheEnglisch
TitelMachine Learning and Knowledge Discovery in Databases
Redakteure/-innenMassih-Reza Amini, Stéphane Canu, Asja Fischer, Tias Guns, Petra Kralj Novak, Grigorios Tsoumakas
Herausgeber (Verlag)Springer, Cham
Seiten581–598
Seitenumfang18
ISBN (elektronisch)978-3-031-26409-2
ISBN (Print)978-3-031-26408-5
PublikationsstatusVeröffentlicht - 2023
Peer-Review-StatusJa
Extern publiziertJa

Publikationsreihe

ReiheLecture Notes in Computer Science, Volume 13715
ISSN0302-9743

Externe IDs

Scopus 85151051582

Schlagworte

Schlagwörter

  • Data mining, Feature engineering, Monte Carlo tree search, Reinforce learning