In plain words: It encodes each pipeline's parts separately, then blends the codes into one representation that captures how they interact, so a search can predict which setup scores best. Across three collections of past results, this beat the usual search that scores parts on their own.
Abstract · Deep Pipeline Embeddings for AutoML
Automated Machine Learning (AutoML) is a promising direction for democratizing AI by automatically deploying Machine Learning systems with minimal human expertise. The core technical challenge behind AutoML is optimizing the pipelines of Machine Learning systems (e.g. the choice of preprocessing, augmentations, models, optimizers, etc.). Existing Pipeline Optimization techniques fail to explore deep interactions between pipeline stages/components. As a remedy, this paper proposes a novel neural architecture that captures the deep interaction between the components of a Machine Learning pipeline. We propose embedding pipelines into a latent representation through a novel per-component encoder mechanism. To search for optimal pipelines, such pipeline embeddings are used within deep-kernel Gaussian Process surrogates inside a Bayesian Optimization setup. Furthermore, we meta-learn the parameters of the pipeline embedding network using existing evaluations of pipelines on diverse collections of related datasets (a.k.a. meta-datasets). Through extensive experiments on three large-scale meta-datasets, we demonstrate that pipeline embeddings yield state-of-the-art results in Pipeline Optimization.
Sebastian Pineda Arango, Josif Grabocka
arXiv:2305.14009 · cs.LG · submitted May 23, 2023 · updated May 24, 2023
abstract · pdf · html · 9 pages