Deep transfer learning strategy for efficient domain generalisation in machine fault diagnosis

Abstract

AbstractAutomated fault diagnosis algorithms based on vibration sensor recordings play an important role in determining the state of health of the machines. Data-driven approaches demand a large amount of labelled data to build reliable models. The performance of such lab-trained models degrades when deployed in practical use cases in the presence of distinct distribution target domain datasets. In this work, we present a novel deep transfer learning strategy that fine-tunes the trainable parameters of the lower (convolutional) layers with respect to the changing target domain datasets and transfers the parameters of the deeper (dense) layers from the source domain for efficient domain generalisation and fault classification. The performance of this strategy is evaluated by considering two different target domain datasets and studying the sensitivity of fine-tuning individual layers in the networks using time-frequency representations of the vibration signals (scalograms) as inputs. We observe that the proposed transfer learning strategy yields near-perfect accuracy, even for use cases where low-precision sensors are used for data collection and unlabelled run-to-failure data with a limited number of training samples.