让分子设计瞄准整体构象分布,而非单一结构。
Boltzmann-Expected Molecular Design with Decoupled Annealing Flows

- 将分子3D设计重构为玻尔兹曼期望问题,用双流模型解耦图与坐标生成。
- 在GEOM-Drugs上实现目标半径和表面积的稳定逼近,对大分子更优。
- 支持多目标权衡与高阶矩设计,可生成特定构象偏好分子。
分子设计中多数三维属性(如自由能、形状描述符)是分子图在三维构象上的玻尔兹曼分布期望。现有模型将每个性质绑定于单一结构,忽略其背后的构象集合。本文提出玻尔兹曼期望设计,并基于去耦退火流(DECAF)实现:将图与坐标的联合分布分解为两个条件流模型——以图条件的流 $p(xackslashmid ext{G})$ 作为玻尔兹曼模拟器,以坐标条件的流 $p( ext{G}ackslashmid x)$ 从三维信息中生成新图。通过交替优化两流,并采用模拟退火接受准则,其评分函数基于 $p(xackslashmid ext{G})$ 抽样的集合统计量,使集合特征成为设计目标。该循环无需重训练即可切换目标。在GEOM-Drugs数据集上,集合感知优化使平均回转半径与溶剂可及表面积持续向目标逼近,而单构象优化在大药物分子上表现下降;且该方法可扩展至多目标权衡,首次实现3D生成模型中的高阶矩设计——联合优化分布方差与偏度,生成具有特定构象分布偏好的柔性分子,经全原子分子动力学模拟验证其构象分布特性。
原文摘要 · Abstract (English)
Most 3D properties relevant to molecular design, including free energies and shape descriptors, are $\textit{expectations}$ over the Boltzmann distribution over 3D configurations of a molecular graph. However, existing property-guided generative models tie each property to a single structure, ignoring the underlying ensemble. We recast 3D molecular design as $\textbf{Boltzmann-expected design}$ and realise it with $\textbf{DECAF}$ (Decoupled Annealing Flows), which factorise the joint distribution over graphs and coordinates into two conditional flow models: a graph-conditioned flow $p(x\mid\mathcal{G})$, acting as a $\textit{Boltzmann emulator}$, and a coordinate-conditioned flow $p(\mathcal{G}\mid x)$, proposing new graphs from 3D information. By alternating the two flows, DECAF optimises molecular graphs with a simulated-annealing acceptance rule whose scoring function is evaluated on ensembles drawn from $p(x\mid\mathcal{G})$, making ensemble statistics, not single-conformer properties, the design target. The resulting loop requires no retraining to change objectives. On GEOM-Drugs, we show that ensemble-aware optimisation produces graphs whose mean radius of gyration and solvent-accessible surface area consistently shift toward targets, while single-conformer optimisation degrades on larger drug-like molecules where Boltzmann distributions are broadest. DECAF extends to multi-objective trade-offs and, uniquely among 3D generative models, to $\textbf{higher-moment design}$: jointly optimising an ensemble property's variance and skewness to produce flexible molecules biased to a prescribed conformational regime: we verify the conformational distributions of these higher-moment designs with all-atom MD simulations.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。