arXiv:2411.01600cs.LGphysics.chem-ph2024-11被引 3

用频域分解+连续时间模型,提升分子动力学长期预测精度

Graph Fourier Neural ODEs: Modeling Spatial-temporal Multi-scales in Molecular Dynamics

  • 通过图傅里叶变换分离分子的高频局部振动与低频全局构象变化
  • 在多个真实分子系统上实现优于现有方法的长期轨迹预测准确率
  • 适合需要高精度长期模拟的分子动力学研究者

精确预测长时间分子动力学(MD)轨迹仍具挑战性,现有深度学习方法常难以保持长期模拟的保真度。我们假设其关键瓶颈在于难以捕捉跨越不同空间和时间尺度的相互作用,涵盖高频局部振动到低频全局构象变化。为此,提出图傅里叶神经微分方程(GF-NODE),结合图傅里叶变换进行空间频率分解与神经微分方程框架实现连续时间演化。具体而言,GF-NODE首先利用图拉普拉斯矩阵将分子构型分解为多组空间频率模式,再通过可学习的神经微分方程模块演化各频率成分以捕获局部与全局动态,最后通过逆图傅里叶变换重构更新后的分子几何结构。该统一流程显式建模高低频现象,更有效地捕捉长程相关性与局部波动。通过简化扩散模型的热方程分析,理论上揭示图拉普拉斯特征值决定时间动态尺度,并在真实分子动力学轨迹上通过全面实证分析验证了这一对应关系,展示了多样分子系统中的定量时空相关性。在具有挑战性的MD基准测试中,GF-NODE实现了最先进的精度,同时在长时间模拟中保持关键几何特征。结果表明,将谱分解与连续时间建模结合,显著提升了MD模拟的鲁棒性与预测能力。

原文摘要 · Abstract (English)

Accurately predicting long-horizon molecular dynamics (MD) trajectories remains a significant challenge, as existing deep learning methods often struggle to retain fidelity over extended simulations. We hypothesize that one key factor limiting accuracy is the difficulty of capturing interactions that span distinct spatial and temporal scales, ranging from high-frequency local vibrations to low-frequency global conformational changes. To address these limitations, we propose Graph Fourier Neural ODEs (GF-NODE), integrating a graph Fourier transform for spatial frequency decomposition with a Neural ODE framework for continuous-time evolution. Specifically, GF-NODE first decomposes molecular configurations into multiple spatial frequency modes using the graph Laplacian, then evolves the frequency components in time via a learnable Neural ODE module that captures both local and global dynamics, and finally reconstructs the updated molecular geometry through an inverse graph Fourier transform. By explicitly modeling high- and low-frequency phenomena in this unified pipeline, GF-NODE captures long-range correlations and local fluctuations more effectively. We provide theoretical insight through heat equation analysis on a simplified diffusion model, demonstrating how graph Laplacian eigenvalues can determine temporal dynamics scales, and crucially validate this correspondence through comprehensive empirical analysis on real molecular dynamics trajectories showing quantitative spatial-temporal correlations across diverse molecular systems. Experimental results on challenging MD benchmarks demonstrate that GF-NODE achieves state-of-the-art accuracy while preserving essential geometrical features over extended simulations. These findings highlight the promise of bridging spectral decomposition with continuous-time modeling to improve the robustness and predictive power of MD simulations.

分子动力学神经微分方程频域建模时序预测

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。