Search papers, labs, and topics across Lattice.
This paper introduces GAIA, a Geometry-Adaptive Integral Autoencoder designed for operator learning in partial differential equations (PDEs) that efficiently handles both forward and inverse problems across arbitrary geometries. By encoding domain boundaries and interior field distributions into geometry tokens and employing cross-attention in integral transform layers, GAIA adapts its kernel locally to geometric features, eliminating the need for retraining or iterative optimization. The model achieves state-of-the-art performance on various benchmarks, significantly reducing median relative \(L^2\) error in tasks such as airfoil flow reconstruction and electrical impedance tomography compared to existing methods.
GAIA achieves a 64% reduction in error for airfoil flow reconstruction, setting a new benchmark for operator learning in complex geometries.
Operator learning for partial differential equations (PDEs) on arbitrary geometries builds fast neural surrogates for large-scale simulation. Although recent geometry-adaptive neural operators have made substantial progress, they are mainly designed for forward problems in which inputs and outputs share the same spatial domain. This limits their applicability for boundary value problems (BVPs) and inverse problems, where inputs and outputs may live on different domains. We introduce the Geometry-Adaptive Integral Autoencoder (GAIA), an operator learning model that encodes the domain boundary and the interior field distribution into geometry tokens, and conditions integral transform layers on these tokens via cross-attention, allowing the kernel to adapt locally to geometric features. This yields a single architecture for forward (including BVPs) and inverse problems on arbitrary domains in one pass, without retraining, iterative optimization, or graph construction. We evaluate GAIA on seven 2D and 3D benchmarks, four of which are new or substantially extended benchmarks for inverse problems and BVP: electrical impedance tomography, optical tomography, 3D Darcy flow on varying geometries, and a modified setting of Poisson BVP on mechanical components benchmark (MCB). GAIA sets new state-of-the-art results on every inverse and BVP task, reducing median relative $L^2$ error by 64% on airfoil flow reconstruction and 27% on EIT relative to the next best amortized method, and outperforming all baselines on every shape category of MCB. On other forward problems, GAIA is competitive with specialized solvers while maintaining stable accuracy across point resolutions on which transformer-based baselines degrade.
