Classifying the Percentage of Broken Magnets in Permanent Magnet Synchronous Motors Using Combined Short-Time Fourier Transform and a Pre-Trained Convolutional Neural Network

Abstract

In critical applications of electrical machines, ensuring validity and safety is paramount to prevent system failures with potentially hazardous consequences. The integration of machine learning (ML) technologies plays a crucial role in monitoring system performance and averting failures. Among various motor types, permanent magnet synchronous motors (PMSMs) are widely favored for their versatile speed range, enhanced power density, and ease of control, finding applications in both industrial settings and electric vehicles. This study focuses on the detection and classification of the percentage of broken magnets in PMSMs using a pre-trained AlexNet convolutional neural network (CNN) model. The dataset was generated by combining finite element methods (FEMs) and short-time Fourier transform (STFT) applied to stator phase currents, which exhibited significant variations due to diverse broken magnet structures. Leveraging transfer learning, the pre-trained AlexNet model underwent adjustments, including the elimination and rearrangement of the final three layers and the introduction of new layers tailored for electrical machine applications. The resulting pre-trained CNN model achieved a remarkable performance, boasting a 99.94% training accuracy and 0.0004% training loss in the simulation dataset, utilizing a PMSM with 4% magnet damage for experimental validation. The model’s effectiveness was further affirmed by an impressive 99.95% area under the receiver operating characteristic (ROC) curve in the experimental dataset. These results underscore the efficacy and robustness of the proposed pre-trained CNN method in detecting and classifying the percentage of broken magnets, even with a limited dataset.