改进TPE算法,让黑箱组合优化更高效
Tree-Structured Parzen Estimator Can Solve Black-Box Combinatorial Optimization More Efficiently
- 将类别型核函数与数值型融合,引入距离结构
- 新方法在合成任务中用更少评估次数找到更好解
- 适合化学、生物等组合优化场景的超参搜索
树形帕尔森估计器(TPE)是一种广泛应用于超参数优化(HPO)的方法,常见于深度学习领域。然而,其在黑箱组合优化中的应用尚未被充分探索。本文提出一种针对组合优化的高效算法:首先将类别型核函数与数值型核函数统一,引入类别变量的距离结构;随后设计优化策略以应对大规模组合搜索空间,显著降低核函数计算的时间复杂度。在合成问题上的实验表明,所提方法在更少评估次数下即可找到更优解。该算法已集成至开源超参优化框架Optuna中。
原文摘要 · Abstract (English)
Tree-structured Parzen estimator (TPE) is a versatile hyperparameter optimization (HPO) method supported by popular HPO tools. Since these HPO tools have been developed in line with the trend of deep learning (DL), the problem setups often used in the DL domain have been discussed for TPE such as multi-objective optimization and multi-fidelity optimization. However, the practical applications of HPO are not limited to DL, and black-box combinatorial optimization is actively utilized in some domains, e.g., chemistry and biology. As combinatorial optimization has been an untouched, yet very important, topic in TPE, we propose an efficient combinatorial optimization algorithm for TPE. In this paper, we first generalize the categorical kernel with the numerical kernel in TPE, enabling us to introduce a distance structure to the categorical kernel. Then we discuss modifications for the newly developed kernel to handle a large combinatorial search space. These modifications reduce the time complexity of the kernel calculation with respect to the size of a combinatorial search space. In the experiments using synthetic problems, we verified that our proposed method identifies better solutions with fewer evaluations than the original TPE. Our algorithm is available in Optuna, an open-source framework for HPO.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。