0% found this document useful (0 votes)
15 views9 pages

Auto ML

This paper investigates the current state of Automated Machine Learning (AutoML) tools, evaluating their performance and comparing their functionalities across various datasets. It highlights the challenges faced by ML engineers and the need for automation in repetitive tasks such as data preprocessing, model selection, and hyperparameter optimization. The study also assesses the trade-offs between optimization speed and accuracy, as well as the reproducibility of results across different AutoML platforms.
Copyright
© All Rights Reserved
We take content rights seriously. If you suspect this is your content, claim it here.
Available Formats
Download as PDF, TXT or read online on Scribd
0% found this document useful (0 votes)
15 views9 pages

Auto ML

This paper investigates the current state of Automated Machine Learning (AutoML) tools, evaluating their performance and comparing their functionalities across various datasets. It highlights the challenges faced by ML engineers and the need for automation in repetitive tasks such as data preprocessing, model selection, and hyperparameter optimization. The study also assesses the trade-offs between optimization speed and accuracy, as well as the reproducibility of results across different AutoML platforms.
Copyright
© All Rights Reserved
We take content rights seriously. If you suspect this is your content, claim it here.
Available Formats
Download as PDF, TXT or read online on Scribd

Towards Automated Machine Learning: Evaluation

and Comparison of AutoML Approaches and Tools


Anh Truong∗ , Austin Walters∗ , Jeremy Goodsitt∗ , Keegan Hines∗ , C. Bayan Bruss∗ , Reza Farivar∗†
∗ Applied Research, Center for Machine Learning † Department of Computer Science
Capital One University of Illinois
McLean, VA, USA Urbana-Champaign, IL, USA
{[Link], [Link], [Link] farivar2@[Link]
[Link], [Link]}@[Link]
arXiv:1908.05557v2 [[Link]] 3 Sep 2019

Abstract—There has been considerable growth and interest in Scikit-learn (2007-2010), H2O (2011), and Spark MLlib
industrial applications of machine learning (ML) in recent years. (2013) among many others. Deep Neural Network platforms
ML engineers, as a consequence, are in high demand across the have also gained popularity in the last 5 years. Tensorflow
industry, yet improving the efficiency of ML engineers remains a
fundamental challenge. Automated machine learning (AutoML) (2015), Keras (2015) and MXNet (2015) contributed to the
has emerged as a way to save time and effort on repetitive tasks wide adoption of deep learning models.
in ML pipelines, such as data pre-processing, feature engineering, During this time period it became evident to many ML
model selection, hyperparameter optimization, and prediction practitioners that extracting the best performance from ma-
result analysis. In this paper, we investigate the current state chine learning models requires substantial human expertise.
of AutoML tools aiming to automate these tasks. We conduct
various evaluations of the tools on many datasets, in different Developing good models from a dataset is almost an art
data segments, to examine their performance, and compare their form involving intuition, experience, and many tedious manual
advantages and disadvantages on different test cases. tasks to tune algorithmic parameters. The combination of
Index Terms—AutoML, automated machine learning, driver- market pressure for more ML engineers, and the tedious nature
less AI, model selection, hyperparameter optimization of developing ‘optimal’ ML solutions sparked the idea of
automating the ML tasks.
I. I NTRODUCTION AutoML’s initial effort came out of academia and ML
Automated Machine Learning (AutoML) promises major practitioners first, and later startups. One of the first at-
productivity boosts for data scientists, ML engineers and ML tempts was Auto-Weka (2013) [1] from Universities of British
researchers by reducing repetitive tasks in machine learning Columbia (UBC) and Freiburg, which utilizes algorithms
pipelines. There are currently a number of different tools provided by Weka [2]. Auto-sklearn (2014) [3] came next
and platforms (both open-source and commercially available from the University of Freiburg. TPOT [4] was developed
solutions) that try to automate these tasks. The goal of this at the University of Pennsylvania (2015). Auto-ml [5], an
paper is to address the following questions: (i) what are the open-source python package, was released in 2016 (to avoid
available ML functionalities provided by the tools; (ii) how confusion with the general term ‘AutoML’, please note the
the tools perform when facing a wide spectrum of real world spelling for this tool). Auto-sklearn, Auto-ml, and TPOT are
datasets; (iii) find the trade-off between optimization speed all built on the well-known ‘scikit-learn’ ML package. Other
and accuracy of the results; and (iv) the reproducibility of the tools followed, including Auto-Keras (2017) [6] from the
results (a.k.a. tool robustness). Texas A&M University running on top of Keras, Tensorflow
The rest of the paper is organized as follows. Section II and Scikit-learn. MLjar (2018) [7] also uses Scikit-learn, in
covers the history and background of AutoML tools. Next, in conjunction with Tensorflow. On the same timeframe, some
Section III we compare the tools’ features and functionalities startups started developing their tools for AutoML. Datarobot
on an automated ML pipeline including data preprocess- [8], [9], [10] launched its automated machine learning tool
ing, model selection, hyperparameter optimization, and model in 2015. H2O-Automl [11], [12] was introduced by the H2O
interpretation. After that, in Section IV we experimentally (2016), using ML models from the H2O platform. The H2O
evaluate the performance of a selected subset of these tools team later released their commercialized H2O-DriverlessAI
on a large variety of datasets and a range of supervised ML product (2017) [13], and SparkCognition introduced Darwin
tasks. Finally, we conclude the paper in Section V. (2018) [14] utilizing their own ML platform.
After a while, the large cloud providers and technology
II. BACKGROUND AND H ISTORY companies followed suit, offering Automated Machine Learn-
Between 1995 to 2015 many ML libraries and tools were ing as a Service (AMLaaS) or standalone products. Google
developed, spanning from Weka (1990s), RapidMiner (2001), Cloud Automl (2017) [15] runs on Google Cloud platform.
Microsoft AzureML (2018) [16] takes advantage of algorithms input to convert categorical data into integers (e.g., using label
on Azure, and Salesforce’s TransmogrifAI (2018) [17] runs on encoder) before any other transformation. TPOT and Auto-
top of Spark ML, and Uber’s Ludwig (2019) [18] runs model keras do not provide either data preprocessing or feature gener-
training on Horovod, Uber’s open-source distributed training ation steps and instead require users to manually perform data
framework. pre-processing, and only accept numerical feature matrices.
The aforementioned platforms emphasize different as-
pects of the AutoML space. For example, Darwin, H2O-
DriverlessAI and DataRobot provide the functionality of de-
B. Model Selection, Hyperparameter Optimization, and Archi-
tecting and processing time-series data. They also offer inter-
tecture Search
active UI to help customers experiment quickly with different
machine learning tasks. H2O-DriverlessAI exports a Plain Old
In this step, the extracted features from the previous step are
Java Object (POJO) or a Model ObJect Optimized (MOJO)
used to train many different types of models, each with many
for the optimized models to be easily deployed in any Java-
different sets of parameters (hyperparameter optimization),
supported platform. TPOT exports optimized code for develop-
then the best model (or an ensemble of models) is selected
ers. Auto-ml offers ‘categorical ensembling’, where segments
as the final model. Each tool supports a collection of existing
of categories in a column can have different models. Google
machine learning algorithms to build models. They include,
Cloud AutoML and Auto-keras conduct neural network search
but not limited to, Logistic Regression, tree-based algorithms,
[19], [20], for both image and text data.
SVM, and neural network models. H2O-Automl, Ludwig,
III. AUTO ML PLATFORMS ’ FEATURES AND DataRobot, Darwin, Auto-ml, Auto-sklearn, MLjar, Transmo-
FUNCTIONALITY COMPARISON : T HE COMMON PIPELINE grifAI, and TPOT all work in this fashion for supervised
methods. DataRobot, H2O-DriverlessAI and Darwin provide
additional unsupervised methods such as clustering and outlier
detection. TPOT and Darwin also utilize genetic algorithms to
iteratively select the best traits of each model and pass them
to the next generation. Google Cloud AutoML and Auto-keras
Fig. 1. The common AutoML pipeline. work differently, utilizing neural architecture search to select
the best neural network model.
For hyperparameter optimization, some of the most popular
Most AutoML tools follow a common three stage pipeline methods are grid search, random search, and Bayesian search.
illustrated in Figure 1. In general, these three components Auto-Weka uses SMAC (Sequential Model-based Algorithm
are optimized iteratively to obtain the best outcome. Figure 2 Configuration, [21]) while Auto-sklearn utilizes SMAC3, a
briefly summarizes the comparison across the surveyed tools. re-implementation of SMAC to efficiently perform Bayesian
More detailed comparisons follow in the subsequent sections. optimization. H2O-Automl and MLjar apply random search
on the parameter spaces, while H2O-DriverlessAI, Ludwig,
A. Data Preprocessing and Feature Engineering Auto-ml, TransmogrifAI and Auto-keras use both random and
Data preprocessing is typically the first task in ML Bayesian search.
pipelines. At the moment, this task is not handled very well In order to reduce time for model search and hyperparameter
by any of the AutoML tools and still requires considerable optimization, it is common to prune the parameter space. In
human intervention. In particular, this task requires data type the first approach, the tools attempt to quickly find an initial
and schema detection which have not been widely supported parameter set. Auto-sklearn and Darwin use pre-processed
among the AutoML tools. However, once data types are ‘meta-features’ from previously trained datasets, each with
identified, the tools provide appropriate feature engineering a known ‘meta-learner’. Given a target dataset, they find a
for the next component in the pipeline. TransmogrifAI seems similar dataset based on ‘meta-feature’, and use the closest
to be further ahead in this regard by supporting different ’meta-learners’ as the initial model. The second approach is
detailed data types detection (e.g., addresses, phone numbers, to use the relationship between model selection and hyper-
names, currency, etc), however this functionality appears to parameter optimization. H2O-Automl uses the combination
not be very stable on multiple datasets. H2O-Automl, H2O- of random grid search with stacked ensembles, as diversified
DriverlessAI, DataRobot, MLjar and Darwin gain some ad- models improve the accuracy of ensemble method. The third
vantage by offering the ability to detect basic data types or approach is to fix an allowed runtime for the tools to search
schemas, currently limited to numerical, categorical and time- for a best model. All AutoML tools, except Auto-ml, currently
series data. Auto-ml, Auto-sklearn, AzureML and Ludwig offer this option. The fourth approach (only applies for H2O-
are less favorable here, in the sense that they can only do Automl and Auto-sklearn) is to restrict the parameters that
feature engineering from user-input specifications, e.g. data cause a slow optimization. For example, non-linear feature
types for each column. The other tools need much more human approximation combined with KNN models is restricted as it
interaction on feature engineering. Auto-sklearn requires users dramatically slows down the optimization.
Fig. 2. Comparison table of functionality for AutoML tools. (+): commercialized tools; (∗ ): the function is not very stable, it fails for some datasets; (2∗ ):
categorical input must be converted into integers; (3∗ ): datasets have to include headers; (4∗ ): missing values must be represented as NA; (5∗ ): multiclass
classification not provided; (6∗ ): need some users’ input for dataset description such as column types; (7∗ ): ability to detect primitive data types and rich data
types such as: text (id, url, phone), numerical (integer, real); (8∗ ): advanced feature processing: bucketing of values, removing features with zero variance
or features with drift over time; (9∗ ): supervised learning includes binary classification, multiclass classification, regression; (10∗ ): unsupervised learning
includes clustering and anomaly detection; (11∗ ): model interpretation and explainability refers to techniques such as LIME, Shapley, Decision Tree Surrogate,
Partial Dependence, Individual Conditional Expectation, Lift chart, feature fit, prediction distribution plot, accuracy over time, hot spot and reason codes;
(12∗ ): confirmed by a company spokesperson, we could not find public documentation at the time of publication; In a few empty cells, it is not clear that
the functionality is provided from documentations of the tools, to the best of our knowledge.

C. Model Interpretation and Prediction Analysis


This component is currently applied to most commercial-
ized tools such as H2O-DriverlessAI, DataRobot and Darwin
whereas it is not the concentration of non-commercialized
tools. In essence, it provides detailed result representation Fig. 3. Data segments used for evaluation. Each cell is referred to as a
through model dashboards, feature importance and different ‘data segment’. For example, in the first row, ’Less than one third’ stands for
datasets with the categorical proportion less than 1/3.
visualization methods, e.g., lift chart and prediction distribu-
tion. Those tools even highlight outlier data points that the best
model was not confident in predictions, and support ‘reason
code’, LIME, Shapley, and partial dependence, etc., for better datasets are already pre-processed into the numerical features
2
model interpretation. , therefore the same data will be fed to all AutoML tools,
minimizing the risk of bias from data selection process; and (ii)
IV. E XPERIMENTAL EVALUATION guarantee a fair comparison among the tools as some do not
We evaluate a selected subset 1 of AutoML tools on nearly provide the pre-processing steps for raw datasets. In order to
300 datasets collected from Openml [22], which allows users evaluate AutoML tools on a variety of dataset characteristics,
to query data for different use cases. Detailed descriptions we selected multiple datasets according to the criteria depicted
on the datasets are given on the Table I in the Appendix. in Figure 3. For the sake of clarity, each cell in this table
The two advantages of using Openml datasets are: (i) the is referred to as a ‘data segment’, each containing datasets
with different sample sizes, feature dimensions, categorical
1 Due to the unavailability of the licensed or trial versions, we have not
evaluated most commercialized tools. Some other open-sourced tools have 2 Although there are still a few datasets containing text or non-numerical
not been evaluated due to the lack of widely supported Python wrappers. features, those are not included in this paper.
Fig. 4. Evaluation of AutoML tools on binary classification task across ten data segments (depicted in Figure 3). Each diagram refers to a data segment. All
experiments are run up to 15 minutes. Some experiments are completed faster but in some other cases, several tools cannot obtain results after that time limit.
Specifically, the percentage of experiments that did not finish in 15 minutes are: Ludwig 4%, H2O-Automl 8%, TPOT 13%, Darwin 26%, Auto-sklearn 30%

features ratio (defined as the ratio of number of categorical AutoML tools proved to be capable of hitting the 70% target
features over total number of features), missing proportion within 15 minutes for binary classification. 5 out of the 6
(proportion of samples with at least one missing feature), and tools (all but Darwin) hit 70% target for regression, and 4
class imbalance (samples in minority class vs. in majority out of 6 tools hit the 70% target in multiclass classification,
class). Each dataset is divided into two parts, one for training (TPOT nearly reaches the target, and Darwin misses the target
and another for testing with the ratio 4 : 1. All AutoML tools again). As Darwin appears to be slow in convergence, and to
are applied to the same training and testing proportions of all be fair to the other tools, it is excluded from our completion
datasets. For all evaluations, the following tools and associated target analysis. We therefore decide to run all our extensive
versions are used: Darwin 1.6, Auto-sklearn 0.5.2, Auto-keras experiments (5,400) for 15 minutes time-limits, for a total
0.4.0, Auto-ml 2.9.10, Ludwig 0.1.2, H2O-Automl [Link], of 1,350-hour EC2 run-time (which includes the overhead
TPOT 0.10.1. of benchmark harness code), where the results are detailed
In the next subsections, we will evaluate AutoML tools in Section IV-A. We then run another experiment with a
on different test cases, each with three different supervised randomized subset of our datasets for longer time limits to
learning tasks: binary classification, multiclass classification, evaluate the performance of the tools when more time is given
and regression. All experiments are run on Amazon EC2 to finish. The results of this latter experiment (3 tasks, 5 data
[Link] instances, which provide 1 Tesla K80 GPU, 4 vCPUs segments, 6 run-time periods, 7 tools, for a total of 717 EC2
(Intel Xeon E5-2686, 2.30Ghz) and 61 GiB of host memory. hours including benchmark harness overhead) are detailed in
Setting a time-limit for all experiments is not straightfor- Section IV-B. Note that the Auto-ml tool was not included in
ward. On the one hand, we would like to let each tool run the extensive experiments as it does not offer an option to limit
as long as it takes to produce the best results. On the other its run-time from a user-input value (15 minutes in our case),
hand, with 3 ML tasks, 300 datasets and 6 tools, we have it simply can only run to completion. As such, its results are
5,400 experiments to run. To keep the experiment run-time only included among the experiments in Section IV-B.
and cost to practical limits, we aim for a ‘completion target’
of 70%, i.e., we select a run-time for which all tools are able
to finish the AutoML tasks for 70% of the datasets. All the
Fig. 5. Evaluation of AutoML tools on multiclass classification task across ten data segments (depicted in Figure 3). Each diagram refers to a data segment.
All experiments are run up to 15 minutes. Some experiments are completed faster but in some other cases, several tools cannot obtain results after that time
limit. Specifically, the percentage of experiments that did not finish in 15 minutes are: Auto-keras 2%, H2O-Automl 21%, Ludwig 24%, Auto-sklearn 30%,
TPOT 35%, Darwin 51%.

A. Evaluation on multiple data segments except Ludwig and Darwin. Comparing the tools against each
In this section, we investigate the performance of the tools other, H2O-Automl and Darwin slightly outperform the rest,
across many datasets and applications (please see Table I in the however it is worth reiterating that Darwin cannot deliver re-
Appendix for the detailed descriptions on the datasets). To that sults for 26% of all datasets. Auto-sklearn and TPOT perform
end, the evaluated data is divided into ten segments (as shown slightly worse than the aforementioned tools. Auto-keras does
in Figure 3), each including ten random datasets. ‘Accuracy’ not perform as well as other tools for most datasets in binary
is the comparison metric used for binary and multiclass classification. As noted before, in this experiment, we limit
classification tasks and ‘Mean Squared Error (MSE)’ is used the optimization time to 15 minutes. Here, TPOT manages
for regression tasks. to complete and deliver results within the 15-minutes time
Figure 4 shows the performance of AutoML tools for binary limit for 87% of datasets, while Darwin and Auto-sklearn
classification task in different data segments. In this Figure, the suffer slightly higher non-delivering ratios of 26% and 30%
performance is represented in the box-whisker format, where respectively. Ludwig’s performance appears to fluctuate the
each box shows the median, and the first and third quartiles most compared to other tools.
of the performance at the two ends. Note that, for the data The performance of the tools for muticlass classification
segment with class imbalance (third row in Figure 4), F1- is illustrated in Figure 5. Here, minimal variation was found
score is used instead of the regular accuracy as it is a more when evaluating between data segments of the same categories
appropriate metric for imbalanced data. (the two graphs in each row). For this multiclass classification
It can be observed from Figure 4 that, the performance task, Auto-keras and Auto-sklearn slightly outperform the rest,
of AutoML tools fluctuate more with a larger number of even though Auto-sklearn cannot deliver results within the
categorical features, and fluctuate less with more data samples. time limit for 30% of datasets. TPOT comes next after these
This makes intuitive sense, as the tools will learn better with two tools. Finally, Ludwig, H2O-Automl and Darwin perform
more data samples, and each tool has different approaches to slightly worse than the rest.
encode categorical values that result in different performance. Figure 6 shows the performance of the tools for regression
In addition, most tools suffer from the imbalanced datasets task. The results from this task has similar trends to binary
Fig. 6. Evaluation of AutoML tools on regression task across ten data segments (depicted in Figure 3). Each diagram refers to a data segment. All experiments
are run up to 15 minutes. Some experiments are completed faster but in some other cases, several tools cannot obtain results after that time limit. Specifically,
the percentage of experiments that did not finish in 15 minutes are: Auto-keras 4%, H2O-Automl 11%, Auto-sklearn 20%, Ludwig 24%, TPOT 25%, Darwin
56%

classification on categorical features. Furthermore, the perfor- better results given more time availability. We performed var-
mance variance tends to increase for all tools when the features ious time-limit experiments for datasets with different sample
dimensions decreases, or missing proportion increases. For this sizes. Here, we randomly select a dataset given a sample size
task, H2O-Automl and Auto-sklearn slight outperform Auto- range (i.e., we pick a uniformly random dataset among all
keras and TPOT while Darwin cannot deliver results on half datasets in each sample size range) and evaluate each tools’
of the datasets. accuracy bounded by the time limits: 5 minutes, 15 minutes,
To summarize what we have seen from the three different 30 minutes, 1 hour, 2 hours and 3 hours. Since the dataset sizes
ML tasks, Auto-keras does not perform as well as other tools do not exceed one million samples, the maximum allotted time
for some datasets in binary classification. In other words, of 3 hours should allow the tools to converge. Figure 7 shows
whether Auto-keras can perform well or not (in binary clas- the results of this evaluation. As observed from the figure,
sification) depends significantly on the nature of the dataset. most tools can generally improve the performance (increase
For multiclass classification, H2O-Automl performs slightly the accuracy for classification tasks and decrease the mean-
worse than the rest. For the regression task, Auto-keras, H2O- squared error for regression task) given more time for their
Automl and Auto-sklearn outperform the rest for most data optimization. Among the tools, H2O-Automl, Auto-keras and
segments (even though Auto-sklearn struggles somewhat more Ludwig converge to the optimal performance very quickly
to complete results in the allotted 15 minutes, failing in 26% for most cases, roughly within 15 minutes. Auto-sklearn
of datasets). TPOT performs slightly worse than those three needs almost 2-3 hours to obtain reasonable results while
tools, Ludwig’s performance varies across the datasets, and TPOT converges slightly faster. Darwin appears fluctuating its
Darwin can only complete work on about half of the datasets performance even with more time for optimization.
in the allotted 15 minutes.
C. Evaluation on robustness
B. Evaluation on time limit In this evaluation, we test the robustness of AutoML tools,
Our next targeted evaluation is to explore the impact of time i.e., whether the tools deliver similar results across multiple
limit in order to investigate how quickly the tools can deliver runs on the same input datasets. For each task, we select a
the results, and whether the tools can consistently guarantee random dataset with the sample size from 10000 to 50000
Fig. 7. Evaluation of AutoML on multiple time limits. The left (middle) subgraphs show the accuracy of tools for binary (multiclass) classification. The right
subgraphs show the mean squared error of tools for regression. From top-to-bottom: each row shows a random dataset in the increasing order of the sample
size, from 1000 to 100000. Note: In the left graph of the third row, all tools except Darwin obtain the same performance although the graph displays only
the result for TPOT; in the graphs at the rows 3 & 4, column 3, all tools except Auto-keras and Auto-ml cannot deliver results due to the large number of
features, roughly 62000 and 21000, respectively; in the second graph of third column, we omit the results of Ludwig as its error is roughly 100-times larger
than the others.

Fig. 8. Evaluation of AutoML tools on robustness.

(this is a common sample size for many real-world datasets) V. C ONCLUSIONS AND F UTURE W ORK
and run each tool on it for ten different times, each times in
In this paper, we have evaluated AutoML tools on their
10 minutes. The results are illustrated in Figure 8. We observe
capabilities in the common machine learning pipeline. At
that H2O-Automl and Ludwig obtain very stable performance
the current state, different tools have different approaches
across three different tasks. Darwin, Auto-keras and Auto-ml
for model selection and hyperparameter optimization. Com-
get slightly less stable performance than H2O-Automl. TPOT
mercialized tools such as H2O-DriverlessAI, DataRobot and
and Auto-sklearn are somewhat unstable in regression task.
Darwin extend their offering functionality on the first and the
It is worth noting that even though Ludwig’s performance is
third component of the pipeline where they are able to detect
very stable, it deviates largely from others.
the data schema, run feature engineering, and analyze the
detailed results for interpretation purpose. In contrast, open
source tools focus more on the second task in the pipeline, [4] R. S. Olson, N. Bartley, R. J. Urbanowicz, and J. H. Moore, “Evaluation
which is training and selecting the best model itself. of a tree-based pipeline optimization tool for automating data science,” in
Proceedings of the Genetic and Evolutionary Computation Conference
In addition, we have evaluated tools across many datasets (GECCO) 2016. New York, NY, USA: ACM, 2016, pp. 485–492.
on different data segments. We observed that most AutoML [5] “Auto-ml: Automated machine learning for production and analytics,”
[Link] ml$, accessed: 2019-04-10.
tools obtain reasonable results in terms of their performance [6] H. Jin, Q. Song, and X. Hu, “Auto-keras: An efficient neural architecture
across many datasets. However, there is no perfect tool at the search system,” in arXiv, 2018.
current state yet, no tool managed to outperform all others [7] “Mljar,” [Link] accessed: 2019-04-
10.
on a plurality of tasks. Across the various evaluations and [8] “Datarobot usage examples,” [Link]
benchmarks we have tested, H2O-Automl, Auto-keras and datarobot-sagemaker-examples, accessed: 2019-04-10.
Auto-sklearn performed better than Ludwig, Darwin, TPOT [9] “Datarobot documentation,” [Link] ac-
cessed: 2019-04-10.
and Auto-ml. In particular, H2O-Automl slightly outperforms [10] “Datarobot python client,” [Link]
the rest for binary classification and regression, and quickly [Link]/en/v2.11.0/setup/getting{ }[Link],
converges to the optimal results. However, it suffers from low accessed: 2019-04-10.
[11] “[Link] automl github,” [Link] accessed: 2019-
performance in multiclass classification. Auto-keras is very 04-10.
stable across all tasks and performs slightly better than the [12] “[Link] automl documentation,” [Link]
rest for multiclass classification and ties with H2O-Automl for h2o-docs/[Link], accessed: 2019-04-10.
[13] “H2o-driverlessai,” [Link]
regression, but suffers from low performance in binary classi- userguide/[Link], accessed: 2019-04-10.
fication. For a production environment where the computation [14] “Darwin-sparkcognition,” [Link]
speed and performance stability are key requirements, these accessed: 2019-04-10.
[15] “Google cloud automl,” [Link] accessed:
two tools might be good candidates depending on applications 2019-04-10.
and machine learning tasks. Auto-sklearn ties with H2O- [16] “Automated machine learning with azureml,” [Link]
Automl and Auto-keras for all tasks but it is comparatively Azure/MachineLearningNotebooks/tree/master/how-to-use-azureml/
automated-machine-learning, accessed: 2019-04-10.
slower than the other two and usually requires longer run [17] “Transmogrifai,” [Link] accessed:
time. Other tools such as Ludwig, Darwin, TPOT and Auto- 2019-04-10.
ml showed more varying results depending on the dataset and [18] “Ludwig,” [Link] accessed: 2019-04-10.
[19] B. Zoph and Q. Le, “Neural architecture search with reinforcement
task. learning,” in arXiv, Nov. 2016.
Ultimately there is no one AutoML tool at this point that [20] H. Pham, M. Guan, B. Zoph, Q. Le, and J. Dean, “Efficient neural
can clearly outperform every other tool. We are at an early architecture search via parameters sharing,” in Proceedings of the
35th International Conference on Machine Learning, vol. 80, Stock-
juncture for Automated Machine Learning, and there are holmsmssan, Stockholm Sweden, 10–15 Jul 2018, pp. 4095–4104.
many innovations announced at a rapid pace. We believe as [21] F. Hutter, H. H. Hoos, and K. Leyton-Brown, “Sequential model-based
the tools mature and borrow ideas from each other, they will optimization for general algorithm configuration,” in Proceedings of the
5th International Conference on Learning and Intelligent Optimization,
gain more strength in their core task. We also observed a gap ser. LION’05, 2011, pp. 507–523.
in the AutoML tools’ support for the first and third stages [22] J. Vanschoren, J. N. van Rijn, B. Bischl, and L. Torgo, “Openml:
of the AutoML pipeline, and expect major developments to Networked science in machine learning,” ACM SIGKDD Explorations
Newsletter, vol. 15, pp. 49–60, Jun. 2014.
happen in those areas in near future.
A PPENDIX
Disclaimer: For commercialized tools, our analyses
and descriptions are consistent with our understanding
derived from publicly available documentation and product
descriptions. In some of these cases, we are unable to explore
source code and regret any factual errors that may arise.
A subsidiary of Capital One - Capital One Ventures - is an
investor in [Link]. During the course of our research we were
neither in contact with [Link] nor Capital One Ventures.

R EFERENCES

[1] L. Kotthoff, C. Thornton, H. H. Hoos, F. Hutter, and K. Leyton-


Brown, “Auto-weka 2.0: Automatic model selection and hyperparameter
optimization in weka,” The Journal of Machine Learning Research,
vol. 18, no. 1, pp. 826–830, Jan. 2017.
[2] M. Hall, E. Frank, G. Holmes, B. Pfahringer, P. Reutemann, and I. H.
Witten, “The weka data mining software: An update,” ACM SIGKDD
Explorations Newsletter, vol. 11, no. 1, pp. 10–18, 2009.
[3] M. Feurer, A. Klein, K. Eggensperger, J. Springenberg, M. Blum,
and F. Hutter, “Efficient and robust automated machine learning,” in
Advances in Neural Information Processing Systems 28, 2015, pp. 2962–
2970.
TABLE I
DATASET DESCRIPTIONS .
Binary classification Multiclass classfication Regression
Id Name Id Name Id Name
954 spectrometer 342 squash-unstored 3584 QSAR-TID-12665
23499 breast-cancer-dropped-missing-attributes-values 385 [Link] 3536 QSAR-TID-12868
862 sleuth-ex2016 48 tae 3682 QSAR-TID-100790
905 chscase-adopt 1565 heart-h 4096 QSAR-TID-30028
724 analcatdata-vineyard 1516 robot-failures-lp1 4057 QSAR-TID-10547
40978 Internet-Advertisements 1535 volcanoes-b5 197 cpu-act
983 cmc 40708 allrep 3394 QSAR-TID-20154
40649 GAMETES-Heterogeneity-20atts-1600-Het-0.4-0.2-50-EDM-2-001 40476 thyroid-allhypo 573 cpu-act
41156 ada 181 yeast 1028 SWD
40648 GAMETES-Epistasis-3-Way-20atts-0.2H-EDM-1-1 28 optdigits 558 bank32nh
959 nursery 1044 eye-movements 1594 news20
881 mv 41163 dilbert 1583 w3a
977 letter 40926 CIFAR-10-small 564 fried
734 ailerons 1536 volcanoes-b6 344 mv
357 vehicle-sensIT 119 ” BNG(cmc, nominal, 55296)” 1578 real-sim
1111 KDDCup09-appetency 255 BNG(cmc) 1588 w8a
1112 KDDCup09-churn 554 mnist-784 1591 connect-4
41150 MiniBooNE 23397 COMET-MC-SAMPLE 273 [Link]
1212 BNG(SPECTF) 148 ” BNG(zoo, nominal, 1000000)” 1191 BNG(pbc)
1369 ” BNG(kr-vs-kp, 1000, 1)” 157 RandomRBF-10-1E-3 1192 BNG(autoHorse)
875 analcatdata-chlamydia 455 cars 533 arsenic-female-bladder
472 lupus 1413 MyIris 556 analcatdata-apnea2
461 analcatdata-creditscore 1528 volcanoes-a2 689 chscase-vine2
796 cpu 11 balance-scale 551 analcatdata-michiganacc
37 diabetes 1499 seeds 678 visualizing-environmental
1590 adult 41169 helena 4553 TurkiyeStudentEvaluation
1059 ar1 4535 Census-Income 287 wine-quality
1442 MegaWatt1 188 eucalyptus 1191 BNG(pbc)
917 fri-c1-1000-25 453 analcatdata-bondrate 608 fri-c3-1000-10
806 fri-c3-1000-50 1361 ” BNG([Link], 1000, 5)” 1572 [Link]
742 fri-c4-500-100 40670 dna 1586 w6a
1116 musk 41082 USPS 315 us-crime
1563 dbworld-subjects-stemmed 313 spectrometer 422 topo-2-1
40910 Speech 1548 autoUniv-au4-2500 1584 w4a
357 vehicle-sensIT 41166 volkert 412 Phen
1157 AP-Endometrium-Kidney 1082 rsctc2010-6 3626 QSAR-TID-103437
1123 AP-Endometrium-Breast 1078 rsctc2010-2 3363 QSAR-TID-19905
4134 Bioresponse 41163 dilbert 3574 QSAR-TID-10958
1164 AP-Endometrium-Uterus 391 [Link] 3586 QSAR-TID-104390
41159 guillermo 1514 micro-mass 3955 QSAR-TID-12406
734 ailerons 1086 ovarianTumour 3266 QSAR-TID-11169
41142 christine 40996 Fashion-MNIST 3836 QSAR-TID-142
1448 KnuggetChase3 1544 volcanoes-e3 40753 delays-zurich-transport
1021 page-blocks 477 fl2000 3707 QSAR-TID-25
868 fri-c4-100-25 377 synthetic-control 4050 QSAR-TID-218
990 eucalyptus 1539 volcanoes-d2 620 fri-c1-1000-25
749 fri-c3-500-5 1087 hepatitisC 3230 QSAR-TID-10930
162 SEA(50000) 1523 vertebra-column 223 stock
874 rabe-131 1117 desharnais 3298 QSAR-TID-12163
1130 OVA-Lung 277 [Link] 1093 Brainsize
950 arsenic-female-lung 41000 jungle-chess-2pcs-endgame-panther-elephant 200 pbc
804 hutsof99-logis 155 pokerhand 534 cps-85-wages
43 haberman 380 SyskillWebert-Bands 665 sleuth-case2002
963 heart-h 457 prnn-cushings 515 baseball-team
40536 SpeedDating 379 SyskillWebert-Goats 553 kidney
992 analcatdata-broadway 148 ” BNG(zoo, nominal, 1000000)” 31 credit-g
960 postoperative-patient-data 184 kropt 203 lowbwt
72 BNG(kr-vs-kp) 4541 Diabetes130US 213 pharynx
1010 dermatology 1363 ” BNG([Link], 5000, 1)” 224 breastTumor
40680 mofn-3-7-10 130 BNG(segment) 301 ozone-level
40647 GAMETES-Epistasis-2-Way-20atts-0.4H-EDM-1-1 400 [Link] 3915 QSAR-TID-11109
804 hutsof99-logis 1520 robot-failures-lp5 3668 QSAR-TID-100071
1462 banknote-authentication 1113 KDDCup99 3925 QSAR-TID-101090
714 fruitfly 1404 ” BNG(lymph, 1000, 10)” 3760 QSAR-TID-117
719 veteran 1356 ” BNG(anneal, 5000, 10)” 193 bolts
726 fri-c2-100-5 186 braziltourism 1432 colon-cancer
933 fri-c4-250-25 74 ” BNG(letter, nominal, 1000000)” 3833 QSAR-TID-12898
808 fri-c0-100-10 1177 BNG(primary-tumor) 619 fri-c4-250-50
904 fri-c0-1000-50 375 JapaneseVowels 3625 QSAR-TID-11000
40994 climate-model-simulation-crashes 401 [Link] 3522 QSAR-TID-100629
55 hepatitis 171 primary-tumor 200 pbc
56 vote 327 bridges 222 echoMonths
172 shuttle-landing-control 328 bridges 232 fishcatch
757 meta 40966 MiceProtein 516 pbcseq
802 pbcseq 2 anneal 524 pbc
1114 KDDCup09-upselling 5 arrhythmia 566 meta
1000 hypothyroid 7 audiology 231 hungarian
470 profb 57 hypothyroid 315 us-crime
38 sick 378 ipums-la-99-small 1072 qqdefects-numeric
51 heart-h 381 ipums-la-98-small 41021 Moneyball
40713 dis 71 ” BNG([Link], nominal, 1000000)” 383 [Link]
1142 OVA-Endometrium 1086 ovarianTumour 384 [Link]
951 arsenic-male-lung 388 [Link] 385 [Link]
1018 ipums-la-99-small 39 ecoli 387 [Link]
40910 Speech 279 [Link] 388 [Link]
1056 mc1 183 abalone 395 [Link]
977 letter 40498 wine-quality-white 397 [Link]
1001 sponge 1402 ” BNG(lymph, 1000, 1)” 398 [Link]
450 analcatdata-lawsuit 30 page-blocks 400 [Link]
1045 kc1-top5 1541 volcanoes-d4 40596 slashdot
803 delta-ailerons 1554 autoUniv-au7-500 31 credit-g
920 fri-c2-500-50 61 iris 273 [Link]
739 sleep 1548 autoUniv-au4-2500 300 isolet
1486 nomao 40996 Fashion-MNIST 386 [Link]
1167 pc1-req 46 splice 390 [Link]
1125 AP-Omentum-Prostate 6 letter 392 [Link]
787 witmer-census-1980 1493 one-hundred-plants-texture 393 [Link]
481 biomed 1385 ” BNG(letter, 10000, 5)” 394 [Link]
935 fri-c1-250-10 1041 gina-prior2 396 [Link]
763 fri-c0-250-10 147 ” BNG(waveform-5000, nominal, 1000000)” 399 [Link]

You might also like