论文标题

通过深内核学习进行序列预测的逐步模型选择

Stepwise Model Selection for Sequence Prediction via Deep Kernel Learning

论文作者

Zhang, Yao, Jarrett, Daniel, van der Schaar, Mihaela

论文摘要

自动化机器学习(AUTOML)中的一个基本问题是模型选择。顺序环境中的一个独特挑战是,最佳模型本身可能会随着时间的推移而变化,具体取决于每个时间点可用的特征和标签的分布。在本文中,我们提出了一种新颖的贝叶斯优化(BO)算法,以应对这种情况下的模型选择挑战。这是通过将每个时间步骤的性能视为其自己的黑框功能来完成的。为了共同有效地解决所得的多个黑框函数优化问题,我们使用深内核学习(DKL)利用了黑框函数之间的潜在相关性。据我们所知,我们是第一个为序列预测提出逐步模型选择(SMS)问题的问题,并为此目的设计和演示有效的联合学习算法。使用多个现实世界数据集,我们验证我们所提出的方法在各种序列预测任务上都优于标准BO和多目标BO算法。

An essential problem in automated machine learning (AutoML) is that of model selection. A unique challenge in the sequential setting is the fact that the optimal model itself may vary over time, depending on the distribution of features and labels available up to each point in time. In this paper, we propose a novel Bayesian optimization (BO) algorithm to tackle the challenge of model selection in this setting. This is accomplished by treating the performance at each time step as its own black-box function. In order to solve the resulting multiple black-box function optimization problem jointly and efficiently, we exploit potential correlations among black-box functions using deep kernel learning (DKL). To the best of our knowledge, we are the first to formulate the problem of stepwise model selection (SMS) for sequence prediction, and to design and demonstrate an efficient joint-learning algorithm for this purpose. Using multiple real-world datasets, we verify that our proposed method outperforms both standard BO and multi-objective BO algorithms on a variety of sequence prediction tasks.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源