ByteFF-Pol:GNN参数化极化力场,溶剂性质误差较AMBER降低42%
ByteFF-Pol:GNN参数化极化力场,溶剂性质误差较AMBER降低42%
【免费下载链接】byteff2项目地址: https://ai.gitcode.com/hf_mirrors/ByteDance-Seed/byteff2
ByteFF-Pol 是一个由图神经网络(GNN,按分子结构组织数据的网络)参数化的极化力场模型,在高层级量子化学数据上训练,无需实验数据校准。它面向做分子动力学模拟的研究者(药物发现、电解液配方等方向),用经典方法的速度给出接近量子化学精度的小分子液体与电解质性质预测。
模型权重与训练配置已随 ByteFF2 仓库开源,克隆方式见下文"怎么用上"一节。
旧方案卡在哪:传统力场与QM计算的三个硬伤
传统力场(如 AMBER)的参数多由实验数据拟合而来。换到训练时没见过的新化学体系,参数外推性不足,模拟结果容易失真。
其次是形式问题:传统力场普遍采用"固定电荷",每个原子的电荷锁死为一个常数,无法描述分子间的电荷转移——而电荷转移在离子溶剂化这类极性环境中是常态。
另一端,从头算量子化学方法(如 DFT)精度足够,但计算量随体系规模快速增长,上千原子体系、更长时间尺度的模拟很难直接落地。
它是怎么做到的:GNN 替代参数表
GNN 如何替代人工调参
模型采用 EGT 架构(一种坐标对称不变的 GNN Transformer 变体,开源配置中可见 16 个注意力头),直接读取分子拓扑并输出相互作用参数,无需人工逐对定义键参数、角参数。
类比:相当于不再查一张"参数表"来算相互作用,而是让一个经验丰富的化学家看一眼分子结构,直接写出相互作用的方式。
效果:README 说明,其在小分子液体与电解质的热力学、输运性质上超过当前主流的传统力场与机器学习力场。
极化效应是如何实现的
非键作用块采用 Exp-6 极化形式(Exp6Pol),原子电荷与极化响应随分子环境动态输出;配置中还带有独立的锂参数(fix_li_alpha、li_damp_clip),针对电池电解液场景。
类比:像连通器里的水位——每个原子的电荷随周围环境涨落,而不是一次锁死的定值。
效果:电荷转移、极化屏蔽这类动态过程可以被直接描述,固定电荷力场做不到的场景可以覆盖。
训练数据从哪来
模型在高层级量子化学(QM)数据上训练,官方发布资料显示,配套的量子化学数据库覆盖 10 万种分子。
类比:相当于用一套完整的"标准科目"练过车,遇到没跑过的新路线(未见过的分子),输出依然稳定。
效果:对未见分子直接给出接近 QM 水平的预测,不需要再做实验校准。
效果与数据:精度和速度对比 AMBER、DFT 如何
以下数字均来自官方发布资料(对应论文 arXiv:2508.08575):
| 指标 | ByteFF-Pol | 对比方案 |
|---|---|---|
| 密度/粘度 MAE(100 种常见有机溶剂) | 较 AMBER 降低 42% | AMBER 力场 |
| 精度对齐从头算量化学 | 达从头算方法约 95% 精度水平 | 从头算 QM |
| 模拟速度(1000 原子溶液体系) | 较 DFT 快约 800 倍 | DFT |
| kinase 抑制剂结合能 RMSE(50 例) | ≤ 1.2 kcal/mol | 虚拟筛选实用门槛 |
| 训练内存占用 | 混合精度策略降低 60% | 常规全精度训练 |
这些数字对使用者的含义:原本要"排队等 DFT 集群"的性质计算,现在用类力场的速度即可完成,且误差落在虚拟筛选可用的范围内。
怎么用上:模型、代码与数据在哪里
仓库包含两部分开箱文件:训练好的模型权重与配置(trained_models/,含 optimal.pt 与 fftrainer_config_in_use.yaml)、验证数据(valid_data/,含 processed_data_shard0.pkl 与 dataset_config.yaml)。
git clone https://gitcode.com/hf_mirrors/ByteDance-Seed/byteff2注意:模型推理与训练代码、使用示例在配套代码仓库中(README 指向 ByteDance-Seed/byteff2 项目),克隆本仓库后按代码仓库文档接入。
硬件方面,官方白皮书称单张 A100 GPU 可完成百万原子体系的微秒级模拟;模型运行的最低资源要求请以代码仓库文档为准。
许可为 CC BY-NC 4.0(仅限非商业用途)。引用可参考:《Bridging Quantum Mechanics to Organic Liquid Properties via a Universal Force Field》(arXiv:2508.08575)。
哪些场景最受益
- 💊 药物发现:结合能预测达到虚拟筛选精度要求,可直接作为先导化合物优化的定量依据
- 🔋 电池电解液:可模拟 Li+ 溶剂化壳层动态演化,为高导电率电解液配方提供计算依据
- 🧪 溶剂物性筛选:批量预测常见有机溶剂的密度、粘度,减少实验测量工作量
- 🧬 生物大分子延伸:按官方方向,可拓展到蛋白质构象动力学、DNA-药物分子相互作用等极化体系
官方已确认的下一步,是将模型继续优化到蛋白质构象动力学、DNA-药物相互作用等生物体系。对读者而言:仓库中权重与配置开箱即用,克隆后即可在自己关注的分子体系上验证精度,或作为对比其他力场的基线。
【免费下载链接】byteff2项目地址: https://ai.gitcode.com/hf_mirrors/ByteDance-Seed/byteff2
创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
