Spatio-spectral graph neural operator for solving computational mechanics problems on irregular domain and unstructured grid
About
Scientific machine learning has seen significant progress with the emergence of operator learning. However, existing methods encounter difficulties when applied to problems on unstructured grids and irregular domains. Spatial graph neural networks utilize local convolution in a neighborhood to potentially address these challenges, yet they often suffer from issues such as over-smoothing and over-squashing in deep architectures. Conversely, spectral graph neural networks leverage global convolution to capture extensive features and long-range dependencies in domain graphs, albeit at a high computational cost due to Eigenvalue decomposition. In this paper, we introduce a novel approach, referred to as Spatio-Spectral Graph Neural Operator (Sp$^2$GNO) that integrates spatial and spectral GNNs effectively. This framework mitigates the limitations of individual methods and enables the learning of solution operators across arbitrary geometries, thus catering to a wide range of real-world problems. Sp$^2$GNO demonstrates exceptional performance in solving both time-dependent and time-independent partial differential equations on regular and irregular domains. Our approach is validated through comprehensive benchmarks and practical applications drawn from computational mechanics and scientific computing literature.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Segmentation | Human Body 12k-vertex meshes | Accuracy70.2 | 14 | |
| Shape classification | SHREC-11 30-class | Accuracy63.8 | 14 | |
| Segmentation | RNA Surface 640 meshes | Accuracy73.9 | 14 | |
| PDE solving | Poisson | L2 Error0.0069 | 13 | |
| Fluid Dynamics Prediction | Shape-Net Car | Pressure L2 Error0.1197 | 13 |