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 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.

By Drori, Iddo, Krishnamurthy, Yamuna, Rampin, Remi, Lourenco, Raoni de Paula, Ono, Jorge Piazentin, Cho, Kyunghyun, Silva, Claudio, Freire, JulianaResearch paper

Resource information

Authors: Drori, Iddo, Krishnamurthy, Yamuna, Rampin, Remi, Lourenco, Raoni de Paula, Ono, Jorge Piazentin, Cho, Kyunghyun, Silva, Claudio, Freire, Juliana
Research area: Artificial intelligence
Research topics: Machine Learning (cs.LG), FOS: Computer and information sciences, FOS: Computer and information sciences
Publication date: 2021
DOI: 10.48550/arxiv.2111.02508
Upstream last modified: Feb 19, 2022
License: arXiv.org perpetual, non-exclusive license
Last record update: Nov 6, 2026