by Shibo Li, Wei Xing, Mike Kirby and Shandian Zhe
This is the python implementation of the our papr Multi-Fidelity Bayesian Optimization via Deep Neural Networks. Bayesian Optimization(BO) is a popular framework to optimize black-box functions. In many applications, the objective function can be evaluated at multiple fidelities to enable a trade-off between the cost and accuracy. We preposed DNN-MFBO that can flexibly capture all kinds of complicated relationships between the fidelities to improve the objective function estimation and hence the optimization performance, please refer our paper for more details.
We tested our code with python 3.6 on Ubuntu 18.04. Our implementation relies on TensorFlow 1.15. Other packages include scikit-learn for data standarlization and hdf5stroage for saving the results to mat file. Please use pip or conda to install those dependencies.
pip install hdf5storage
pip install scikit-learn
We highly recommend to use Docker to freeze the running experiments. We attach our docker build file.
Please find the details of running configuration from run-*.sh
DNN-MFBO is released under the MIT License, please refer the LICENSE for details
Please cite our work if you would like to use the code
@article{li2020multi,
title={Multi-Fidelity Bayesian Optimization via Deep Neural Networks},
author={Li, Shibo and Xing, Wei and Kirby, Robert and Zhe, Shandian},
journal={Advances in Neural Information Processing Systems},
volume={33},
year={2020}
}
If you have any questions, please email me at shibo 'at' cs.utah.edu, or create an issue on github. The datasets used of the last two applications in the paper are proprietary datasests, please contact our data provider if you are interested. We attach examples of three well-known sythetic multi-fidelity functions from https://www.sfu.ca/~ssurjano/index.html