TR2026-123

A BERT-Based Surrogate Model for Permanent Magnet Motor Cogging Torque Prediction


    •  Sun, S., Wang, Y., Koike-Akino, T., Yamamoto, T., Sakamoto, Y., Wang, B., "A BERT-Based Surrogate Model for Permanent Magnet Motor Cogging Torque Prediction", International Conference on Electrical Machines (ICEM), September 2026.
      BibTeX TR2026-123 PDF
      • @inproceedings{Sun2026sep,
      • author = {{Sun, Siyuan and Wang, Ye and Koike-Akino, Toshiaki and Yamamoto, Tatsuya and Sakamoto, Yusuke and Wang, Bingnan}},
      • title = {{A BERT-Based Surrogate Model for Permanent Magnet Motor Cogging Torque Prediction}},
      • booktitle = {International Conference on Electrical Machines (ICEM)},
      • year = 2026,
      • month = sep,
      • url = {https://www.merl.com/publications/TR2026-123}
      • }
  • MERL Contacts:
  • Research Areas:

    Electric Systems, Machine Learning

Abstract:

Motor performances such as cogging torque and torque ripple are difficult to predict accurately with surrogate models. In this work, we propose text-transformer-based models to tackle the problem. We first formatted the motor design parameters into natural-language sentences describing the design, which are fed into a text transformer model, and fine-tuned on the motor design dataset to make predictions. Compared with conventional artificial neural network models, our approach is more flexible in the input parameter dimensions, and can be jointly trained on multiple motor types. Numerical tests are performed on two datasets with different types of interior permanent magnet motor designs, and initial results show substantial improvements on cogging torque prediction accuracy over artificial neural network models on both datasets.