ICML2026

Fractional is Better: Learnable Derivative Orders in Neural Operator Learning

Fares B. Mehouachi, Saif Jabari

摘要

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))(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 rates from O(W(sm)/d)O(W^{-(s-m)/d}) to O(Ws/d)O(W^{-s/d}), where WW is network width, ss is input regularity, mm is the PDE order, and dd is spatial dimension. Our central finding, however, is a surprise: the optimal derivative order β\beta^* is strictly less than the PDE order mm. This gap arises from a bias-variance tradeoff in spectral space that we characterize in closed form. Learning β\beta 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.