通过潜在 Kronecker 结构实现高斯过程学习曲线预测的尺度化
机器学习
2024-10-15 v1 机器学习
摘要
AutoML 中的关键任务是将机器学习模型的学习曲线联合建模,作为模型超参数和训练进程的函数。虽然高斯过程 (Gaussian processes, GPs) 适用于此任务,但朴素的 GPs 需要 的时间和 的空间,其中 为超参数配置数, 为每个超参数的学习曲线观测数。由于缺失学习曲线值导致早停不兼容,通常不兼容的 Kronecker 结构推理效率较低。我们引入潜在 Kronecker 结构,以利用高效乘积核同时处理缺失值。具体而言,我们将观察值联合协方差矩阵解释为潜在 Kronecker 乘积的投影。结合迭代线性求解器和结构化矩阵向量乘法,我们的方法仅需 的时间和 的空间。我们展示了该 GP 模型在学习曲线预测任务中可匹配 Transformer 的性能。
关键词
引用
@article{arxiv.2410.09239,
title = {Scaling Gaussian Processes for Learning Curve Prediction via Latent Kronecker Structure},
author = {Jihao Andreas Lin and Sebastian Ament and Maximilian Balandat and Eytan Bakshy},
journal= {arXiv preprint arXiv:2410.09239},
year = {2024}
}
备注
Bayesian Decision-making and Uncertainty Workshop at NeurIPS 2024