Periodic Bayesian Flow Networks with Additive Accuracy
Peijia Lin, Zihan Zhang, zhangrui zhao, Shaohao Rui, Junyi An, Yun-Fei Shi, Fenglei Cao, Weijie Ma, Yutong Lu
摘要
Generating periodic data---such as fractional atomic coordinates in crystal structures and phase patterns in compressive light-field (CLF) displays---is challenging because wrap-around boundaries complicate probabilistic modeling and learning. While Bayesian Flow Networks (BFNs) offer a powerful generative framework with strictly additive accuracy in Euclidean space, existing periodic adaptations typically sacrifice additivity and become sensitive to schedule heuristics. We introduce PeriodicBFN, which embeds each periodic scalar into a two-dimensional unit-circle representation and performs Gaussian Bayesian updates in the resulting Cartesian space, thereby restoring strictly additive accuracy. To address invariance in periodic generative modeling, we further derive a Rao--Blackwellized objective that analytically marginalizes global periodic translations, producing a translation-invariant target with reduced gradient variance. Experiments on crystal structure prediction and multi-layer phase synthesis for CLF displays demonstrate improved training stability and strong performance. To our knowledge, this is the first work to extend periodic-data generative modeling to phase synthesis for modern glasses-free 3D display systems.
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
它引用的顶会 Paper10
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 被引用 35,902 次
- Score-Based Generative Modeling through Stochastic Differential EquationsYang Song, Jascha Sohl-Dickstein, Diederik P. Kingma, Abhishek Kumar 等ICLR 2021 · 被引用 1,270 次
- Torsional Diffusion for Molecular Conformer GenerationBowen Jing, Gabriele Corso, Jeffrey Chang, Regina Barzilay 等NeurIPS 2022 · 被引用 413 次
- Crystal Diffusion Variational Autoencoder for Periodic Material GenerationTian Xie, Xiang Fu, Octavian-Eugen Ganea, Regina Barzilay 等ICLR 2022 · 被引用 394 次
- Riemannian Score-Based Generative ModellingValentin De Bortoli, Emile Mathieu, Michael J. Hutchinson, James Thornton 等NeurIPS 2022 · 被引用 306 次
相关 Paper
- A Periodic Bayesian Flow for Material GenerationHanlin Wu, Yuxuan Song, Jingjing Gong, Ziyao Cao 等ICLR 2025
- Full-Atom Peptide Design via Riemannian-Euclidean Bayesian Flow NetworksHao Qian, Shikui Tu, Lei XuAAAI 2026
- Variational Bayesian Flow Network for Graph GenerationYida Xiong, Jiameng Chen, Xiuwen Gong, Jia Wu 等ICML 2026
- Periodic Activation Functions Induce StationarityLassi Meronen, Martin Trapp, Arno SolinNeurIPS 2021 · 被引用 31 次
- DynGFN: Towards Bayesian Inference of Gene Regulatory Networks with GFlowNetsLazar Atanackovic, Alexander Tong, Bo Wang, Leo J. Lee 等NeurIPS 2023 · 被引用 37 次
