DeepONet-grid-UQ: A trustworthy deep operator framework for predicting the power grid's post-fault trajectories.

• A Deep Operator Network (DeepONet) for data-driven transient response prediction. • The Bayesian DeepONet enables a reliable prediction via uncertainty quantification. • The Bayesian DeepONet enables learning even when data is scarce. • The Probabilistic DeepONet enables a reliable prediction at virtually no extra cost This paper proposes a novel data-driven method for the reliable prediction of the power grid's post-fault trajectories, i.e., the power grid's dynamic response after a disturbance or fault. The proposed method is based on the recently proposed concept of Deep Operator Networks (DeepONets). Unlike traditional neural networks that learn to approximate functions, DeepONets are designed to approximate nonlinear operators, i.e., mappings between infinite-dimensional spaces. Under this operator framework, we design a novel and efficient DeepONet that (i) takes as inputs the trajectories collected before and during the fault and (ii) outputs the predicted post-fault trajectories. In addition, we endow our method with the much-needed ability to balance efficiency with reliable/trustworthy predictions via uncertainty quantification. To this end, we propose and compare two novel methods that enable quantifying the predictive uncertainty. First, we propose a Bayesian DeepONet (B-DeepONet) that uses stochastic gradient Hamiltonian Monte-Carlo to sample from the posterior distribution of the DeepONet trainable parameters. Then, we design a Probabilistic DeepONet (Prob-DeepONet) that uses a probabilistic training strategy to enable quantifying uncertainty at virtually no extra computational cost. Finally, we validate the proposed methods' predictive power and uncertainty quantification capability using the New York-New England power grid model. [ABSTRACT FROM AUTHOR]