基于流匹配的反应路径生成:MolGEN框架实现过渡态与产物的统一高效采样

超能小量子
2026-07-29 17:28:27
人工智能
化学科学
算法解析

化学反应机理的阐明依赖于对过渡态(TS)和产物的高效生成。传统方法如攀爬图像微推弹性带法(CI-NEB)和生长弦法虽然提供了强大的计算框架,但构建一个中等规模的反应网络往往需要数千个基元步骤和数百万次DFT单点计算,计算瓶颈极为突出。


近年来,扩散模型作为分子生成的前沿方法,通过学习逆向随机扩散过程将简单先验分布转化为复杂数据分布。TSDiffOA-ReactDiff等方法已被用于生成反应网络中各基元反应的过渡态。然而,扩散模型依赖随机微分方程(SDE),其固有的随机噪声增加了训练难度、降低了采样效率,且条件采样需要复杂的任务特定引导方案。


流匹配(Flow Matching)提供了一种确定性替代方案。它学习常微分方程(ODE)的速度场,将先验分布连续传输到目标分布,以平滑、解析定义的流替代SDE公式,具有训练稳定、轨迹可控、采样快速等优势。本文提出的MolGEN框架正是基于这一范式,首次实现了由生成模型同时驱动过渡态和产物采样的端到端反应网络生成。




MolGEN框架设计


MolGEN采用条件流匹配框架,通过最优传输路径将标准高斯分布映射到化学分布。其核心架构基于消息传递神经网络(MPNN)。


Image Fig.1 | 该过程将所有三种状态的键信息整合到反应消息中。类似地,对于反应产物生成任务,我们将反应物消息与产物插值消息拼接以形成反应消息。随后,模型将这些反应消息通过消息传递层来预测演化分布的速度场。MolGEN从标准高斯分布开始初始化,因此它还包含随机性。


如图1所示,对于过渡态生成任务,MolGEN将过渡态插值点的消息与反应物-产物对的消息拼接形成"反应消息",将三种状态的键信息整合在一起。对于产物生成任务,则拼接反应物消息与产物插值点消息。模型随后通过消息传递层预测驱动分布演化的速度场。


MolGEN从标准高斯分布起始,因此在确定性前向传播中仍包含随机性,其模拟的概率路径与扩散模型类似——均实现从高斯分布到复杂目标分布的变换。训练采用两个损失函数:流匹配损失用于衡量预测速度场与最优传输速度场之间的偏差;条件修正Jensen-Shannon散度损失则度量预测概率路径与最优传输概率路径的差异。


这一设计兼具扩散模型和流匹配两种范式的优势:通过最优传输路径采样实现约0.8秒的快速推理,同时能以更高精度生成过渡态和产物。


过渡态生成性能


研究者在Transition1x数据集上训练并评估MolGEN。该数据集包含11,961个反应,涉及至多7个重原子(CNO)和23个原子的体系,所有数据均以分子式分层划分。


MolGENTSDiffOA-ReactDiff(扩散模型)和ReactOT(双端流匹配模型)进行了对比。关键区别在于:ReactOTTS预测视为从反应物-产物中点到唯一TS的确定性映射,而MolGEN从高斯先验出发生成随机分布。这意味着MolGEN的生成发生在势能面的驻点附近,能够通过并行生成30个独立样本并选取最低能量构型来提升预测质量。


Image Table 1 | 均方根距离(RMSD)和能垒误差的统计摘要


1总结了各方法在均方根偏差(RMSD)、势垒高度误差、有效过渡态比例和更新TS比例等指标上的表现。MolGEN将过渡态几何预测误差降低至扩散模型的一半,同时采样时间缩减至亚秒级。值得注意的是,MolGEN还能发现比参考数据具有更低势垒的替代过渡态构型,这是双端流匹配模型所不具备的能力,凸显了其揭示前所未知反应通道的潜力。


反应产物生成


MolGEN利用相同的骨干网络进行反应产物生成,通过将反应物信息作为条件输入,从高斯先验中采样产物结构。该方法在top-k准确率指标上达到了与当前最优序列模型相当的水平,同时天然保持了质量守恒和电子守恒约束——这是序列模型难以保证的优势。


反应网络生成:γ-酮氢过氧化物分解案例


MolGEN最具实际意义的应用在于无模板的反应网络探索。研究者将其应用于γ-酮氢过氧化物(KHP)的分解网络,这是一个大气化学中的重要体系。


Image


Image


 


 


Table 2 | γ-酮氢过氧化物(KHP)第一步反应的比较


2展示了MolGEN与传统弦方法在KHP分解第一步中的对比。传统方法需要1,156次量子化学计算,而MolGEN仅需12次量子化学评估即可产生更多有效过渡态,并识别出一条比文献报道最低势垒更低的新分解路径。这一结果充分展示了生成式方法在反应网络探索中的巨大效率优势。


MolGEN的工作流程为:首先从给定反应物生成可能的产物,然后为每对反应物-产物生成过渡态,通过迭代展开即可构建完整的反应网络——整个过程无需模板、无需手工规则,完全由生成推理驱动。


总结与展望


MolGEN的主要贡献在于:(1)首次将流匹配范式成功应用于反应机理生成的完整流程;(2)以确定性最优传输路径替代随机扩散,实现亚秒级采样和更高几何精度;(3)统一了过渡态和产物的生成框架,使端到端反应网络探索成为可能;(4)在实际化学体系中展现了数量级的计算效率提升。


该研究的局限性在于当前训练数据仅涉及小分子体系(至多7个重原子),向更大分子体系和更复杂反应类型的扩展仍需验证。此外,生成的过渡态仍需少量量子化学验证以确保物理合理性。


对于从事计算化学和机器学习交叉研究的学者,MolGEN表明流匹配作为一种通用的生成范式,在化学空间探索中具有比扩散模型更优的效率-精度平衡,值得在更广泛的分子设计和反应发现任务中进一步探索。





参考文献:Tuo, P., Chen, J. & Li, J. Flow matching for reaction pathway generation. Nat Commun (2026). https://doi.org/10.1038/s41467-026-75654-w


102
1
0
0
关于作者
相关文章
  • 基于残基相互作用的GAT模型:识别非同源蛋白口袋隐藏药物靶点 ...
    了解详情 
  • 伊辛机求解无线资源分配问题:从QUBO建模到MIMO波束赋形与无人机 ...
    组合优化是无线通信资源分配中的核心挑战,许多实际问题如信道分配、功率控制、用户调度等均可归 ...
    了解详情 
  • Chemeleon2:强化学习引导生成模型发现新型晶体结构 ...
    本研究将强化学习引入晶体生成模型,用奖励函数替代似然目标,让模型从"模仿已知材料"转向"主动 ...
    了解详情 
  • 量子计算解决湍流非线性难题:专用量子计算机实现复杂流场高效模 ...
    湍流作为典型的非线性复杂系统,始终是经典计算“逐点推进”逻辑的算力天花板。北京大 ...
    了解详情 
领取成功
本月5个550bit真机配额已发放给您,配额将在2个月后到期,请及时使用哦~
活动中心
联系我们
二维码
返回顶部
返回
活动中心

完成任务,轻松获取真机配额

×
每日必做
新手任务
长期任务
其他任务
快速回复 返回顶部 返回列表
玻色有奖小调研
填写问卷,将免费赠送您1个1000bit真机配额
(单选) 您是从哪个渠道得知我们的?*
您是从哪个社交媒体得知我们的?*
您是通过哪个学校的校园宣讲得知我们的呢?
取消

提交成功

真机配额已发放到您的账户,可前往【云平台】查看

量子AI开发者认证

考核目标

开发者能够成功搭建Kaiwu-PyTorch-Plugin项目基础环境,并成功运行示例代码,根据示例提示,输出指定的值并填写至相应的输入框中。

通过奖励

5个一年效期的1000量子比特真机配额

专属「量子AI开发者」社区认证标识

开发者权益

每月固定权益:5个550量子比特真机配额
前往考核

第一步

按照README提示成功安装Kaiwu-PyTorch-Plugin库环境依赖
前往GitHub

第二步

运行 community-assessment 分支下的 run_rbm.py 代码示例

第三步

理解示例代码,手动打印并填写如下数值:

正相采样的状态

负相采样的状态

正相的能量值

负相的能量值

*

提交答案

开发者权益

每月固定权益:5个550量子比特的真机配额

恭喜您完成考核

您将获得量子AI开发者认证标识及考核奖励

1000 bit*5

配额

Quantum AI Developer Certification

Assessment Objectives

Developers should successfully set up the basic environment for the Kaiwu-PyTorch-Plugin project, run the QBM-VAE sample code, and calculate the correct FID value based on the random seed value provided by the system.

Pass Rewards

10 quotas for 550-qubit real quantum machines with a one-year validity period

Exclusive "Quantum AI Developer" Community Certification Badge

Developer Benefits

Fixed Monthly Benefits: 5 quotas for 550-qubit real quantum machines
Proceed to Assessment

Step 1

Install the environment dependencies for the Kaiwu-PyTorch-Plugin library according to the README instructions
Go to GitHub

Step 2

Replace the Seed Value

Your seed value is

Step 3

Enter the FID Value You Calculated

*

Submit Answer

Developer Benefits

Fixed Monthly Benefits: 5 quotas of 550-qubit real machines

Congratulations on Completing the Assessment

You will receive the Quantum AI Developer Certification Badge and Assessment Rewards

550bit*10

Quotas