Frequency Bias and OOD Generalization in Neural Operators under a Variable-Coefficient Wave Equation

Hugging Face Daily Papers Papers

Summary

This paper investigates the generalization behavior of Fourier Neural Operators and Deep Operator Networks under distribution shifts in a variable-coefficient wave equation, revealing that FNO struggles with high-frequency inputs while DeepONet shows milder degradation.

Neural operators learn to map initial conditions to the terminal solution of partial differential equations (PDEs), providing a surrogate for the full operator mapping. This enables rapid prediction across different input configurations. While recent neural operator architectures have demonstrated strong performance on diverse PDE tasks, their behavior under structured distribution shifts remains insufficiently understood. To investigate this, we study operator learning in a wave propagation setting governed by a one-dimensional variable-coefficient wave equation, using two representative architectures, the Fourier Neural Operator (FNO) and the Deep Operator Network (DeepONet). To examine their generalization under distribution shifts, we consider structured out-of-distribution (OOD) settings that independently vary input frequency and coefficient smoothness. The results show that under smoothness shifts, both models maintain stable performance, with FNO achieving lower error. In contrast, under frequency shifts, FNO exhibits a sharp increase in error under unseen high-frequency inputs, whereas DeepONet shows milder degradation despite higher overall error. Our analysis reveals that these differences arise from how each architecture represents and responds to variations in frequency structure. Together, these findings highlight a fundamental gap between strong in-distribution performance and generalization under distribution shifts in operator learning, underscoring the role of architectural representation bias in developing more reliable neural operators for physics-based PDE simulations beyond the training distribution.
Original Article
View Cached Full Text

Cached at: 05/14/26, 04:17 AM

Paper page - Frequency Bias and OOD Generalization in Neural Operators under a Variable-Coefficient Wave Equation

Source: https://huggingface.co/papers/2605.12997 Published on May 13

·

Submitted byhttps://huggingface.co/lainmn

An Luoon May 14

Abstract

Neural operators for PDE solving show different generalization behaviors under distribution shifts, with Fourier Neural Operators and Deep Operator Networks exhibiting distinct responses to smoothness and frequency variations.

Neural operatorslearn to map initial conditions to the terminal solution ofpartial differential equations(PDEs), providing a surrogate for the full operator mapping. This enables rapid prediction across different input configurations. While recent neural operator architectures have demonstrated strong performance on diverse PDE tasks, their behavior under structureddistribution shiftsremains insufficiently understood. To investigate this, we study operator learning in a wave propagation setting governed by a one-dimensional variable-coefficient wave equation, using two representative architectures, theFourier Neural Operator(FNO) and theDeep Operator Network(DeepONet). To examine their generalization underdistribution shifts, we consider structured out-of-distribution (OOD) settings that independently vary input frequency and coefficient smoothness. The results show that under smoothness shifts, both models maintain stable performance, with FNO achieving lower error. In contrast, under frequency shifts, FNO exhibits a sharp increase in error under unseen high-frequency inputs, whereas DeepONet shows milder degradation despite higher overall error. Our analysis reveals that these differences arise from how each architecture represents and responds to variations infrequency structure. Together, these findings highlight a fundamental gap between strong in-distribution performance and generalization underdistribution shiftsin operator learning, underscoring the role of architecturalrepresentation biasin developing more reliableneural operatorsfor physics-based PDE simulations beyond the training distribution.

View arXiv pageView PDFAdd to collection

Get this paper in your agent:

hf papers read 2605\.12997

Don’t have the latest CLI?curl \-LsSf https://hf\.co/cli/install\.sh \| bash

Models citing this paper0

No model linking this paper

Cite arxiv.org/abs/2605.12997 in a model README.md to link it from this page.

Datasets citing this paper0

No dataset linking this paper

Cite arxiv.org/abs/2605.12997 in a dataset README.md to link it from this page.

Spaces citing this paper0

No Space linking this paper

Cite arxiv.org/abs/2605.12997 in a Space README.md to link it from this page.

Collections including this paper0

No Collection including this paper

Add this paper to acollectionto link it from this page.

Similar Articles

Sequential Physics-Constrained Neural Operator Forward Modeling for the $\textit{Norne}$ Reservoir System

arXiv cs.LG

This paper presents a comprehensive mathematical framework for sequential surrogate modeling of three-phase black-oil reservoir dynamics using Fourier Neural Operators (FNO) and physics-informed variants (PINO), applied to the Norne benchmark reservoir. Theoretical contributions include functional-analytic formulation, covariate shift analysis, physics-constrained spectral stability, and truncated backpropagation gradient analysis.

Iterative Refinement Neural Operators are Learned Fixed-Point Solvers: A Principled Approach to Spectral Bias Mitigation

arXiv cs.LG

This paper introduces the Iterative Refinement Neural Operator (IRNO), which augments pretrained neural operators with a learned refinement module applied via fixed-point iteration to mitigate spectral bias. IRNO progressively corrects high-frequency errors, achieving up to 56% improvement on turbulent flow and showing stable extrapolation beyond the trained iteration count.