kth.sePublications KTH
Change search
CiteExportLink to record
Permanent link

Direct link
Cite
Citation style
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Other style
More styles
Language
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Other locale
More languages
Output format
  • html
  • text
  • asciidoc
  • rtf
MMET: A Multi-Input and Multi-Scale Transformer for Efficient PDEs Solving
KTH, School of Electrical Engineering and Computer Science (EECS), Information Science and Engineering.ORCID iD: 0000-0001-5696-3103
Xian Jiaotong Liverpool Univ, Sch Adv Technol, Suzhou, Peoples R China.
Techforgood AS, Oslo, Norway.
Techforgood AS, Oslo, Norway.
Show others and affiliations
2025 (English)In: PROCEEDINGS OF THE THIRTY-FOURTH INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE (IJCAI 2025) / [ed] Kwok, J, International Joint Conferences on Artificial Intelligence , 2025, p. 7634-7642Conference paper, Published paper (Refereed)
Abstract [en]

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.

Place, publisher, year, edition, pages
International Joint Conferences on Artificial Intelligence , 2025. p. 7634-7642
National Category
Computer graphics and computer vision
Identifiers
URN: urn:nbn:se:kth:diva-378130DOI: 10.24963/ijcai.2025/849ISI: 001634925300207Scopus ID: 2-s2.0-105021818973OAI: oai:DiVA.org:kth-378130DiVA, id: diva2:2046506
Conference
34th International Joint Conference on Artificial Intelligence-IJCAI, AUG 16-22, 2025, Montreal, CANADA
Note

Part of ISBN 978-1-956792-06-5

QC 20260317

Available from: 2026-03-17 Created: 2026-03-17 Last updated: 2026-03-17Bibliographically approved

Open Access in DiVA

No full text in DiVA

Other links

Publisher's full textScopus

Authority records

Luo, YichenPang, Zhibo

Search in DiVA

By author/editor
Luo, YichenPang, Zhibo
By organisation
Information Science and Engineering
Computer graphics and computer vision

Search outside of DiVA

GoogleGoogle Scholar

doi
urn-nbn

Altmetric score

doi
urn-nbn
Total: 23 hits
CiteExportLink to record
Permanent link

Direct link
Cite
Citation style
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Other style
More styles
Language
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Other locale
More languages
Output format
  • html
  • text
  • asciidoc
  • rtf