DPA-3.1-MPtrj
Predictions
Convex hull distance prediction errors projected onto elements
Trained By
Model Info
- Model Version v0.3
- Model Type UIP
- Targets EFSG
- Openness OSOD
- Train Task S2EFS
- Test Task IS2RE-SR
- Trained for Benchmark Yes
Training Set
MPtrj: 1.58M structures from 146k materials
Description
DPA3 is an advanced interatomic potential leveraging the message passing architecture, implemented within the DeePMD-kit framework, available on GitHub. Designed as a large atomic model (LAM), DPA3 is tailored to integrate and simultaneously train on datasets from various disciplines, encompassing diverse chemical and materials systems across different research domains. Its model design ensures exceptional fitting accuracy and robust generalization both within and beyond the training domain. Furthermore, DPA3 maintains energy conservation and respects the physical symmetries of the potential energy surface, making it a dependable tool for a wide range of scientific applications.
Hyperparameters
- max_force:
0.05 - max_steps:
500 - ase_optimizer:
"FIRE" - cell_filter:
"ExpCellFilter" - n_layers:
24 - e_rcut:
6 - a_rcut:
4.5 - n_dim:
128 - e_dim:
64 - a_dim:
32 - optimizer:
"AdamW" - round1:
{"loss":"MSE","loss_weights":{"energy":"0.2 -> 20","force":"100 -> 20","virial":"0.02 -> 1"},"initial_learning_rate":0.001,"learning_rate_schedule":"ExpLR - start_lr=0.001, decay_steps=5000, stop_lr=0.00001","training_steps":2000000} - round2:
{"loss":"Huber","loss_weights":{"energy":15,"force":1,"virial":2.5},"initial_learning_rate":0.0002,"learning_rate_schedule":"ExpLR - start_lr=0.0002, decay_steps=5000, stop_lr=0.00001","training_steps":1000000} - batch_size:
64 - epochs:
120 - graph_construction_radius:
6 - max_neighbors:
null