Commit
·
8d3de70
1
Parent(s):
d802af4
update readme
Browse files- README.md +30 -0
- trained_models/fftrainer_config_in_use.yaml +67 -0
- trained_models/optimal.pt +3 -0
README.md
CHANGED
|
@@ -1,3 +1,33 @@
|
|
| 1 |
---
|
| 2 |
license: apache-2.0
|
|
|
|
|
|
|
|
|
|
| 3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
license: apache-2.0
|
| 3 |
+
tags:
|
| 4 |
+
- chemistry
|
| 5 |
+
- biology
|
| 6 |
---
|
| 7 |
+
# ByteFF2
|
| 8 |
+
|
| 9 |
+
This repository contains the model used for the paper [Bridging Quantum Mechanics to Organic Liquid Properties via a Universal Force Field](https://arxiv.org/abs/2508.08575)。
|
| 10 |
+
|
| 11 |
+
[ByteFF-Pol](https://arxiv.org/abs/2508.08575) is a polarizable force field parameterized by a graph neural network (GNN), trained on high-level quantum mechanics (QM) data, thus eliminating the need for experimental calibration. ByteFF-Pol achieves exceptional accuracy in predicting the thermodynamic and transport properties of small-molecule liquids and electrolytes, outperforming SOTA traditional and ML force fields
|
| 12 |
+
|
| 13 |
+
# Trained Models
|
| 14 |
+
The `trained_models`` folder contains the trained model for ByteFF-Pol and its corresponding configuration (.yaml) file.
|
| 15 |
+
|
| 16 |
+
# How to use
|
| 17 |
+
Code and examples are available in the [byteff2](https://github.com/ByteDance-Seed/byteff2) repository.
|
| 18 |
+
|
| 19 |
+
## Citation
|
| 20 |
+
If you find ByteFF-Pol is useful for your research and applications, feel free to give us a star ⭐ or cite us using:
|
| 21 |
+
|
| 22 |
+
```bibtex
|
| 23 |
+
|
| 24 |
+
@misc{zheng2025bridgingquantummechanicsorganic,
|
| 25 |
+
title = {Bridging Quantum Mechanics to Organic Liquid Properties via a Universal Force Field},
|
| 26 |
+
author = {Tianze Zheng and Xingyuan Xu and Zhi Wang and Xu Han and Zhenliang Mu and Ziqing Zhang and Sheng Gong and Kuang Yu and Wen Yan},
|
| 27 |
+
year = {2025},
|
| 28 |
+
eprint = {2508.08575},
|
| 29 |
+
archivePrefix = {arXiv},
|
| 30 |
+
primaryClass = {physics.comp-ph},
|
| 31 |
+
url = {https://arxiv.org/abs/2508.08575}
|
| 32 |
+
}
|
| 33 |
+
```
|
trained_models/fftrainer_config_in_use.yaml
ADDED
|
@@ -0,0 +1,67 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
model:
|
| 2 |
+
ff_block:
|
| 3 |
+
- act: gelu
|
| 4 |
+
out_mlp_dims:
|
| 5 |
+
- 256
|
| 6 |
+
- 4
|
| 7 |
+
post_mlp_dims:
|
| 8 |
+
- 256
|
| 9 |
+
- 256
|
| 10 |
+
- 1
|
| 11 |
+
pre_mlp_dims:
|
| 12 |
+
- 256
|
| 13 |
+
- 256
|
| 14 |
+
- 3
|
| 15 |
+
tanh_output: 20.0
|
| 16 |
+
type: MMBondedConj
|
| 17 |
+
- act: gelu
|
| 18 |
+
out_mlp_dims:
|
| 19 |
+
- 256
|
| 20 |
+
- 3
|
| 21 |
+
pre_mlp_dims:
|
| 22 |
+
- 256
|
| 23 |
+
- 256
|
| 24 |
+
- 3
|
| 25 |
+
type: ChargeVolume
|
| 26 |
+
- act: gelu
|
| 27 |
+
charge14: 0.5
|
| 28 |
+
disp_damping_factor: 120.0
|
| 29 |
+
fix_li_alpha: 1.0e-06
|
| 30 |
+
ind14: 0.5
|
| 31 |
+
li_damp_clip: 0.2
|
| 32 |
+
out_mlp_dims:
|
| 33 |
+
- 256
|
| 34 |
+
- 3
|
| 35 |
+
pol_damping_factor: 0.39
|
| 36 |
+
pre_mlp_dims:
|
| 37 |
+
- 256
|
| 38 |
+
- 256
|
| 39 |
+
- 3
|
| 40 |
+
s12: 1.5
|
| 41 |
+
type: Exp6Pol
|
| 42 |
+
vdw14: 0.5
|
| 43 |
+
graph_block:
|
| 44 |
+
feature_layer:
|
| 45 |
+
act: gelu
|
| 46 |
+
atom_embedding_dim: 32
|
| 47 |
+
edge_mlp_dims:
|
| 48 |
+
- 64
|
| 49 |
+
- 256
|
| 50 |
+
- 2
|
| 51 |
+
node_mlp_dims:
|
| 52 |
+
- 64
|
| 53 |
+
- 256
|
| 54 |
+
- 2
|
| 55 |
+
gnn_layer:
|
| 56 |
+
act: gelu
|
| 57 |
+
at_channels: 16
|
| 58 |
+
ffn_dims:
|
| 59 |
+
- 256
|
| 60 |
+
- 2
|
| 61 |
+
gnn_dims:
|
| 62 |
+
- 256
|
| 63 |
+
- 256
|
| 64 |
+
- 4
|
| 65 |
+
gnn_type: EGT
|
| 66 |
+
heads: 16
|
| 67 |
+
jk: cat
|
trained_models/optimal.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ae47a6e6860b563908a2e0a83d4a3f6adc1c36b48f544e2241d24066d43d539c
|
| 3 |
+
size 111892932
|