MMET: A Multi-Input and Multi-Scale Transformer for Efficient PDEs Solving

TL;DR

This paper introduces MMET, a transformer-based framework that efficiently solves multi-input, multi-scale PDEs by reducing computational costs and improving accuracy, enabling real-time applications in engineering and physics.

Contribution

The paper presents a novel multi-input, multi-scale transformer architecture with Gated Condition Embedding and Hilbert curve-based input reduction, enhancing PDE solving efficiency and scalability.

Findings

Outperforms state-of-the-art methods in accuracy.

Reduces computational costs significantly.

Supports large-scale, multi-input PDE problems.

Abstract

Partial Differential Equations (PDEs) are fundamental for modeling physical systems, yet solving them in a generic and efficient manner using machine learning-based approaches remains challenging due to limited multi-input and multi-scale generalization capabilities, as well as high computational costs. This paper proposes the Multi-input and Multi-scale Efficient Transformer (MMET), a novel framework designed to address the above challenges. MMET decouples mesh and query points as two sequences and feeds them into the encoder and decoder, respectively, and uses a Gated Condition Embedding (GCE) layer to embed input variables or functions with varying dimensions, enabling effective solutions for multi-scale and multi-input problems. Additionally, a Hilbert curve-based reserialization and patch embedding mechanism decrease the input length. This significantly reduces the computational cost when dealing with large-scale geometric models. These innovations enable efficient representations and support multi-scale resolution queries for large-scale and multi-input PDE problems. Experimental evaluations on diverse benchmarks spanning different physical fields demonstrate that MMET outperforms SOTA methods in both accuracy and computational efficiency. This work highlights the potential of MMET as a robust and scalable solution for real-time PDE solving in engineering and physics-based applications, paving the way for future explorations into pre-trained large-scale models in specific domains. This work is open-sourced at https://github.com/YichenLuo-0/MMET.