arXiv:2411.09388cs.LGcond-mat.dis-nn2024-11综述被引 1

对比三类生成模型在分子数据上的表现,助你选对工具。

A survey of probabilistic generative frameworks for molecular simulations

  • 按流形与扩散两类框架,比较模型生成能力。
  • 低维非对称数据用神经样条流,高维简单数据用条件流匹配。
  • 扩散模型适合复杂低维分子构象生成,有实用参考价值。

生成式人工智能在分子科学中广泛应用,但针对分子数据的生成模型性能基准测试仍不足。本文系统介绍两类概率生成框架:基于流的模型与扩散模型,并选取神经样条流(Neural Spline Flows)、条件流匹配(Conditional Flow Matching)和去噪扩散概率模型(Denoising Diffusion Probabilistic Models)作为代表,评估其在可调维度、复杂度与模态不对称性数据集上的精度、计算成本与生成速度。实验使用高斯混合模型及通过分子动力学模拟生成的Aib₉肽二面角分布数据。结果表明:(i)神经样条流在低维数据中模式不对称性捕捉最佳;(ii)条件流匹配在高维低复杂度数据上表现最优;(iii)去噪扩散模型适用于低维高复杂度场景。研究为分子生成任务的模型选择提供分类框架与实证依据。

原文摘要 · Abstract (English)

Generative artificial intelligence is now a widely used tool in molecular science. Despite the popularity of probabilistic generative models, numerical experiments benchmarking their performance on molecular data are lacking. In this work, we introduce and explain several classes of generative models, broadly sorted into two categories: flow-based models and diffusion models. We select three representative models: Neural Spline Flows, Conditional Flow Matching, and Denoising Diffusion Probabilistic Models, and examine their accuracy, computational cost, and generation speed across datasets with tunable dimensionality, complexity, and modal asymmetry. Our findings are varied, with no one framework being the best for all purposes. In a nutshell, (i) Neural Spline Flows do best at capturing mode asymmetry present in low-dimensional data, (ii) Conditional Flow Matching outperforms other models for high-dimensional data with low complexity, and (iii) Denoising Diffusion Probabilistic Models appears the best for low-dimensional data with high complexity. Our datasets include a Gaussian mixture model and the dihedral torsion angle distribution of the Aib\textsubscript{9} peptide, generated via a molecular dynamics simulation. We hope our taxonomy of probabilistic generative frameworks and numerical results may guide model selection for a wide range of molecular tasks.

生成模型分子模拟概率建模模型对比

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