Predicting Machine Learning Pipeline Runtimes in the Context of Automated Machine Learning

Publikation: Beitrag in FachzeitschriftArtikelForschungPeer-Review

Autoren

Externe Organisationen

  • Universidad de la Sabana
  • Heinz Nixdorf Institut (HNI)
  • Universität Paderborn
Forschungs-netzwerk anzeigen

Details

OriginalspracheEnglisch
Aufsatznummer9347828
Seiten (von - bis)3055-3066
Seitenumfang12
FachzeitschriftIEEE Transactions on Pattern Analysis and Machine Intelligence
Jahrgang43
Ausgabenummer9
PublikationsstatusVeröffentlicht - 1 Sept. 2021
Extern publiziertJa

Abstract

Automated machine learning (AutoML) seeks to automatically find so-called machine learning pipelines that maximize the prediction performance when being used to train a model on a given dataset. One of the main and yet open challenges in AutoMLis an effective use of computational resources: An AutoML process involves the evaluation of many candidate pipelines, which are costly but often ineffective because they are canceled due to a timeout. In this paper, we present an approach to predict the runtime of two-step machine learning pipelines with up to one pre-processor, which can be used to anticipate whether or not a pipeline will time out. Separate runtime models are trained offline for each algorithm that may be used in a pipeline, and an overall prediction is derived from these models. We empirically show that the approach increases successful evaluations made by an AutoML tool while preserving or even improving on the previously best solutions.

ASJC Scopus Sachgebiete

Zitieren

Predicting Machine Learning Pipeline Runtimes in the Context of Automated Machine Learning. / Mohr, Felix; Wever, Marcel; Tornede, Alexander et al.
in: IEEE Transactions on Pattern Analysis and Machine Intelligence, Jahrgang 43, Nr. 9, 9347828, 01.09.2021, S. 3055-3066.

Publikation: Beitrag in FachzeitschriftArtikelForschungPeer-Review

Download
@article{49316f6e9f9d449f846952458027bee8,
title = "Predicting Machine Learning Pipeline Runtimes in the Context of Automated Machine Learning",
abstract = "Automated machine learning (AutoML) seeks to automatically find so-called machine learning pipelines that maximize the prediction performance when being used to train a model on a given dataset. One of the main and yet open challenges in AutoMLis an effective use of computational resources: An AutoML process involves the evaluation of many candidate pipelines, which are costly but often ineffective because they are canceled due to a timeout. In this paper, we present an approach to predict the runtime of two-step machine learning pipelines with up to one pre-processor, which can be used to anticipate whether or not a pipeline will time out. Separate runtime models are trained offline for each algorithm that may be used in a pipeline, and an overall prediction is derived from these models. We empirically show that the approach increases successful evaluations made by an AutoML tool while preserving or even improving on the previously best solutions.",
keywords = "Automated machine learning, hierarchical runtime prediction, runtime prediction for classifiers and pipelines",
author = "Felix Mohr and Marcel Wever and Alexander Tornede and Eyke Hullermeier",
note = "Publisher Copyright: {\textcopyright} 1979-2012 IEEE.",
year = "2021",
month = sep,
day = "1",
doi = "10.1109/tpami.2021.3056950",
language = "English",
volume = "43",
pages = "3055--3066",
journal = "IEEE Transactions on Pattern Analysis and Machine Intelligence",
issn = "0162-8828",
publisher = "IEEE Computer Society",
number = "9",

}

Download

TY - JOUR

T1 - Predicting Machine Learning Pipeline Runtimes in the Context of Automated Machine Learning

AU - Mohr, Felix

AU - Wever, Marcel

AU - Tornede, Alexander

AU - Hullermeier, Eyke

N1 - Publisher Copyright: © 1979-2012 IEEE.

PY - 2021/9/1

Y1 - 2021/9/1

N2 - Automated machine learning (AutoML) seeks to automatically find so-called machine learning pipelines that maximize the prediction performance when being used to train a model on a given dataset. One of the main and yet open challenges in AutoMLis an effective use of computational resources: An AutoML process involves the evaluation of many candidate pipelines, which are costly but often ineffective because they are canceled due to a timeout. In this paper, we present an approach to predict the runtime of two-step machine learning pipelines with up to one pre-processor, which can be used to anticipate whether or not a pipeline will time out. Separate runtime models are trained offline for each algorithm that may be used in a pipeline, and an overall prediction is derived from these models. We empirically show that the approach increases successful evaluations made by an AutoML tool while preserving or even improving on the previously best solutions.

AB - Automated machine learning (AutoML) seeks to automatically find so-called machine learning pipelines that maximize the prediction performance when being used to train a model on a given dataset. One of the main and yet open challenges in AutoMLis an effective use of computational resources: An AutoML process involves the evaluation of many candidate pipelines, which are costly but often ineffective because they are canceled due to a timeout. In this paper, we present an approach to predict the runtime of two-step machine learning pipelines with up to one pre-processor, which can be used to anticipate whether or not a pipeline will time out. Separate runtime models are trained offline for each algorithm that may be used in a pipeline, and an overall prediction is derived from these models. We empirically show that the approach increases successful evaluations made by an AutoML tool while preserving or even improving on the previously best solutions.

KW - Automated machine learning

KW - hierarchical runtime prediction

KW - runtime prediction for classifiers and pipelines

UR - http://www.scopus.com/inward/record.url?scp=85100855731&partnerID=8YFLogxK

U2 - 10.1109/tpami.2021.3056950

DO - 10.1109/tpami.2021.3056950

M3 - Article

C2 - 33539291

VL - 43

SP - 3055

EP - 3066

JO - IEEE Transactions on Pattern Analysis and Machine Intelligence

JF - IEEE Transactions on Pattern Analysis and Machine Intelligence

SN - 0162-8828

IS - 9

M1 - 9347828

ER -

Von denselben Autoren