Uncertainty Quantification with Bayesian Higher Order ReLU KANs

James Giroux, Cristiano Fanelli

arXiv:2410.01687·cs.LG·Published 2024-10-02·Updated 2024-10-03

We introduce the first method of uncertainty quantification in the domain of Kolmogorov-Arnold Networks, specifically focusing on (Higher Order) ReLUKANs to enhance computational efficiency given the computational demands of Bayesian methods. The method we propose is general in nature, providing access to both epistemic and aleatoric uncertainties. It is also capable of generalization to other various basis functions. We validate our method through a series of closure tests, including simple one-dimensional functions and application to the domain of (Stochastic) Partial Differential Equations. Referring to the latter, we demonstrate the method's ability to correctly identify functional dependencies introduced through the inclusion of a stochastic term. The code supporting this work can be found at https://github.com/wmdataphys/Bayesian-HR-KAN

TopicsDynamical Systems & PDE Learning

Tagspartial-differential-equations uncertainty-quantification

arXiv categoriescs.LG, cs.AI, physics.data-an

arXiv abstract pagePDF