/ieee-etfa2020-paper

Machine Learning Based Unbalance Detection of a Rotating Shaft Using Vibration Data

Primary LanguageJupyter NotebookApache License 2.0Apache-2.0

Machine Learning Based Unbalance Detection of a Rotating Shaft Using Vibration Data

Oliver Mey, Willi Neudeck, André Schneider and Olaf Enge-Rosenblatt

Fraunhofer IIS/EAS, Fraunhofer Institute for Integrated Circuits, Division Engineering of Adaptive Systems, Dresden, Germany

This GitHub repository is part of a paper submission to the 25th IEEE International Conference on Emerging Technologies and Factory Automation, ETFA 2020 (https://ieeexplore.ieee.org/document/9212000). It contains a series five freely available Python Jupyter notebooks with code examples. The code was used to get the classifications results described within the ETFA paper. The dataset used for the detection of unbalances based on vibration data along with this study is available via the Fraunhofer Fordatis database (https://fordatis.fraunhofer.de/handle/fordatis/151.2). Both, the Python code and the data, allow all interested engineers, data scientists, and Machine Learning / Deep Learning / AI freaks to check the classification results described in the IEEE paper and to supplement their own analyzes. We look forward to your comments and suggestions for improvements.

Data from vibration sensor 1 for a complete measurement for the case of no unbalance and the largest unbalance. Data from vibration sensor 1 for a complete measurement for the case of no unbalance and the largest unbalance.

For both cases, also a one second sample is extracted. For both cases, also a one second sample is extracted.

The FFT transformation of the a measurement cycle. The FFT transformation of the a measurement cycle.

Classification results based on Convolutional Neural Networks for different unbalance cases.

Classification results based on raw timeseries/vibration signals using Convolutional Neural Networks (CNN) for different unbalance cases (Approach 1).

Classification results  based on FFT-transformed signals for different unbalance cases.

Classification results based on FFT-transformed signals using Fully Connected Neural Networks (FCN) for different unbalance cases (Approach 2).