Lightweight CNN Architecture Design Based on Spatial-Temporal Tensor and Its Application in Bearing Fault Diagnosis

Bearing is a failure-prone component in rotating machinery equipment, and various task conditions make bearing fault diagnosis (BFD) a challenging task. Despite the great success of convolutional neural network (CNN) in feature extraction, most of them are based on manual tuning, and it is not easy...

Celý popis

Uloženo v:
Podrobná bibliografie
Vydáno v:IEEE transactions on instrumentation and measurement Ročník 73; s. 1 - 12
Hlavní autoři: Wang, Zan, Lu, Hui, Shi, Yuhui, Wang, Xianpeng
Médium: Journal Article
Jazyk:angličtina
Vydáno: New York IEEE 2024
The Institute of Electrical and Electronics Engineers, Inc. (IEEE)
Témata:
Artificial neural networks
Automatic architecture design
bearing fault diagnosis (BFD)
convolutional neural network (CNN)
Convolutional neural networks
Deep learning
Fault diagnosis
Feature extraction
Floating point arithmetic
genetic algorithm
Genetic algorithms
Lightweight
Machine learning
Mathematical analysis
Rotating machinery
Space exploration
Tensors
Vibrations
ISSN:0018-9456, 1557-9662
On-line přístup:Získat plný text
Tagy: Přidat tag
Žádné tagy, Buďte první, kdo vytvoří štítek k tomuto záznamu!
Popis
Shrnutí:Bearing is a failure-prone component in rotating machinery equipment, and various task conditions make bearing fault diagnosis (BFD) a challenging task. Despite the great success of convolutional neural network (CNN) in feature extraction, most of them are based on manual tuning, and it is not easy to determine a unified CNN architecture with an effective balance between accuracy and computational resources [parameters and floating point operations (FLOPs)]. In this article, an automatic CNN architecture design method called AUTO-CNN is proposed to address the above issues. First, a novel representation learning method (RLM) based on spatial-temporal tensors (STTs) is proposed, in which AUTO-CNN can use 2-D multiscale convolutions to capture more feature hierarchy, i.e., temporal variabilities and spatial characteristics. Second, unlike most of the current nontask-specific studies that establish deeper and more complex networks for better performance, this article investigates lightweight CNNs with limited resources under task-specific applications. Finally, to speed up the design process, an efficient hierarchical encoding strategy and space exploration strategy are proposed to avoid inefficient architecture generation and reduce CNNs training time. Experiments on four real-world BFD tasks demonstrate that the proposed method outperforms the peer competitors, including machine learning, deep learning, and other CNN-based methods.
Bibliografie:ObjectType-Article-1
SourceType-Scholarly Journals-1
ObjectType-Feature-2
content type line 14
ISSN:0018-9456
1557-9662
DOI:10.1109/TIM.2023.3336435