School of Naval Architecture and Ocean Engineering, Huazhong University of Science and Technology, Wuhan 430074, China.
Wuhan University Shenzhen Research Institute, Shenzhen 518057, China.
Sensors (Basel). 2021 Dec 28;22(1):192. doi: 10.3390/s22010192.
Convolution neural network (CNN)-based fault diagnosis methods have been widely adopted to obtain representative features and used to classify fault modes due to their prominent feature extraction capability. However, a large number of labeled samples are required to support the algorithm of CNNs, and, in the case of a limited amount of labeled samples, this may lead to overfitting. In this article, a novel ResNet-based method is developed to achieve fault diagnoses for machines with very few samples. To be specific, data transformation combinations (DTCs) are designed based on mutual information. It is worth noting that the selected DTC, which can complete the training process of the 1-D ResNet quickly without increasing the amount of training data, can be randomly used for any batch training data. Meanwhile, a self-supervised learning method called 1-D SimCLR is adopted to obtain an effective feature encoder, which can be optimized with very few unlabeled samples. Then, a fault diagnosis model named DTC-SimCLR is constructed by combining the selected data transformation combination, the obtained feature encoder and a fully-connected layer-based classifier. In DTC-SimCLR, the parameters of the feature encoder are fixed, and the classifier is trained with very few labeled samples. Two machine fault datasets from a cutting tooth and a bearing are conducted to evaluate the performance of DTC-SimCLR. Testing results show that DTC-SimCLR has superior performance and diagnostic accuracy with very few samples.
基于卷积神经网络(CNN)的故障诊断方法由于其出色的特征提取能力,已被广泛用于获取代表性特征并用于分类故障模式。然而,由于需要大量的有标签样本来支持 CNN 算法,在有标签样本数量有限的情况下,这可能导致过拟合。在本文中,开发了一种新的基于 ResNet 的方法,以实现少量样本机器的故障诊断。具体来说,基于互信息设计了数据变换组合(DTC)。值得注意的是,选择的 DTC 可以在不增加训练数据量的情况下快速完成 1-D ResNet 的训练过程,可以随机用于任何批处理训练数据。同时,采用称为 1-D SimCLR 的自监督学习方法来获得有效的特征编码器,该编码器可以使用很少的未标记样本进行优化。然后,通过结合选定的数据变换组合、获得的特征编码器和基于全连接层的分类器,构建了一个名为 DTC-SimCLR 的故障诊断模型。在 DTC-SimCLR 中,特征编码器的参数是固定的,并且使用很少的有标签样本对分类器进行训练。使用来自切齿和轴承的两个机器故障数据集来评估 DTC-SimCLR 的性能。测试结果表明,DTC-SimCLR 在使用很少的样本时具有优越的性能和诊断准确性。
Med Biol Eng Comput. 2023-9
IEEE Trans Neural Netw Learn Syst. 2024-5
Sensors (Basel). 2024-1-29
Sensors (Basel). 2021-7-13
Sensors (Basel). 2019-6-19