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
Abstract
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.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 368207af-fd0d-4333-a7d5-9e97c06d918bBuilds on10
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 35,902 citations
- Score-Based Generative Modeling through Stochastic Differential EquationsYang Song, Jascha Sohl-Dickstein, Diederik P. Kingma, Abhishek Kumar et al.ICLR 2021 · 1,270 citations
- Torsional Diffusion for Molecular Conformer GenerationBowen Jing, Gabriele Corso, Jeffrey Chang, Regina Barzilay et al.NeurIPS 2022 · 413 citations
- Crystal Diffusion Variational Autoencoder for Periodic Material GenerationTian Xie, Xiang Fu, Octavian-Eugen Ganea, Regina Barzilay et al.ICLR 2022 · 394 citations
- Riemannian Score-Based Generative ModellingValentin De Bortoli, Emile Mathieu, Michael J. Hutchinson, James Thornton et al.NeurIPS 2022 · 306 citations
Related papers
- A Periodic Bayesian Flow for Material GenerationHanlin Wu, Yuxuan Song, Jingjing Gong, Ziyao Cao et al.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 et al.ICML 2026
- Periodic Activation Functions Induce StationarityLassi Meronen, Martin Trapp, Arno SolinNeurIPS 2021 · 31 citations
- DynGFN: Towards Bayesian Inference of Gene Regulatory Networks with GFlowNetsLazar Atanackovic, Alexander Tong, Bo Wang, Leo J. Lee et al.NeurIPS 2023 · 37 citations
