arXiv:2502.02016cs.LGcs.AI2025-02ICLR被引 25

用周期性贝叶斯流生成晶体,速度比传统方法快100倍。

A Periodic Bayesian Flow for Material Generation

  • 提出周期性贝叶斯流,突破传统高斯假设限制。
  • 在MP-20数据集上实现约100倍采样加速(10步对2000步)。
  • 适合需要高效生成晶体结构的研究者或材料设计场景。

晶体生成的分布建模因晶体独特的周期性物理对称性而具有挑战性。基于扩散的方法在该领域已初显潜力。近期,贝叶斯流网络通过聚合噪声潜变量,实现了方差降低的参数空间,对具有结构约束的欧几里得数据建模具有优势(Song et al., 2023)。受此启发,我们致力于将其拓展至非欧几何流形(如晶体结构),克服理论难题。本文提出CrysBFN,一种新型晶体生成方法,核心是周期性贝叶斯流,其熵动态非单调,与原始高斯基底的BFN本质不同。为实现这一概念,CrysBFN引入新的熵条件机制,并实证其优于时间条件。在晶体从头生成和结构预测任务上的大量实验表明,CrysBFN在所有基准测试中均达到新最优。令人意外的是,其采样效率显著提升,例如在MP-20数据集上相较以往扩散方法提速约100倍(10步对2000步前向计算)。代码已开源:https://github.com/wu-han-lin/CrysBFN。

原文摘要 · Abstract (English)

Generative modeling of crystal data distribution is an important yet challenging task due to the unique periodic physical symmetry of crystals. Diffusion-based methods have shown early promise in modeling crystal distribution. More recently, Bayesian Flow Networks were introduced to aggregate noisy latent variables, resulting in a variance-reduced parameter space that has been shown to be advantageous for modeling Euclidean data distributions with structural constraints (Song et al., 2023). Inspired by this, we seek to unlock its potential for modeling variables located in non-Euclidean manifolds e.g. those within crystal structures, by overcoming challenging theoretical issues. We introduce CrysBFN, a novel crystal generation method by proposing a periodic Bayesian flow, which essentially differs from the original Gaussian-based BFN by exhibiting non-monotonic entropy dynamics. To successfully realize the concept of periodic Bayesian flow, CrysBFN integrates a new entropy conditioning mechanism and empirically demonstrates its significance compared to time-conditioning. Extensive experiments over both crystal ab initio generation and crystal structure prediction tasks demonstrate the superiority of CrysBFN, which consistently achieves new state-of-the-art on all benchmarks. Surprisingly, we found that CrysBFN enjoys a significant improvement in sampling efficiency, e.g., ~100x speedup 10 v.s. 2000 steps network forwards) compared with previous diffusion-based methods on MP-20 dataset. Code is available at https://github.com/wu-han-lin/CrysBFN.

晶体生成贝叶斯流扩散模型高效采样

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