用扩散模型加速纳米流体密度预测,省数据又快十倍。
Generative Quasi-Continuum Modeling of Confined Fluids at the Nanoscale
- 用条件去噪扩散模型预测受限流体的力分布,连接连续理论
- 在训练外的通道宽度上实现从头算精度的密度分布重建
- 比传统模拟快数个数量级,训练数据需求大幅降低
我们提出一种数据高效、多尺度框架,用于预测纳米尺度受限流体的密度分布。尽管精确密度估计需要远超从头算分子动力学(AIMD)可及的时间尺度,机器学习分子动力学(MLMD)虽能以更低计算成本实现从头算精度的力预测,但仍受限于飞秒级时间步长,难以获得准确的长期平均密度。为此,我们提出基于条件去噪扩散概率模型(DDPM)的准连续方法,以少量AIMD数据提取的噪声力为条件,预测沿受限方向的长期力分布。预测的平滑力通过Nernst-Planck方程与连续理论关联,揭示密度行为。我们在石墨烯纳米狭缝间水的系统上验证该框架,发现对训练域外通道宽度的密度分布仍可恢复从头算精度。相比AIMD和MLMD,该方法在运行时间上实现数量级加速,且所需训练数据显著少于以往工作。
原文摘要 · Abstract (English)
We present a data-efficient, multiscale framework for predicting the density profiles of confined fluids at the nanoscale. While accurate density estimates require prohibitively long timescales that are inaccessible by ab initio molecular dynamics (AIMD) simulations, machine-learned molecular dynamics (MLMD) offers a scalable alternative, enabling the generation of force predictions at ab initio accuracy with reduced computational cost. However, despite their efficiency, MLMD simulations remain constrained by femtosecond timesteps, which limit their practicality for computing long-time averages needed for accurate density estimation. To address this, we propose a conditional denoising diffusion probabilistic model (DDPM) based quasi-continuum approach that predicts the long-time behavior of force profiles along the confinement direction, conditioned on noisy forces extracted from a limited AIMD dataset. The predicted smooth forces are then linked to continuum theory via the Nernst-Planck equation to reveal the underlying density behavior. We test the framework on water confined between two graphene nanoscale slits and demonstrate that density profiles for channel widths outside of the training domain can be recovered with ab initio accuracy. Compared to AIMD and MLMD simulations, our method achieves orders-of-magnitude speed-up in runtime and requires significantly less training data than prior works.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。