Let There Be Light: Reflection, Refraction and Scattering for Neural Operators
2026-06-02 • Machine Learning
Machine Learning
AI summaryⓘ
The authors introduce a new kind of neural operator called LiNO, inspired by how light behaves with reflection, refraction, and scattering. Their method breaks down the process of learning complex transformations into simple parts that handle local changes and global communication separately. They also improve efficiency by reducing the computational cost from growing very fast to growing more slowly with the size of the data. This makes their model easier to understand and better at handling complex spatial relationships in physics problems.
neural operatorspartial differential equationslatent evolutionreflectionrefractionscatteringkernel methodsnonlocal communicationcomputational complexityanisotropic modulation
Authors
Keke Wu, Yixuan Zhang, Jingrun Chen
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.