Search papers, labs, and topics across Lattice.
This paper introduces a novel Mixture of Experts (MoE) architecture for steady PDE surrogate modeling on coarse unstructured meshes, where individual transformer experts specialize in distinct spectral scales of the solution. A learnable Hodge-Laplacian layer acts as a spectral filter, guiding the partitioning of the solution space. Evaluated on a remeshed RANS airfoil benchmark, this approach significantly outperforms Transolver, achieving a 31% reduction in relative L2 error and a 47% reduction in interior mean squared error.
Spectral decomposition via a learnable Hodge-Laplacian unlocks a 47% reduction in interior MSE for steady-PDE surrogates on coarse meshes.
Current approaches for steady partial differential equations surrogate models on unstructured meshes apply a single mixing mechanism to the whole solution. This paper proposes an architecture that partitions the solution among vanilla transformer experts, each focused on a different spectral scale. The partition is built from a discrete Hodge Laplacian $L_0=\star_0^{-1} d_0^{\top}\star_1 d_0$, that serves as a fitting layer measuring high-frequency content. A gating mechanism takes the mesh as input and produces the per-vertex weights that recombine the expert outputs. AirFRANS is a steady Reynolds-averaged Navier--Stokes (RANS) airfoil benchmark. We evaluate our approach in a reduced remeshed version of the benchmark with each unstructured mesh counting 9.6K faces and 4.9K vertices. We compare our approach with the Transolver on the same dataset and dataset slit. Our approach reduces relative $L_2$ error by 31\% and interior mean squared error by 47\%.