4.7 Article

Scalable uncertainty quantification for deep operator networks using randomized priors

出版社

ELSEVIER SCIENCE SA
DOI: 10.1016/j.cma.2022.115399

关键词

Deep learning; Uncertainty quantification; Operator learning

资金

  1. US Department of Energy under the Advanced Scientific Computing Research program [DE-SC0019116]
  2. US Air Force [AFOSR FA9550-20-1-0060]
  3. US Department of Energy/Advanced Research Projects Agency [DE-AR0001201]

向作者/读者索取更多资源

This study presents a simple and effective method for quantifying posterior uncertainty in deep operator networks (DeepONets). The approach utilizes a frequentist approach with randomized prior ensembles and introduces an efficient vectorized implementation for fast parallel inference. The proposed method exhibits four main advantages: more robust and accurate predictions compared to deterministic DeepONets, reliable uncertainty estimates for sparse data sets with multi-scale function pairs, effective detection of out-of-distribution and adversarial examples, and seamless quantification of uncertainty due to model bias and data noise. Additionally, the study provides an optimized JAX library called UQDeepONet that can handle large model architectures, ensemble sizes, and data sets with excellent parallel performance on accelerated hardware, enabling uncertainty quantification for DeepONets in realistic large-scale applications.
We present a simple and effective approach for posterior uncertainty quantification in deep operator networks (DeepONets); an emerging paradigm for supervised learning in function spaces. We adopt a frequentist approach based on randomized prior ensembles, and put forth an efficient vectorized implementation for fast parallel inference on accelerated hardware. Through a collection of representative examples in computational mechanics and climate modeling, we show that the merits of the proposed approach are fourfold. (1) It can provide more robust and accurate predictions when compared against deterministic DeepONets. (2) It shows great capability in providing reliable uncertainty estimates on scarce data sets with multi-scale function pairs. (3) It can effectively detect out-of-distribution and adversarial examples. (4) It can seamlessly quantify uncertainty due to model bias, as well as noise corruption in the data. Finally, we provide an optimized JAX library called UQDeepONet that can accommodate large model architectures, large ensemble sizes, as well as large data sets with excellent parallel performance on accelerated hardware, thereby enabling uncertainty quantification for DeepONets in realistic large-scale applications. All code and data accompanying this manuscript will be made available at https://github.com/PredictiveIntelligenceLab/UQDeepONet.(c) 2022 Elsevier B.V. All rights reserved.

作者

我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。

评论

主要评分

4.7
评分不足

次要评分

新颖性
-
重要性
-
科学严谨性
-
评价这篇论文

推荐

暂无数据
暂无数据