Despite the widespread applications of machine learning force field (MLFF) on solids and small molecules, there is a notable gap in applying MLFF to complex liquid electrolytes. In this work, we introduce BAMBOO (ByteDance AI Molecular Simulation Booster), a novel framework for molecular dynamics (MD) simulations, with a demonstration of its capabilities in the context of liquid electrolytes for lithium batteries. We design a physics-inspired graph equivariant transformer architecture as the backbone of BAMBOO to learn from quantum mechanical simulations. Additionally, we pioneer an ensemble knowledge distillation approach and apply it on MLFFs to improve the stability of MD simulations. Finally, we propose the density alignment algorithm to align BAMBOO with experimental measurements. BAMBOO demonstrates state-of-the-art accuracy in predicting key electrolyte properties such as density, viscosity, and ionic conductivity across various solvents and salt combinations. Our current model, trained on more than 15 chemical species, achieves the average density error of 0.01 g/cm$^3$ on various compositions compared with experimental data. Moreover, our model demonstrates transferability to molecules not included in the quantum mechanical dataset. We envision this work as paving the way to a "universal MLFF" capable of simulating properties of common organic liquids.
翻译:尽管机器学习力场(MLFF)已在固体和小分子领域得到广泛应用,但在复杂液态电解质中的应用仍存在显著空白。本文提出BAMBOO(字节跳动人工智能分子模拟助推器)——一种面向分子动力学(MD)模拟的新型框架,并展示了其在锂电池液态电解质场景中的能力。我们设计了基于物理启发的图等变Transformer架构作为BAMBOO的核心,以学习量子力学模拟数据。此外,我们首创性地将集成知识蒸馏方法应用于MLFF,提升了MD模拟的稳定性。最终,我们提出密度对齐算法,使BAMBOO与实验测量结果保持一致。在预测不同溶剂-盐组合的关键电解质性质(如密度、粘度、离子电导率)时,BAMBOO展现出最先进的精度。当前模型基于15种以上化学成分训练,在多种组分下与实验数据相比,平均密度误差为0.01 g/cm$^3$。同时,模型展现了向量子力学数据集未包含分子的迁移能力。我们期望该工作为能够模拟常见有机液体性质的"通用MLFF"铺平道路。