arXiv:2502.02016v1 宣告类型: cross 摘要: 生成晶体数据分布的建模是一个重要但具有挑战性的任务，由于晶体的独特的周期性物理对称性。基于扩散的方法在建模晶体分布方面显示出了早期的潜力。最近，贝叶斯流网络被引入以聚合嘈杂的潜在变量，从而产生一个减小方差的参数空间，该参数空间已被证明在具有结构约束的欧几里得数据分布建模中具有优势（Song et al., 2023）。受到这一工作的启发，我们寻求通过克服复杂的理论问题来解锁其在非欧几里得流形上的潜力，例如晶体结构内的变量。我们提出了CrysBFN，一种新颖的晶体生成方法，通过提出一种周期性的贝叶斯流，本质上与原始的基于高斯的BFN有所区别，其熵动力学是非单调的。为了成功实现周期性贝叶斯流的概念，CrysBFN集成了一个新的熵条件机制，并且在与时间条件机制相比时，其重要性得到了实证证明。在晶体从头生成和晶体结构预测任务上的广泛实验表明，CrysBFN在其所有基准测试上表现出了优越性，始终实现了新的性能最佳。令人惊讶的是，我们发现CrysBFN在采样效率上取得了显著的改进，例如，在MP-20数据集上与先前的基于扩散的方法相比，10步网络前向传播的速度提高了约100倍（2000步）。代码可在 https://github.com/wu-han-lin/CrysBFN 获取。