Abstract:Graph Neural Networks (GNNs) have gained significant traction for simulating complex physical systems, with models like MeshGraphNet demonstrating strong performance on unstructured simulation meshes. However, these models face several limitations, including scalability issues, requirement for meshing at inference, and challenges in handling long-range interactions. In this work, we introduce X-MeshGraphNet, a scalable, multi-scale extension of MeshGraphNet designed to address these challenges. X-MeshGraphNet overcomes the scalability bottleneck by partitioning large graphs and incorporating halo regions that enable seamless message passing across partitions. This, combined with gradient aggregation, ensures that training across partitions is equivalent to processing the entire graph at once. To remove the dependency on simulation meshes, X-MeshGraphNet constructs custom graphs directly from CAD files by generating uniform point clouds on the surface or volume of the object and connecting k-nearest neighbors. Additionally, our model builds multi-scale graphs by iteratively combining coarse and fine-resolution point clouds, where each level refines the previous, allowing for efficient long-range interactions. Our experiments demonstrate that X-MeshGraphNet maintains the predictive accuracy of full-graph GNNs while significantly improving scalability and flexibility. This approach eliminates the need for time-consuming mesh generation at inference, offering a practical solution for real-time simulation across a wide range of applications. The code for reproducing the results presented in this paper is available through NVIDIA Modulus: <a class="link-external link-http" href="http://github.com/NVIDIA/modulus/tree/main/examples/cfd/xaeronet" rel="external noopener nofollow">this http URL</a>.

TCAD Device Simulation With Graph Neural Network

Restructuring TCAD System: Teaching Traditional TCAD New Tricks

Comprehensive studies on deep learning applicable to TCAD

Revolutionizing TCAD Simulations with Universal Device Encoding and Graph Attention Networks

Dynamic graph-based convergence acceleration for topology optimization in unstructured meshes

Proposing an intelligent mesh smoothing method with graph neural networks

Framework for TCAD Augmented Machine Learning on Multi-I-v Characteristics Using Convolutional Neural Network and Multiprocessing

Device Modeling Based on Cost-Sensitive Densely Connected Deep Neural Networks

Learning Mesh-Based Simulation with Graph Networks

Accelerate Microstructure Evolution Simulation Using Graph Neural Networks with Adaptive Spatiotemporal Resolution

Accelerated Modelling of Interfaces for Electronic Devices using Graph Neural Networks

Developing a novel structured mesh generation method based on deep neural networks

Intelligent mesh generation for crack simulation using graph neural networks

Scientific Computing Algorithms to Learn Enhanced Scalable Surrogates for Mesh Physics

TCAD Simulation of Novel Semiconductor Devices

Mesh Neural Networks Based on Dual Graph Pyramids

Multiscale graph neural networks with adaptive mesh refinement for accelerating mesh-based simulations

X-MeshGraphNet: Scalable Multi-Scale Graph Neural Networks for Physics Simulation

Up-sampling-only and Adaptive Mesh-based GNN for Simulating Physical Systems

Transistor Compact Model Based On Gradient Constrainted Convolution Neural Network

The Device Compact Model Based on Multi-gradient Neural Network and Its Application on MoS2 Field Effect Transistors