arXiv:2502.02016v1 宣告类型: cross 摘要: 生成晶格数据分布是一个重要但具有挑战性的任务，这是因为晶体具有独特的周期物理对称性。基于扩散的方法在建模晶格分布方面早期显示出潜在的可行性。最近，引入了贝叶斯流网络来聚合噪声的潜在变量，从而形成一个方差减少的参数空间，该参数空间已被证明适用于具有结构约束的欧几里得数据分布建模（Song等，2023）。受到这一进展的启发，我们致力于探索其在建模位于非欧几里得流形上的变量（例如，处于晶体结构中的变量）中的潜力，从而克服存在的理论难题。我们提出了CrysBFN，这是一种新颖的晶体生成方法，通过提出周期性贝叶斯流来实现，本质上与原始基于高斯的BFN不同，显示出非单调熵动力学。为成功实现周期性贝叶斯流的概念，CrysBFN 结合了一种新的熵条件机制，并在与时间条件机制的对比中展示了其重要性。在晶体从头生成和晶体结构预测任务上的广泛实验证明了CrysBFN 的优越性，且在所有基准测试中均能实现新的最先进水平。令人惊讶的是，我们发现CrysBFN 在采样效率方面取得了显著改进，例如，在MP-20数据集上与之前的基于扩散的方法相比，10次迭代的速度提高了约100倍（10步vs. 2000步网络前向）。代码可在https://github.com/wu-han-lin/CrysBFN获取。