arXiv:2409.18582cs.LGcs.AI2024-09ICLR被引 3

用博弈论方法解决组合型贝叶斯优化难题,加速蛋白质设计

Optimistic Games for Combinatorial Bayesian Optimization with Application to Protein Design

  • 构建变量间的合作博弈,以均衡点替代传统优化目标
  • 在长度达20^X的蛋白空间中快速发现高活性变体
  • 适合大规模组合优化场景,如药物与蛋白设计

贝叶斯优化(BO)是一种通过序列交互优化昂贵黑箱函数的强大框架。然而在药物发现、电路设计、神经网络架构搜索等重要问题中,这些函数定义在巨大且无结构的组合空间上,导致现有BO算法因难以最大化获取函数而不可行。为此,我们提出GameOpt,一种基于博弈论的组合型贝叶斯优化新方法。GameOpt在不同优化变量间建立合作博弈,选择上置信界获取函数的博弈均衡点。这些均衡点是各变量无偏离动机的稳定配置,类似连续域中的局部最优。关键在于,这使我们能将组合域复杂度分解为独立决策集,实现对大规模组合空间的高效扩展。我们在具有挑战性的蛋白质设计问题上验证了GameOpt性能,使用四个真实蛋白数据集进行测试。每个蛋白最多可有20^X种可能构型(X为蛋白长度),标准BO方法在此失效。而我们的方法迭代选取信息丰富的蛋白构型,在极短时间内发现高活性蛋白变体,优于其他基线方法。

原文摘要 · Abstract (English)

Bayesian optimization (BO) is a powerful framework to optimize black-box expensive-to-evaluate functions via sequential interactions. In several important problems (e.g. drug discovery, circuit design, neural architecture search, etc.), though, such functions are defined over large $\textit{combinatorial and unstructured}$ spaces. This makes existing BO algorithms not feasible due to the intractable maximization of the acquisition function over these domains. To address this issue, we propose $\textbf{GameOpt}$, a novel game-theoretical approach to combinatorial BO. $\textbf{GameOpt}$ establishes a cooperative game between the different optimization variables, and selects points that are game $\textit{equilibria}$ of an upper confidence bound acquisition function. These are stable configurations from which no variable has an incentive to deviate$-$ analog to local optima in continuous domains. Crucially, this allows us to efficiently break down the complexity of the combinatorial domain into individual decision sets, making $\textbf{GameOpt}$ scalable to large combinatorial spaces. We demonstrate the application of $\textbf{GameOpt}$ to the challenging $\textit{protein design}$ problem and validate its performance on four real-world protein datasets. Each protein can take up to $20^{X}$ possible configurations, where $X$ is the length of a protein, making standard BO methods infeasible. Instead, our approach iteratively selects informative protein configurations and very quickly discovers highly active protein variants compared to other baselines.

贝叶斯优化蛋白质设计组合优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。