Transformer-Based PINN for Semi-Supervised Electromagnetic Forward Simulations

Menglin Zhai, Yujiao Ji, Rui Pei, Longting Xu*, Yaobo Chen, Weibing Lu

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

Abstract

Traditional numerical methods face serious efficiency challenges for solving complex electromagnetic problems. In this paper, a Transformer-based physical information neural network (PINN), which is highly efficient in capturing long-range dependencies of the sequence, is proposed to accelerate electromagnetic forward simulations. To improve model performance on limited data, semi-supervised learning is further introduced by using pseudo-label strategy. Numerical examples show good prediction accuracy of the proposed framework when compared to other PINNs. Through numerical experiments under different electromagnetic scenarios, its generalization ability can also be verified.

Original languageEnglish
JournalIEEE Antennas and Wireless Propagation Letters
DOIs
Publication statusAccepted/In press - 2025

Keywords

  • Electromagnetic forward simulations
  • PINN
  • Semi-supervised learning
  • Transformer

Cite this