arXiv:2606.26657cs.LG2026-06

用强化学习策略高效筛选海量化学分子,降低计算成本。

Target-Aware Bandit Allocation for Scalable Surrogate Optimization in Chemical Space

论文配图:Target-Aware Bandit Allocation for Scalable Surrogate Optimization in Chemical Space
图 1 · 摘自论文原文
  • 将化学空间分块并视作老虎机的“筹码”,动态分配计算资源。
  • 在真实库上实现更高筛选效率,推理开销降低90%以上。
  • 适合超大规模药物分子虚拟筛选,兼顾速度与精度。

在昂贵评估条件下从巨大离散空间中识别高价值候选物是科学领域常见挑战,结构导向药物发现即为典型场景。尽管代理优化可提升采样效率,但现代分子库已达百亿至万亿规模,全库代理推断本身成为主要计算瓶颈。我们提出BOBa框架,通过多臂老虎机机制引导代理优化,避免全库推断,自适应地将计算资源分配到潜在优势的行动空间分区。将分区视为老虎机中的“臂”,BOBa在保持合理探索的同时,集中资源于经验上更有潜力的区域。真实合成按需库的实验表明,不确定性下的乐观主义策略结合有意义的空间划分,对有效分配推断与评估至关重要。研究揭示了筛查性能与代理推断成本间的可调权衡,支持当前库的实用优化,并为超大规模库虚拟筛选提供了可行路径。

原文摘要 · Abstract (English)

Identifying high-utility candidates from massive discrete spaces under expensive evaluations is a recurring challenge across the sciences, with structure-based drug discovery as a prominent example. While surrogate-based optimization can increase sample efficiency by reducing the number of expensive evaluations, modern molecular libraries have reached billions to trillions of compounds, making full-library surrogate inference itself a major computational bottleneck. We introduce BOBa, a bandit-guided surrogate optimization framework that eliminates full-library inference by adaptively allocating computation across partitions of the action space. By treating partitions as arms in a multi-armed bandit, BOBa concentrates inference and evaluations on empirically promising partitions while maintaining principled exploration. Experiments on real-world synthesis-on-demand libraries demonstrate that optimism-under-uncertainty bandits, combined with meaningful action space partitioning, are essential for effective allocation of inference and evaluations. Our findings reveal a tunable tradeoff between screening performance and surrogate inference cost, which supports practical optimization over current libraries, and establishes a viable route to ultra-large library virtual screening.

药物发现代理优化强化学习虚拟筛选

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