arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2201.05978cs.LGcs.SYeess.SY

离散仿真优化用于机器学习方法超参数调优

Discrete Simulation Optimization for Tuning Machine Learning Method Hyperparameters

Indian Institute of Technology Delhi(印度理工学院德里分校)

Varun Ramamohan, Shobhit Singhal, Aditya Raj Gupta, Nomesh Bhojkumar Bolia

AI总结:

本研究采用离散仿真优化方法(KN排序与选择及随机标尺随机搜索)调优机器学习超参数,在多种模型上优于或媲美hyperopt和mango等先进库,兼顾统计保证与计算效率。

摘要:

机器学习(ML)方法被广泛应用于大多数技术领域,如图像识别、产品推荐、财务分析、医学诊断和预测性维护。实现机器学习方法的一个重要方面涉及控制其学习过程,以最大化所考虑方法的性能。超参数调优是选择一组合适的机器学习方法参数的过程,这些参数控制其学习过程。在本工作中,我们展示了使用离散仿真优化方法(如排序与选择(R&S)和随机搜索)来识别使机器学习方法性能最大化的超参数集。具体而言,我们使用KN R&S方法和随机标尺随机搜索方法及其一种变体来实现此目的。我们还为应用KN方法构建了理论基础,该方法通过解空间枚举以统计保证确定最优解。相比之下,随机标尺方法渐近收敛到全局最优,并且计算开销较小。我们将这些方法应用于各种机器学习模型,包括用于时间序列预测和图像分类的深度神经网络模型。我们将这些方法的应用与最先进的超参数优化库(如$hyperopt$和$mango$)进行基准测试。KN方法始终优于$hyperopt$的随机搜索(RS)和Parzen估计器树(TPE)方法。随机标尺方法优于$hyperopt$的RS方法,并在统计上与$hyperopt$的TPE方法和$mango$算法相比具有可比性能。

英文摘要:

Machine learning (ML) methods are used in most technical areas such as image recognition, product recommendation, financial analysis, medical diagnosis, and predictive maintenance. An important aspect of implementing ML methods involves controlling the learning process for the ML method so as to maximize the performance of the method under consideration. Hyperparameter tuning is the process of selecting a suitable set of ML method parameters that control its learning process. In this work, we demonstrate the use of discrete simulation optimization methods such as ranking and selection (R&S) and random search for identifying a hyperparameter set that maximizes the performance of a ML method. Specifically, we use the KN R&S method and the stochastic ruler random search method and one of its variations for this purpose. We also construct the theoretical basis for applying the KN method, which determines the optimal solution with a statistical guarantee via solution space enumeration. In comparison, the stochastic ruler method asymptotically converges to global optima and incurs smaller computational overheads. We demonstrate the application of these methods to a wide variety of machine learning models, including deep neural network models used for time series prediction and image classification. We benchmark our application of these methods with state-of-the-art hyperparameter optimization libraries such as $hyperopt$ and $mango$. The KN method consistently outperforms $hyperopt$'s random search (RS) and Tree of Parzen Estimators (TPE) methods. The stochastic ruler method outperforms the $hyperopt$ RS method and offers statistically comparable performance with respect to $hyperopt$'s TPE method and the $mango$ algorithm.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑