Abstract
The Kolmogorov-Arnold network (KAN) is a regression model that is based on arepresentation of an arbitrary continuous multivariate function by acomposition of functions of a single variable. Experimentally-obtained datasetsfor regression models typically include uncertainties, which in some cases,cannot be neglected. The conventional way to account for the latter is to modelconfidence intervals of the systems' outputs in addition to the expected valuesof the outputs. However, such information may be insufficient, and in somecases, researchers aim to obtain probability distributions of the outputs. Thepresent paper proposes a method for estimating probability distributions of theoutputs in the case of aleatoric uncertainty (i.e. for systems that producedifferent outputs each time an experiment is executed with the same inputs).The suggested approach covers input-dependent probability distributions of theoutputs and is capable of capturing the multi-modality, as well as thevariation of the distribution type with the inputs. Although the method isapplicable to any regression model, the present paper combines it with KANs,since the specific structure of KANs leads to computationally-efficient models'construction. The source code is available online.