Let There Be Light: Reflection, Refraction and Scattering for Neural Operators
Organizations: Suzhou Institute for Advanced Research, University of Science and Technology of China, Suzhou 215123, China · School of Artificial Intelligence and Data Science, University of Science and Technology of China, Hefei 230026, China · Suzhou Big Data & AI Research and Engineering Center, Suzhou 215123, China · School of Mathematical Science, Peking University, Beijing, 100871, China · School of Mathematical Sciences, University of Science and Technology of China, Hefei 230026, China
Abstract
Neural operators learn mappings between infinite-dimensional function spaces and provide a data-driven surrogate modeling paradigm for parametric partial differential equations (PDEs). Existing architectures typically obtain expressivity by parameterizing integral kernels in prescribed transform domains or by applying attention-like interactions over discretized spatial points. While these approaches have achieved substantial progress, they often face a persistent trade-off among physical interpretability, nonlocal spatial communication, mesh scalability, and computational cost. We propose a Light-inspired neural operator(LiNO), an operator-learning architecture whose latent evolution is decomposed into three mechanisms motivated by elementary light transport: reflection, refraction, and scattering. Reflection and refraction act as adaptive pointwise transformations in latent feature space, enabling local feature reorientation and anisotropic modulation, whereas scattering performs input-dependent nonlocal propagation over the physical domain. We first formulate scattering as a normalized pairwise kernel with relative positional bias, and then develop an efficient scattering variant that replaces explicit pairwise interactions with positive-feature global propagation and a local diffusion branch, reducing the dominant spatial complexity from quadratic to linear. This yields a structured neural operator that separates local feature modulation from global spatial communication while retaining a modular and interpretable latent evolution.