Search papers, labs, and topics across Lattice.
We frame neural operator layers as minimizers of Bregman regularized optimization problems in Banach function spaces, interpreting activation operators as Bregman proximity operators. This perspective unifies existing neural operators and introduces Bregman neural operators, which incorporate inverse activation operators and match the expressivity of standard models. Numerical results demonstrate that Bregman variants of Fourier neural operators enable training of deeper and more accurate models.
Neural operators are Bregman proximity operators, unlocking a new class of models with enhanced depth and accuracy.
We present several advances on neural operators by viewing the action of operator layers as the minimizers of Bregman regularized optimization problems over Banach function spaces. The proposed framework allows interpreting the activation operators as Bregman proximity operators from dual to primal space. This novel viewpoint is general enough to recover classical neural operators as well as a new variant, coined Bregman neural operators, which includes the inverse activation operator and features the same expressivity of standard neural operators. Numerical experiments support the added benefits of the Bregman variant of Fourier neural operators for training deeper and more accurate models.