arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.00640cs.LG

张量输出的组合带状贝叶斯优化

Combinatorial Bandit Bayesian Optimization for Tensor Outputs

Department of Industrial Engineering, Tsinghua University, Beijing 100084, China(清华大学工业工程系);College of Economics and Management, Nanjing University of Aeronautics and Astronautics, Nanjing 211106, China(南京航空航天大学经济管理学院)

Jingru Huang, Haijie Xu, Jie Guo, Manrui Jiang, Chen Zhang

AI总结:

本文提出了一种针对张量输出的组合带状贝叶斯优化方法,通过引入张量输出高斯过程和UCB获取函数,有效处理部分观测的张量输出问题,并建立了理论遗憾界。

摘要:

贝叶斯优化(BO)已被广泛用于优化昂贵且黑盒函数,但现有BO方法未解决张量输出函数的问题。为此,我们提出了一种新的张量输出BO框架。具体而言,我们首先引入一种张量输出高斯过程(TOGP),其包含两种类别的张量输出核作为张量输出函数的替代模型,能够有效捕捉张量内部的结构依赖性。基于此,我们开发了上置信界(UCB)获取函数来选择查询点。此外,我们引入了一个更具实际性和挑战性的问题设置,称为组合带状贝叶斯优化(CBBO),其中只有张量输出的一部分可以用于目标。为解决此问题,我们提出了一种张量输出CBBO方法,将TOGP扩展以处理部分观测的张量输出,并相应设计了一种新的组合多臂带状UCB2(CMAB-UCB2)准则,以顺序选择查询点和输出子集。我们为两种方法建立了理论遗憾界,保证了子线性遗憾。在合成和实际数据集上的广泛实验展示了我们方法的优越性。

英文摘要:

Bayesian optimization (BO) has been widely used to optimize expensive and black-box functions across various domains. However, existing BO methods have not addressed tensor-output functions. To fill this gap, we propose a novel tensor-output BO framework. Specifically, we first introduce a tensor-output Gaussian process (TOGP) with two classes of tensor-output kernels as a surrogate model of the tensor-output function, which can effectively capture the structural dependencies within the tensor. Based on it, we develop an upper confidence bound (UCB) acquisition function to select query points. Furthermore, we introduce a more practical and challenging problem setting, termed combinatorial bandit Bayesian optimization (CBBO), where only a subset of the tensor outputs can be selected to contribute to the objective. To tackle this, we propose a tensor-output CBBO method, which extends TOGP to handle partially observed tensor outputs, and accordingly design a novel combinatorial multi-arm bandit-UCB2 (CMAB-UCB2) criterion to sequentially select both the query points and the output subset. We establish theoretical regret bounds for both methods, guaranteeing sublinear regret. Extensive experiments on synthetic and real-world datasets demonstrate the superiority of our methods.

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

↑