arXiv:2602.12162cs.LG2026-02

AMORTIX让分子优化一步到位,无需反复调用昂贵的测试接口。

Amortized Molecular Optimization via Group Relative Policy Optimization

  • 基于图变换器的可迁移策略,单次前向传播完成优化
  • 在多靶点激酶抑制剂设计中超越基线方法,PMO基准排名第一
  • 支持多样结构约束,适合需要快速生成新分子的研究者

在结构受限的分子优化中,现有先进方法对每个新输入结构都需从头启动昂贵的测试搜索,难以扩展到大量起始结构或高成本测试场景。虽可迁移的优化策略理论上能缓解此瓶颈,但现有方法在推理时难以泛化至不同结构约束。本文提出AMORTIX,一种原生支持此类约束的可迁移图变换器模型,可在单次前向传播中完成分子结构优化,且推理阶段无需调用任何测试接口。该领域可迁移训练的核心挑战在于起始结构间优化难度差异极大。我们发现,在这种异质性下,标准强化学习方法无法稳定训练,因此提出按起始结构分组内归一化奖励以解决。在结构受限的单目标与多目标激酶抑制剂设计任务及少样本前药案例研究中进行评估。AMORTIX在目标导向的骨架修饰任务中优于所有可迁移与实例优化基线,在PMO基准上位列可迁移方法第一;前药案例进一步验证了所学修饰规则在未见过药物结构上的迁移能力。代码已开源:https://github.com/Hash-hh/AMORTIX/

原文摘要 · Abstract (English)

In structurally constrained molecular optimization, state-of-the-art methods restart an expensive oracle-driven search from scratch for every new input structure, scaling poorly to settings with many starting structures or expensive oracles. While amortized approaches that learn a transferable policy could in principle remove this bottleneck, existing methods struggle to generalize to diverse structural constraints at inference time. We present AMORTIX, an amortized Graph Transformer model that natively supports such constraints, optimizing molecular structures in a single forward pass with zero inference-time oracle calls. A central challenge for amortized training in this domain is that optimization difficulty varies drastically across starting structures. We show that, under this heterogeneity, standard reinforcement learning methods fail to stabilize training, and address this by normalizing rewards within groups of completions sharing the same starting structure. We evaluate on structurally constrained single- and multi-target kinase inhibitor design, and on a few-shot prodrug case study. AMORTIX outperforms both amortized and instance-optimization baselines on goal-directed scaffold decoration and ranks first among amortized methods on the PMO benchmark; the prodrug case study further demonstrates transfer of a learned modification rule to unseen drug structures. Code is available at https://github.com/Hash-hh/AMORTIX/.

分子生成强化学习图神经网络可迁移优化

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