Disentangling Attention in Deep Operator Learning: A Controlled Study of Data-Driven and Physics-Informed Architectures
A controlled study on a variant of DeepONet revealed the impact of the attention mechanism on the accuracy of PDE solutions. The researchers tested five DeepONet variants with different attention mechanisms in both data-driven and physical information modes, and evaluated them on benchmark problems such as source-driven transient one-dimensional nonlinear diffusion-reaction equations, transient one-dimensional viscous Burgers equations, and two-dimensional Poisson heat conduction problems. The results showed that configurations with sensor-based segmentation and cross-attention reduced the average relative L_2 error of the classical DeepONet by 2.4 to 28.0 times across all benchmark training combinations, with the best configuration reducing it by 3.5 to 32.3 times; whereas the branch-specific attention mechanism with only dot-product fusion performed inconsistently, and although adding cross-attention improved all six scenarios, the improvement was minimal. Global pre-mixing did not provide consistent benefits; increasing the depth of cross-attention could improve accuracy but at a significant cost, with diminishing returns. Overall, query-dependent cross-attention was the most…