Bayesian BiLO: Bilevel Local Operator Learning for Efficient Uncertainty Quantification of Bayesian PDE Inverse Problems with Low-Rank Adaptation
Ray Zirui ZhangiD, Christopher E. MilesiD, Xiaohui S. XieiD, John S. LowengrubiD
Abstract
Uncertainty quantification in PDE inverse problems is essential in many applications. Scientific machine learning and AI enable data-driven learning of model components while preserving physical structure, and provide the scalability and adaptability needed for emerging imaging technologies and clinical insights. We develop a Bilevel Local Operator Learning framework for Bayesian inference in PDEs (B-BiLO). At the upper level, we sample parameters from the posterior via Hamiltonian Monte Carlo, while at the lower level we fine-tune a neural network via low-rank adaptation (LoRA) to approximate the solution operator locally. B-BiLO enables efficient gradient-based sampling without synthetic data or adjoint equations and avoids sampling in high-dimensional weight space, as in Bayesian neural networks, by optimizing weights deterministically. We analyze errors from approximate lower-level optimization and establish their impact on posterior accuracy. Numerical experiments across PDE models, including tumor growth, demonstrate that B-BiLO achieves accurate and efficient uncertainty quantification. MSC Classification: 65M32, 62F15, 68T07
02 /Specification & Classification
Click any tag to browse matching papers- Citations
- 1
- Approach
Reproducibility ledger
Nobody has reported trying to reproduce this paper yet. Did you replicate an experiment or run the code? Your report saves the next lab a week.
Sign in to file a reproducibility report.
Related papers
- BiLO: Bilevel Local Operator Learning for PDE Inverse Problems
Ray Zirui Zhang, Christopher E. Miles, … John S. Lowengrub·Journal of Computational PhysicsPeer-reviewed·Jan 11, 2026
- Posterior marginalization accelerates Bayesian inference for dynamical systems
Elba Raimúndez, Michael Fedders, Jan Hasenauer·iSciencePeer-reviewed·Sep 28, 2023
- Scalable approximate Bayesian inference for particle tracking data
Ruoxi Sun, Liam Paninski·bioRxiv·Mar 5, 2018
- Efficient Bayesian inference for ordinary differential equation models from experimental data with uncertain measurement times
Jakob Vanhoefer, Vanessa Nakonecnij, … Jan Hasenauer·bioRxiv·May 13, 2026
- How forgiving are M/EEG inverse solutions to noise level misspecification? An excursion into the BSI-Zoo.
Anuja Negi, Stefan Haufe, … Ali B. Hashemi·IEEE Transactions on Biomedical EngineeringPeer-reviewed·Jan 1, 2026
- Bayesian Neural Networks for Cellular Image Classification and Uncertainty Analysis
Giacomo Deodato, Christopher Ball, Xian Zhang·bioRxiv·May 1, 2020