Deep Learning-based Machine Condition Diagnosis using Short-time Fourier Transformation Variants

TL;DR

This paper explores the use of various Short-time Fourier Transform variants combined with deep learning models to accurately diagnose motor conditions from current signals, achieving high classification accuracy.

Contribution

It introduces a novel application of multiple STFT variants with deep learning for motor fault diagnosis, outperforming previous ML and 2D-plot methods.

Findings

Overlap-STFT achieved 97.65% accuracy.

Synchrosqueezed STFT achieved 88.27% accuracy.

All methods outperformed previous best results.

Abstract

In motor condition diagnosis, electrical current signature serves as an alternative feature to vibration-based sensor data, which is a more expensive and invasive method. Machine learning (ML) techniques have been emerging in diagnosing motor conditions using only motor phase current signals. This study converts time-series motor current signals to time-frequency 2D plots using Short-time Fourier Transform (STFT) methods. The motor current signal dataset consists of 3,750 sample points with five classes - one healthy and four synthetically-applied motor fault conditions, and with five loading conditions: 0, 25, 50, 75, and 100%. Five transformation methods are used on the dataset: non-overlap and overlap STFTs, non-overlap and overlap realigned STFTs, and synchrosqueezed STFT. Then, deep learning (DL) models based on the previous Convolutional Neural Network (CNN) architecture are trained and validated from generated plots of each method. The DL models of overlap-STFT, overlap R-STFT, non-overlap STFT, non-overlap R-STFT, and synchrosqueezed-STFT performed exceptionally with an average accuracy of 97.65, 96.03, 96.08, 96.32, and 88.27%, respectively. Four methods outperformed the previous best ML method with 93.20% accuracy, while all five outperformed previous 2D-plot-based methods with accuracy of 80.25, 74.80, and 82.80%, respectively, using the same dataset, same DL architecture, and validation steps.