Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

FLAML: A Fast and Lightweight AutoML Library

About

We study the problem of using low computational cost to automate the choices of learners and hyperparameters for an ad-hoc training dataset and error metric, by conducting trials of different configurations on the given training data. We investigate the joint impact of multiple factors on both trial cost and model error, and propose several design guidelines. Following them, we build a fast and lightweight library FLAML which optimizes for low computational resource in finding accurate models. FLAML integrates several simple but effective search strategies into an adaptive system. It significantly outperforms top-ranked AutoML libraries on a large open source AutoML benchmark under equal, or sometimes orders of magnitude smaller budget constraints.

Chi Wang, Qingyun Wu, Markus Weimer, Erkang Zhu• 2019

Related benchmarks

TaskDatasetResultRank
Hyperparameter OptimizationHyperFD
Rank@166.42
10
Hyperparameter OptimizationPD1
Normalized Accuracy @ 11.28
9
Hyperparameter OptimizationHPO-B
nAcc@177.84
9
Cell Painting morphology predictionBBBC036 SMILES-based
MSE3.4217
7
Cell Painting morphology predictionBBBC036 Plate-based
MSE2.4918
7
Cell Painting morphology predictionBBBC047 SMILES-based split
MSE2.8126
7
Cell Painting morphology predictionBBBC047 Plate-based
MSE2.6158
7
Cell Painting morphology predictionCPG0016 (SMILES-based split)
MSE1.2865
7
Cell Painting morphology predictionCPG0016 Plate-based
MSE1.1206
7
Showing 9 of 9 rows

Other info

Follow for update