AlphaD3M: Machine Learning Pipeline Synthesis

We introduce AlphaD3M, an automatic machine learning (AutoML) system based on meta reinforcement learning using sequence models with self play. AlphaD3M is based on edit operations performed over machine learning pipeline primitives providing explainability. We compare AlphaD3M with state-of-the-art...

Full description

Saved in:
Bibliographic Details
Published in:arXiv.org 2021-11
Main Authors: Drori, Iddo, Krishnamurthy, Yamuna, Rampin, Remi, Raoni de Paula Lourenco, Ono, Jorge Piazentin, Cho, Kyunghyun, Silva, Claudio, Freire, Juliana
Format: Article
Language:English
Subjects:
Machine learning
Online Access:Get full text
Tags: Add Tag
No Tags, Be the first to tag this record!
Description
Summary:We introduce AlphaD3M, an automatic machine learning (AutoML) system based on meta reinforcement learning using sequence models with self play. AlphaD3M is based on edit operations performed over machine learning pipeline primitives providing explainability. We compare AlphaD3M with state-of-the-art AutoML systems: Autosklearn, Autostacker, and TPOT, on OpenML datasets. AlphaD3M achieves competitive performance while being an order of magnitude faster, reducing computation time from hours to minutes, and is explainable by design.
ISSN:2331-8422