Shearlet Neural Operators for Anisotropic-Shock-Dominated and Multi-scale parametric partial differential equations

TL;DR

The paper introduces Shearlet Neural Operators (SNO), which replace Fourier transforms with shearlet-based representations to better handle anisotropic, multiscale, and shock-dominated PDEs, improving accuracy and feature fidelity.

Contribution

The paper proposes a novel neural operator architecture using shearlet transforms, enhancing the ability to model anisotropic and discontinuous PDE solutions compared to Fourier-based methods.

Findings

SNO outperforms FNO baselines across seven benchmark PDEs.

SNO achieves higher accuracy in anisotropic and shock-dominated regimes.

Shearlet domain learning improves feature localization and representation.

Abstract

Neural operators have emerged as powerful data-driven surrogates for learning solution operators of parametric partial differential equations (PDEs). However, widely used Fourier Neural Operators (FNOs) rely on global Fourier representations, which can be inefficient for resolving anisotropic structures, sharp gradients, and spatially localized discontinuities that arise in shock-dominated and multiscale regimes. To address these limitations, we introduce the Shearlet Neural Operator (SNO), a neural operator architecture that replaces the Fourier transform with a shearlet-based representation. Shearlets offer directional, multiscale, and spatially localized atoms with near-optimal sparse approximation of anisotropic features, providing an inductive bias aligned with PDE solutions containing edges, fronts, and shocks. SNO learns in the shearlet domain and reconstructs predictions via the inverse transform, retaining efficient spectral computation while improving locality and directional selectivity. Across seven benchmark PDE families, including strongly anisotropic advection, anisotropic diffusion, and nonlinear conservation laws with straight, curved, interacting, spiral, and polygonal shock structures, SNO consistently improves predictive accuracy and feature fidelity over FNO baselines, with the largest gains observed in anisotropic and discontinuity-dominated settings.