[2503.03178] Active operator learning with predictive uncertainty quantification for partial differential equations

[2503.03178] Active operator learning with predictive uncertainty quantification for partial differential equations

arXiv - Machine Learning 4 min read Article

Summary

The paper presents a lightweight predictive uncertainty quantification method for neural operators in solving partial differential equations, enhancing accuracy and efficiency in scientific applications.

Why It Matters

As neural operators become more prevalent in solving partial differential equations (PDEs), understanding their predictive accuracy and uncertainty is crucial for reliable deployment in scientific contexts. This research addresses the computational challenges of existing uncertainty quantification methods, providing a faster and more efficient alternative that can significantly improve data efficiency and accuracy in real-time applications.

Key Takeaways

  • Proposes a lightweight predictive uncertainty quantification method for neural operators.
  • Demonstrates unbiased uncertainty estimates and accurate out-of-distribution predictions.
  • Offers significant speed improvements in inference times, reducing evaluation by over five times.
  • Extends the framework to Fourier Neural Operators and other operator networks.
  • Highlights the application of predictive uncertainties in Bayesian optimization and active learning.

Computer Science > Machine Learning arXiv:2503.03178 (cs) [Submitted on 5 Mar 2025 (v1), last revised 25 Feb 2026 (this version, v4)] Title:Active operator learning with predictive uncertainty quantification for partial differential equations Authors:Nick Winovich, Mitchell Daneker, Lu Lu, Guang Lin View a PDF of the paper titled Active operator learning with predictive uncertainty quantification for partial differential equations, by Nick Winovich and 3 other authors View PDF HTML (experimental) Abstract:With the increased prevalence of neural operators being used to provide rapid solutions to partial differential equations (PDEs), understanding the accuracy of model predictions and the associated error levels is necessary for deploying reliable surrogate models in scientific applications. Existing uncertainty quantification (UQ) frameworks employ ensembles or Bayesian methods, which can incur substantial computational costs during both training and inference. We propose a lightweight predictive UQ method tailored for Deep operator networks (DeepONets) that also generalizes to other operator networks. Numerical experiments on linear and nonlinear PDEs demonstrate that the framework's uncertainty estimates are unbiased and provide accurate out-of-distribution uncertainty predictions with a sufficiently large training dataset. Our framework provides fast inference and uncertainty estimates that can efficiently drive outer-loop analyses that would be prohibitively expensive ...

Related Articles

Yupp shuts down after raising $33M from a16z crypto's Chris Dixon | TechCrunch
Machine Learning

Yupp shuts down after raising $33M from a16z crypto's Chris Dixon | TechCrunch

Less than a year after launching, with checks from some of the biggest names in Silicon Valley, crowdsourced AI model feedback startup Yu...

TechCrunch - AI · 4 min ·
Machine Learning

[R] Fine-tuning services report

If you have some data and want to train or run a small custom model but don't have powerful enough hardware for training, fine-tuning ser...

Reddit - Machine Learning · 1 min ·
Machine Learning

[D] Does ML have a "bible"/reference textbook at the Intermediate/Advanced level?

Hello, everyone! This is my first time posting here and I apologise if the question is, perhaps, a bit too basic for this sub-reddit. A b...

Reddit - Machine Learning · 1 min ·
Machine Learning

[D] ICML 2026 review policy debate: 100 responses suggest Policy B may score higher, while Policy A shows higher confidence

A week ago I made a thread asking whether ICML 2026’s review policy might have affected review outcomes, especially whether Policy A pape...

Reddit - Machine Learning · 1 min ·
More in Machine Learning: This Week Guide Trending

No comments

No comments yet. Be the first to comment!

Stay updated with AI News

Get the latest news, tools, and insights delivered to your inbox.

Daily or weekly digest • Unsubscribe anytime