用推理时优化生成更符合实验数据的蛋白质构象集合。
Inference-time optimization for experiment-grounded protein ensemble generation
- 在潜在空间优化提升构象集合似然,避免依赖采样步数和初始结构。
- 结合AlphaFold3与力场先验,生成符合实验数据的玻尔兹曼加权集合。
- 可改进构象多样性与物理合理性,适合药物设计与结构生物学研究。
蛋白质功能依赖于动态构象集合,但当前生成模型如AlphaFold3常无法生成与实验数据一致的集合。现有实验引导方法通过调控反向扩散过程改进结果,但受限于固定采样步长和对初始化敏感,常产生热力学不合理结构。本文提出一种通用的推理时优化框架:首先在潜在表示空间优化以最大化集合对数似然,消除对扩散长度的依赖并减少初始化偏差,同时易于引入外部约束;其次提出新采样方案,融合AlphaFold3结构先验与力场先验,采样其乘积分布以平衡实验似然。实验显示该框架显著优于现有方法,在X射线晶体学与核磁共振数据上提升构象多样性、物理能量与数据吻合度,部分结果优于已存入PDB的结构。此外,通过最大化ipTM分数进行推理时优化发现,扰动AlphaFold3嵌入会人为提高模型置信度,暴露当前评估指标的漏洞,其改进或有助于降低结合剂工程中的误报率。
原文摘要 · Abstract (English)
Protein function relies on dynamic conformational ensembles, yet current generative models like AlphaFold3 often fail to produce ensembles that match experimental data. Recent experiment-guided generators attempt to address this by steering the reverse diffusion process. However, these methods are limited by fixed sampling horizons and sensitivity to initialization, often yielding thermodynamically implausible results. We introduce a general inference-time optimization framework to solve these challenges. First, we optimize over latent representations to maximize ensemble log-likelihood, rather than perturbing structures post hoc. This approach eliminates dependence on diffusion length, removes initialization bias, and easily incorporates external constraints. Second, we present novel sampling schemes for drawing Boltzmann-weighted ensembles. By combining structural priors from AlphaFold3 with force-field-based priors, we sample from their product distribution while balancing experimental likelihoods. Our results show that this framework consistently outperforms state-of-the-art guidance, improving diversity, physical energy, and agreement with data in X-ray crystallography and NMR, often fitting the experimental data better than deposited PDB structures. Finally, inference-time optimization experiments maximizing ipTM scores reveal that perturbing AlphaFold3 embeddings can artificially inflate model confidence. This exposes a vulnerability in current design metrics, whose mitigation could offer a pathway to reduce false discovery rates in binder engineering.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。