Submitted:
07 July 2025
Posted:
08 July 2025
You are already at the latest version
Abstract
Keywords:
1. Introduction
2. Related Works
2.1. RGB Mapping-Based Approach
2.2. Wavelet Transform-Based Approaches
2.3. STFT-Based Approaches
3. Experimental Setup
3.1. Extraction of D-Axis Current Component

3.2. Data Measurement and Class Configuration
3.3. Raw Data Segmentation
3.4. Data-to-Image Conversion
3.4.1. Three-Phase RGB Image Construction
3.4.2. Three-Phase DWT Image Construction
3.4.3. Three-Phase STFT Image Construction
3.4.4. Id-Linear Image Construction
3.4.5. Id-DWT Image Construction
3.4.6. Id-STFT Image Construction
3.5. Data Splitting and Augmentation
3.6. CNN Configuration
4. Experimental Results and Discussion
4.1. Training Results
4.2. Validation Results
4.3. Test Result
4.4. Discussion
5. Conclusions
Funding
Conflicts of Interest
References
- Ali, U. Towards fault diagnosis in induction motor using fractional Fourier transform. arXiv 2024, arXiv:2412.18227.
- Gubarevych, O.; Gerlici, J.; Kravchenko, O.; Melkonova, I.; Melnyk, O. Use of Park’s vector method for monitoring the rotor condition of an induction motor as a part of the built-in diagnostic system of electric drives of transport. Energies 2023, 16, 5109. [CrossRef]
- Enejo, I.S.; Adegboye, B.; Imoru, O.; James, T.O. Fault diagnosis in a three-phase induction motor using enhanced Park vector approach. In Proceedings of the 2022 IEEE Nigeria 4th International Conference on Disruptive Technologies for Sustainable Development (NIGERCON), Lagos, Nigeria, 6–10 June 2022; pp. 1–6. [CrossRef]
- Zhang, S.; Wang, R.; Si, Y.; Wang, L. An improved convolutional neural network for three-phase inverter fault diagnosis. IEEE Trans. Instrum. Meas. 2022, 71, 1–15. [CrossRef]
- Mohammad-Alikhani, A.; Jamshidpour, E.; Dhale, S.; Akrami, M.; Pardhan, S.; Nahid-Mobarakeh, B. Fault diagnosis of electric motors by a channel-wise regulated CNN and differential of STFT. IEEE Trans. Ind. Appl. 2025, 61, 3066–3077. [CrossRef]
- Zhang, B.; Chen, C.; Wang, Y.; Li, J. STFT based CNN for interturn short circuit fault diagnosis of permanent magnet synchronous motor. In Proceedings of the 2024 3rd International Symposium on Semiconductor and Electronic Technology (ISSET), Xi’an, China, 19–21 April 2024; pp. 400–404. [CrossRef]
- Xu, M.; Gao, J.; Zhang, Z.; Wang, H. Bearing-fault diagnosis with signal-to-RGB image mapping and multichannel multiscale convolutional neural network. Entropy 2022, 24, 1569. [CrossRef]
- Łuczak, D. Machine fault diagnosis through vibration analysis: Time series conversion to grayscale and RGB images for recognition via convolutional neural networks. Energies 2024, 17, 1998. [CrossRef]
- Łuczak, D.; Brock, S.; Siembab, K. Fault detection and localisation of a three-phase inverter with permanent magnet synchronous motor load using a convolutional neural network. Actuators 2023, 12, 125. [CrossRef]
- Yu, T.; Jiang, Z.; Ren, Z.; Zhou, Y.; Zhang, Y.; Gao, R. A convolutional multisensor fusion fault diagnosis framework based on multidimensional distance matrix for rotating machinery. Struct. Health Monit. 2024, 0, 0. [CrossRef]
- Ahmed, H.O.A.; Nandi, A.K. Connected components-based colour image representations of vibrations for a two-stage fault diagnosis of roller bearings using convolutional neural networks. Chin. J. Mech. Eng. 2021, 34, 37. [CrossRef]
- Xing, Z.; Liu, Y.; Wang, Q.; Zhao, L.; Chen, Y. Fault diagnosis of rotating parts integrating transfer learning and ConvNeXt model. Sci. Rep. 2025, 15, 190. [CrossRef]
- Piedad, E.; Del Rosario, C.A.; Prieto-Araujo, E.; Gomis–Bellmunt, O. Exploring wavelet transformations for deep learning-based machine condition diagnosis. In Proceedings of the 2024 International Conference on Diagnostics in Electrical Engineering (Diagnostika), Pilsen, Czech Republic, 10–12 September 2024; pp. 1–4. [CrossRef]
- Le, V.D.; Nguyen, N.B.; Le, Q.C. Deep learning method for fault diagnosis in high voltage transmission lines: A case of the Vietnam 220 kV transmission line. Int. J. Electr. Eng. Inform. 2022, 14(2), 254–275. [CrossRef]
- Paraskevopoulos, D.; Spandonidis, C.; Giannopoulos, F. Hybrid wavelet–CNN fault diagnosis method for ships’ power systems. Signals 2023, 4, 150–166. [CrossRef]
- Hsueh, Y.-M.; Ittangihal, V.R.; Wu, W.-B.; Chang, H.-C.; Kuo, C.-C. Fault diagnosis system for induction motors by CNN using empirical wavelet transform. Symmetry 2019, 11, 1212. [CrossRef]
- Pietrzak, P.; Wolkiewicz, M. Application of continuous wavelet transform and convolutional neural networks in fault diagnosis of PMSM stator windings. Bull. Pol. Acad. Sci. Tech. Sci. 2024, 72, 150202. [CrossRef]
- Fu, G.; Wei, Q.; Yang, Y. Bearing fault diagnosis with parallel CNN and LSTM. Math. Biosci. Eng. 2024, 21, 2385–2406. [CrossRef]
- Ni, Y.; Li, S.; Guo, P. Discrete wavelet integrated convolutional residual network for bearing fault diagnosis under noise and variable operating conditions. Sci. Rep. 2025, 15, 16185. [CrossRef]
- Saravanan, N.; Ramachandran, K.I. Incipient gear box fault diagnosis using discrete wavelet transform (DWT) for feature extraction and classification using artificial neural network (ANN). Expert Syst. Appl. 2010, 37, 4168–4181. [CrossRef]
- Appiah, M.K.; Danuor, S.K.; Bienibuor, A.K. Performance of continuous wavelet transform over Fourier transform in feature resolutions. Int. J. Geosci. 2024, 15, 87–105. [CrossRef]
- Piedad, E.; Mayordo, Z.G.; Prieto-Araujo, E.; Gomis-Bellmunt, O. Deep learning-based machine condition diagnosis using short-time Fourier transformation variants. In Proceedings of the 2024 International Conference on Diagnostics in Electrical Engineering (Diagnostika), Pilsen, Czech Republic, 10–12 September 2024; pp. 1–4. [CrossRef]
- Song, X.; Li, Z.; Liu, Y. MVB fault diagnosis based on time-frequency analysis and convolutional neural networks. Sci. Rep. 2025, 15, 5271. [CrossRef]
- Ali, U.; Ali, W.; Ramzan, U. An improved fault diagnosis strategy for induction motors using weighted probability ensemble deep learning. arXiv 2024, arXiv:2412.18249. [CrossRef]
- Son, T.; Hong, D.; Kim, B. Multi–output classification based on convolutional neural network model for untrained compound fault diagnosis of rotor systems with non–contact sensors. Sensors 2023, 23, 3153. [CrossRef]












| Classification | Layer configuration |
Activation function | Output dimension | Remark |
| Feature extractor |
ResNET50 (Top layer removed; weights=none) | 7 x 7 x 2048 | Input 224 x 224 x 3 | |
| Flatten layer FC layer 1 FC layer 2 Output layer |
Flatten Dense (512) Batch normalization Dropout (0.4) Dense (256) Batch normalization Dropout (0.3) Dense (3) |
ReLU ReLU Softmax |
100,352 512 512 512 256 256 256 3 |
7 x 7 x 2048 -> 1D Fully connected layer Feature normalization Overfitting prevention Fully connected layer Feature normalization Overfitting prevention No. of classes: 3 |
| Item | Settings |
|---|---|
| Optimization algorithm | Adam |
| Initial learning rate Loss function No. of epochs Batch size Callback function |
0.0001 Categorical cross-entropy 50 32 ReduceLROnPlateau, CSVLogger, etc. |
| 3Phase -RGB |
3Phase -DWT |
3Phase- STFT |
Id -Linear |
Id -DWT |
Id -STFT |
|
|---|---|---|---|---|---|---|
| Test Acc. | 99.01% | 98.27% | 76.56% | 94.59% | 72.11% | 99.01% |
| Classification | Data | Encoding | Normal | ITSC | 4Tum |
|
Precision Recall F1-Score |
3Phase Id 3Phase Id 3Phase Id |
DWT RGB STFT DWT Linear STFT DWT RGB STFT DWT Linear STFT DWT RGB STFT DWT Linear STFT |
100.00% 100.00% 100.00% 82.66% 99.26% 100.00% 100.00% 100.00% 100.00% 82.96% 100.00% 100.00% 100.00% 100.00% 100.00% 82.81% 99.63% 100.00% |
97.06% 98.52% 65.50% 67.36% 93.80% 98.52% 97.78% 98.52% 62.59% 60.37% 89.63% 98.52% 97.42% 98.52% 64.02% 63.67% 91.67% 98.52% |
97.76% 98.52% 64.18% 66.33% 90.71% 98.52% 97.04% 98.52% 67.04% 72.96% 94.07% 98.52% 97.40% 98.52% 65.58% 69.49% 92.36% 98.52% |
Disclaimer/Publisher’s Note: The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of MDPI and/or the editor(s). MDPI and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content. |
© 2025 by the authors. Licensee MDPI, Basel, Switzerland. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license (http://creativecommons.org/licenses/by/4.0/).