Localized Operator Learning with Adaptive Partition-of-Unity Mixture-of-Expert Networks
We introduce a partition-of-unity mixture-of-experts framework for localized operator learning, using geometry-aware gating networks to blend local experts through smooth spatial partitions. Our HiRefPOU architecture organizes DeepONet representations into nested parent-child partitions with residual refinement while preserving global continuity. We also incorporate spatial adaptivity into Fourier Neural Operators. On heterogeneous Darcy and reaction-diffusion benchmarks, HiRefPOU improves accuracy over global DeepONet and static POU-MoE baselines, with interpretable partitions that align with interfaces and regions of rapid solution variation.