面向分子动力学应用的物理信息神经网络软件
化学物理
2020-11-24 v3 机器学习
计算物理
摘要
我们开发了一种名为 PND 的新型微分方程求解软件,它基于物理信息神经网络(physics-informed neural network, PINN)用于分子动力学模拟器。基于 Pytorch 提供的自动微分技术,我们的软件允许用户灵活地将原子运动方程、初始与边界条件以及守恒定律作为损失函数来实现以训练网络。PND 配备了一个并行分子动力学(MD)引擎,以便用户检验和优化损失函数设计、不同的守恒定律与边界条件以及超参数,从而加速基于 PINN 的分子应用开发。
引用
@article{arxiv.2011.03490,
title = {Physics-informed Neural-Network Software for Molecular Dynamics Applications},
author = {Taufeq Mohammed Razakh and Beibei Wang and Shane Jackson and Rajiv K. Kalia and Aiichiro Nakano and Ken-ichi Nomura and Priya Vashishta},
journal= {arXiv preprint arXiv:2011.03490},
year = {2020}
}