AlphaD3M: Machine Learning Pipeline Synthesis

Abstract

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.

arXiv commentICML 2018 AutoML Workshop

References in corpus (3)

• OpenML: networked science in machine learning
• Mastering Chess and Shogi by Self-Play with a General Reinforcement Learning Algorithm
• Solving the Rubik's Cube Without Human Knowledge

Cited by in corpus (6)

• Automatic selection of clustering algorithms using supervised graph embedding
• AutoML to Date and Beyond: Challenges and Opportunities
• Efficient AutoML Pipeline Search with Matrix and Tensor Factorization
• PipelineProfiler: A Visual Analytics Tool for the Exploration of AutoML Pipelines
• Solving Constrained CASH Problems with ADMM
• Towards A Domain-Customized Automated Machine Learning Framework For Networks and Systems