AI²BMD登Nature，AI生物分子模拟双突破，继AlphaFold后又一里程碑
一个月前，诺贝尔化学奖颁给AlphaFold，给全世界带来了一场认知地震。人们开始意识到，近年来，AI在解析蛋白质结构与设计方面的应用进展迅速。凭借AlphaFold和其他前沿算法，研究人员能够以惊人的速度预测和分析蛋白质的三维结构，破解了这个长期以来困扰生物学界的难题。
如今，AI在静态晶体蛋白质结构的预测上已达到实验精度——这是一个巨大的突破。但它的潜力远不止于此。科学家们正努力将AI的力量拓展至动态表征和分子相互作用的模拟领域。
最近，微软研究院开发的AI²BMD（AI-based ab initio biomolecular dynamics system，基于AI的从头算生物分子动力学系统）在Nature上发表，这一进展代表着在分子动力学模拟领域的显著突破。
效率精度两不误
生物世界的本质在于分子及其相互作用的动态变化。理解生物分子的动态和相互作用对于解读生物过程的机制以及开发生物材料和药物至关重要。然而，通过实验捕捉这些真实的运动几乎是不可能的。
生物分子动力学（MD）模拟是一种结合物理定律和数值模拟的方法，旨在应对理解生物分子动态的挑战，其有效性依赖于模拟的精度和效率。
MD模拟大致可以分为两类：经典MD和量子力学。
经典MD侧重于模拟的效率。经典MD采用了对分子系统的简化表示，能够在较长时间的构象变化上实现快速模拟。该方法于2013年获得诺贝尔奖。不过，尽管速度快，经典MD的准确性却相对较低。
量子力学则侧重于模拟的精度。量子力学方法如密度泛函理论（DFT）提供了从基础原理出发的精确计算。DFT在1998年获得诺贝尔奖，但其计算成本过高，难以处理大型生物分子系统。
为此，微软研究院一直在开发高效的方法，目标是实现具有从头算精度的生物分子模拟。经过四年的研究，AI²BMD诞生了，它能够以从头算精度高效地模拟大型生物分子，真正实现了效率精度两不误。
与标准模拟技术相比，它在生物分子模拟中实现了一个此前难以达到的精度与计算成本的平衡——AI²BMD在精度上超越了经典模拟，但其计算成本远低于DFT的要求，速度上更是快了几个数量级。
这一方法有望为生物分子建模，尤其是在蛋白质-药物相互作用等需要高精度的场景中，提供新的动力。
基于AI的从头算生物分子动力学模拟
AI²BMD能够高效地以从头算精度模拟各种全原子蛋白质，并通过极化力场明确模拟溶剂环境。
AI²BMD的流程图
AI²BMD采用了一种通用的蛋白质分片方法，将蛋白质分割为重叠的单元，从而创建了一个包含2000万快照的数据集，这是DFT级别中规模最大的数据库。
在该研究团队此前设计的ViSNet的基础上，他们使用机器学习训练了AI²BMD的势能函数。ViSnet是一种通用分子几何建模基础模型，已在《Nature Communication》上发表，并已集成到PyTorch Geometry库中。
AI²BMD利用基于ViSNet的势能函数，在每一步模拟中计算蛋白质的能量和原子力，达到从头算的精度。
通过高效的AI²BMD系统，进行了几百纳秒的动力学模拟，展示了其高效探索肽和蛋白质构象空间的能力。在这一过程中，AI²BMD推导出了与核磁共振实验一致的精确3J耦合值，并展示了蛋白质的折叠和展开过程。
通过对动力学和热力学的综合分析，AI²BMD在蛋白质折叠自由能等方面表现出与实验数据的高度一致性，同时展现了与经典MD不同的现象。
蛋白质分片方法
蛋白质由20种氨基酸构成，每种氨基酸都有一个通用的主链（由Cα、C、O、N和H组成）和一个不同的侧链（称为R基）。
二肽是指在其N端和C端分别封端了Ace和Nme基团的氨基酸。鉴于氨基酸是蛋白质的基本单元，这些二肽便被作为分片的基本单元。
二肽的结构
该研究团队基于二肽设计了一种通用的蛋白质分片方法，并据此训练了AI²BMD势能函数，从而确保了对所有蛋白质的泛化能力。
在该方法中，采用滑动窗口技术对多肽链进行切割，因此Ace-Nme片段充当两个相邻二肽之间的重叠区域。
对于多肽链的末端Cα原子，会根据其连接的C-H键长和Cα的连接方向，为其添加额外的氢原子。如果第一个或最后一个氨基酸是甘氨酸，则仅根据C–H键长添加一个连接到Cα的氢原子。如果下一个氨基酸是脯氨酸，则还根据N–H键长添加一个连接到N的氢原子，其中该N与Cδ相连。
然后，为了精确调整氢原子的位置，采用有限记忆Broyden-Fletcher-Goldfarb-Shanno拟牛顿算法来优化添加的氢原子的位置。其它部分则保持不变。
通过蛋白质分片方法，所有蛋白质可以转换为21种蛋白质单元（即20种二肽和1个Ace-Nme），这大幅减少了蛋白质单元的特定类型数量，便于数据集构建和模型训练，帮助探索完整的构象空间，避免了势能面上的空白区域，从而提升了MD模拟的泛化性、效率和稳健性。
蛋白质单元数据集
AI²BMD蛋白质单元数据集的生成过程涉及对蛋白质的基本结构单元——二肽——进行全面的构象采样。
首先，生成了初始的20种二肽和1个Ace-Nme单元，并通过旋转关键化学键（称为二面角）来捕获不同的分子形态。
每一种构型经过几何优化，确保结构合理后用于后续的从头算分子动力学（AIMD）模拟。
在模拟中，研究人员采集了大量的分子形态，并重新计算每种构型的能量和作用力，以用于机器学习模型的训练。
整个数据集涵盖了约2000万个构象，全面捕获了蛋白质单元的构象空间，为AI²BMD提供了可靠的数据支持，以实现高效而准确的分子模拟。
ViSNet作为AI²BMD的势函数
ViSNet是一种通用的几何深度学习模型，能够以原子坐标和原子序数为输入，预测势能、原子力以及多种量子化学性质。
如上图a中所示，ViSNet模型由一个嵌入块和多个堆叠的ViSNet块组成，最后接一个输出块。原子序数和坐标输入嵌入块，随后进入ViSNet块以提取和编码几何表示。这些几何表示随后通过输出块用于预测分子的能量和力。
上图b中展示了ViSNet块的结构，包括一个消息块和一个更新块。这些模块协同工作，构成了称为ViS-MP的向量标量交互消息传递机制。通过ViS-MP传递的丰富几何信息由运行时几何计算模块以线性复杂度提取。
对于每种蛋白质单元，ViSNet被训练为一个能量守恒的势能模型，能够通过预测的势能梯度导出原子间的作用力。
研究人员将蛋白质单元数据集随机分为训练集、验证集和测试集，并在不同类型的蛋白质上进行了训练和验证。训练时使用了多种优化技术来提高模型的准确性和稳定性，并利用了GPU集群进行高效的训练。
AI²BMD模拟程序
为了使用AI²BMD的势能进行模拟，该研究团队设计了一个基于原子模拟环境的AI驱动MD模拟程序。该模拟程序支持云环境，可以将计算结果定期保存到云存储，以应对长时间计算中可能出现的断点。
程序启动时，初始蛋白质结构被输入到预处理模块，在该模块中添加溶剂和离子，并对结构进行弛豫。
然后，整个模拟系统进入MD循环，即程序的核心逻辑组件。在MD循环的每次迭代中，蛋白质首先通过蛋白质分片模块被分解为片段，随后被分配到不同的计算服务器上进行能量和力的计算。
分片后的蛋白质片段会被工作调度器分配到不同的服务器上。用户可以根据系统规模和计算需求，调整调度策略，以最大化GPU的利用率，或平衡各GPU上的计算负载。
分片后的蛋白质片段和溶剂原子以异步方式发送到不同进程中的计算服务器。其中，ViSNet服务器负责基于AI的蛋白质片段计算，溶剂服务器负责溶剂分子的计算。
各服务器完成计算后，能量和力的结果会汇总并用于更新整个系统。
MD模拟的大突破
AI²BMD在分子动力学（MD）模拟领域实现了显著的突破，具体体现在以下几个方面：
（1）从头算精度：AI²BMD引入了一种
原创文章，作者：睿鸿在线，如若转载，请注明出处：https://mip.orihonor.com/2024/11/08/4831.shtml
