Search papers, labs, and topics across Lattice.
This paper investigates the impact of input feature representation on neural operator learning for PDEs, proposing that providing derivative information can improve approximation bounds by reducing the network's implicit differentiation burden. Empirically, it's found that the optimal derivative order is strictly less than the PDE order due to a spectral bias-variance tradeoff, which can be learned from data. The proposed $\partial$-NO architecture-agnostic augmentation consistently enhances prediction accuracy by incorporating learnable fractional derivative features.
Neural operators learn optimal derivative orders strictly below the PDE order, revealing a data-driven spectral regularization mechanism.
Neural operators learn mappings between function spaces, enabling fast surrogate solutions to partial differential equations. Despite remarkable architectural diversity, these methods often share a common input representation: raw coordinate-value pairs $(x, u(x))$. We ask whether inputs aligned with PDE differential structure can improve learning. Through Picard iteration on mild solutions, we show that derivatives of the input appear explicitly in the solution operator, suggesting that providing derivative features should reduce the network's implicit differentiation burden. We prove this intuition: providing derivative features improves approximation bounds from $O(W^{-(s-m)/d})$ to $O(W^{-s/d})$, where $W$ is parameter count, $s$ is input regularity, $m$ is the PDE order, and $d$ is spatial dimension. Our central finding, however, is a surprise: the optimal derivative order $β^*$ is strictly less than the PDE order $m$. This gap arises from a bias-variance tradeoff in spectral space that we characterize in closed form. Learning $β$ from data achieves automatic spectral regularization. We introduce $\partial$-NO (del-NO), for derivative-augmented neural operators, an architecture-agnostic augmentation that provides learnable fractional derivative features to any neural operator backbone. Across benchmark problems and architectures, $\partial$-NO consistently improves prediction accuracy, with learned orders that reflect a representation of known physics modulated by noise and finite-sample constraints.