用新型生成模型高效采样大规模材料平衡态,突破传统方法规模瓶颈。
Scalable Boltzmann Generators for equilibrium sampling of large-scale materials
- 结合图神经网络与增强耦合流,基于局部环境生成样本。
- 训练更快、资源更少,可处理超千原子系统且采样效率更高。
- 适合材料模拟研究者,尤其关注大尺度相变与自由能计算。
利用生成模型对多体系统进行平衡态采样,如最早提出的玻尔兹曼生成器,因其能在一次采样中生成无偏且不相关样本而受到广泛关注。尽管在自然科学领域展现出巨大潜力,但将其扩展至大规模系统仍是重大挑战。本文提出一种新型玻尔兹曼生成器架构,专注于材料科学应用。该模型结合增强耦合流与图神经网络,使生成过程基于局部环境信息,同时支持基于能量的训练和快速推理。相比以往架构,本模型训练速度显著提升,所需计算资源大幅减少,并实现更优采样效率。关键优势在于可迁移至更大系统尺寸,实现前所未有的大规模材料模拟。我们在多种材料体系中验证了该方法:包括Lennard-Jones晶体、mW水的冰相及硅的相图,系统规模均超过一千原子。训练后的生成器能精准生成各类晶型的平衡系综,以及不同系统尺寸下的赫姆霍兹与吉布斯自由能,达到可忽略有限尺寸效应的尺度。
原文摘要 · Abstract (English)
The use of generative models to sample equilibrium distributions of many-body systems, as first demonstrated by Boltzmann Generators, has attracted substantial interest due to their ability to produce unbiased and uncorrelated samples in `one shot'. Despite their promise and impressive results across the natural sciences, scaling these models to large systems remains a major challenge. In this work, we introduce a Boltzmann Generator architecture that addresses this scalability bottleneck with a focus on applications in materials science. We leverage augmented coupling flows in combination with graph neural networks to base the generation process on local environmental information, while allowing for energy-based training and fast inference. Compared to previous architectures, our model trains significantly faster, requires far less computational resources, and achieves superior sampling efficiencies. Crucially, the architecture is transferable to larger system sizes, which allows for the efficient sampling of materials with simulation cells of unprecedented size. We demonstrate the potential of our approach by applying it to several materials systems, including Lennard-Jones crystals, ice phases of mW water, and the phase diagram of silicon, for system sizes well above one thousand atoms. The trained Boltzmann Generators produce highly accurate equilibrium ensembles for various crystal structures, as well as Helmholtz and Gibbs free energies across a range of system sizes, able to reach scales where finite-size effects become negligible.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。