Work

AlphaD3M: Machine Learning Pipeline Synthesis

Year: 2021
Type: preprint
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.
Institution: New York University
Language: English
Open Access status: green
OpenAlex ID: W3210057840
Cites: 16
Cited by: 20
Related to: 20