Geometry-aware LegONet for PDE Learning on Arbitrary Domains

Jiahao Zhang, Yueqi Wang, Guang Lin

Abstract

Learned PDE solvers often entangle governing operators with the geometry, boundary conditions, and discretization used for training. This limits reuse when the same physics is posed on new domains, and it also makes physical-law discovery geometry-dependent. We introduce Geometry-aware LegONet (gLegONet), a boundary-manifold extension of Lego-like operator learning. Physical mechanisms are pretrained once as modular variational blocks on an ambient spectral domain. For a target geometry, sampled boundary constraints define an affine admissible manifold. Its mass-orthonormal tangent coordinates are used to evolve the dynamics and evaluate candidate law-discovery features directly. Changing the domain therefore changes only an algebraic coordinate interface, not the learned operator blocks. This converts arbitrary-domain PDE learning from geometry-specific retraining or soft penalty enforcement into boundary-guaranteed assembly of reusable mechanisms. In forward simulations and sparse identification tests on unseen domains, the method maintains boundary residuals near the algebraic tolerance and yields predictive governing laws from short-time observations.

Disclosure

“are mechanism representations that can be reused when analytic deriva- tion or repeated geometry-specific assembly is difficult. The same realized mechanisms can then support both forward simulation and physical-law identification. Use of AI-assisted tools A large language model, ChatGPT, was used during manuscript preparation to assist with language editing, consistency checks and the identification of passages that could benefit from further clarification. All suggested revisions wer”

PDF page 24
Classification
Drafting limited passages
Multiplier
5
Verified

Structural counts

Pages 33 pdf
Theorems 0 source
Lemmas 0 source
Propositions 0 source
Corollaries 0 source
Definitions 0 source
Displayed equations 40 source
Bibliography entries 32 source
Appendix pages 0 estimated

Count notes

  • Source counts use the expanded primary TeX file main.tex.
  • Appendix pages include the first PDF page with an explicit Appendix heading through the final page.