English

AVATAR -- Machine Learning Pipeline Evaluation Using Surrogate Model

Machine Learning 2020-02-04 v2 Machine Learning

Abstract

The evaluation of machine learning (ML) pipelines is essential during automatic ML pipeline composition and optimisation. The previous methods such as Bayesian-based and genetic-based optimisation, which are implemented in Auto-Weka, Auto-sklearn and TPOT, evaluate pipelines by executing them. Therefore, the pipeline composition and optimisation of these methods requires a tremendous amount of time that prevents them from exploring complex pipelines to find better predictive models. To further explore this research challenge, we have conducted experiments showing that many of the generated pipelines are invalid, and it is unnecessary to execute them to find out whether they are good pipelines. To address this issue, we propose a novel method to evaluate the validity of ML pipelines using a surrogate model (AVATAR). The AVATAR enables to accelerate automatic ML pipeline composition and optimisation by quickly ignoring invalid pipelines. Our experiments show that the AVATAR is more efficient in evaluating complex pipelines in comparison with the traditional evaluation approaches requiring their execution.

Keywords

Cite

@article{arxiv.2001.11158,
  title  = {AVATAR -- Machine Learning Pipeline Evaluation Using Surrogate Model},
  author = {Tien-Dung Nguyen and Tomasz Maszczyk and Katarzyna Musial and Marc-Andre Zöller and Bogdan Gabrys},
  journal= {arXiv preprint arXiv:2001.11158},
  year   = {2020}
}

Comments

The Eighteenth International Symposium on Intelligent Data Analysis, IDA 2020

R2 v1 2026-06-23T13:24:41.695Z