Fault Diagnosis of Bearings and Gears Based on LiteNet With Feature Aggregation
Qiankun Li, Xin Ma, Mingliang Cui, Hu Yu, Jingfeng Zhao, Youqing Wang
Abstract
Significant progress has been made in the current fault diagnosis algorithms. However, they do not consider computational resources and require expensive equipment to complete the training of the models. To immediately complete model training and obtain higher accuracy rates using cheaper equipment, reduce the equipment cost in the industry, and build a bridge for industrial fault diagnosis with neural networks, this paper proposes a convolutional neural network-based architecture that uses a small number of computational resources with high accuracy. Simultaneously, a loss function is proposed that can further improve the accuracy of the network model without consuming too many computational resources. According to experimental comparisons, the proposed technique has clear advantages in terms of accuracy, computational resources, training time, and stability.