A novel framework based on adaptive Multi-Task learning for bearing fault diagnosis
Bearing fault diagnosis is very important for the security and efficiency of electric machines. In recent years, the newly emerging deep learning methods have risen bearing fault diagnosis as a research hotspot again. To achieve better performance and interpretability and absorb novel methods, this...
| Main Authors: | , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
Elsevier
2023-09-01
|
| Series: | Energy Reports |
| Subjects: | |
| Online Access: | http://www.sciencedirect.com/science/article/pii/S2352484723006285 |
| _version_ | 1827820676793237504 |
|---|---|
| author | Jierui Zhang Jianjun Chen Huiwen Deng Weihao Hu |
| author_facet | Jierui Zhang Jianjun Chen Huiwen Deng Weihao Hu |
| author_sort | Jierui Zhang |
| collection | DOAJ |
| description | Bearing fault diagnosis is very important for the security and efficiency of electric machines. In recent years, the newly emerging deep learning methods have risen bearing fault diagnosis as a research hotspot again. To achieve better performance and interpretability and absorb novel methods, this paper proposes a novel framework based on adaptive Multi-Task Learning for bearing fault diagnosis, Gramian Angular Fields - Markov Transition Fields - Convolutional Neural Network (GM-CNN), including data augmentation, image encoding methods and adaptive Multi-Task Learning (MTL). Firstly, data augmentation (DA) methods are applied in data preprocessing for tackling the problems of lack of data and weak generalization ability. They include cropping, flipping, and noise injection. Secondly, Gramian Angular Fields (GAF), Markov Transition Fields (MTF), and their combination GAF-MTF are used to encode time series into images which are more proper for convolutional neural network (CNN) to extract features and classify. Then, the processed data are fed into the proposed MTL framework, and tasks for classifying the fault type and severity are trained jointly as they share some general knowledge and this saves more time. Besides, attention is applied to make the MTL adaptive, which is helpful for more balanced training. Some experiments are carried out. And the experiment results show that the proposed framework is relatively simple but more effective, classifying the fault type and severity with high accuracy (basically higher than 99%). It is shown that every step of the framework is important and essential. This paper provides a reference for future studies on bearing fault diagnosis from the perspective of feature extraction and CNN. It could also be applied to other time series classification situations which could be promising directions. |
| first_indexed | 2024-03-12T01:31:42Z |
| format | Article |
| id | doaj.art-d76d5454f3374a258448110d946542bb |
| institution | Directory Open Access Journal |
| issn | 2352-4847 |
| language | English |
| last_indexed | 2024-03-12T01:31:42Z |
| publishDate | 2023-09-01 |
| publisher | Elsevier |
| record_format | Article |
| series | Energy Reports |
| spelling | doaj.art-d76d5454f3374a258448110d946542bb2023-09-12T04:15:46ZengElsevierEnergy Reports2352-48472023-09-019522531A novel framework based on adaptive Multi-Task learning for bearing fault diagnosisJierui Zhang0Jianjun Chen1Huiwen Deng2Weihao Hu3School of Mechanical and Electrical Engineering, University of Electronic Science and Technology of China, No.2006, Xiyuan Avenue, West Hi-tech Zone, Chengdu 611731, ChinaSchool of Mechanical and Electrical Engineering, University of Electronic Science and Technology of China, No.2006, Xiyuan Avenue, West Hi-tech Zone, Chengdu 611731, China; Corresponding author.School of Mechanical and Electrical Engineering, University of Electronic Science and Technology of China, No.2006, Xiyuan Avenue, West Hi-tech Zone, Chengdu 611731, China; Sichuan Energy Industry Investment Group CO., LTD, Chengdu, Sichuan, 610000, ChinaSchool of Mechanical and Electrical Engineering, University of Electronic Science and Technology of China, No.2006, Xiyuan Avenue, West Hi-tech Zone, Chengdu 611731, ChinaBearing fault diagnosis is very important for the security and efficiency of electric machines. In recent years, the newly emerging deep learning methods have risen bearing fault diagnosis as a research hotspot again. To achieve better performance and interpretability and absorb novel methods, this paper proposes a novel framework based on adaptive Multi-Task Learning for bearing fault diagnosis, Gramian Angular Fields - Markov Transition Fields - Convolutional Neural Network (GM-CNN), including data augmentation, image encoding methods and adaptive Multi-Task Learning (MTL). Firstly, data augmentation (DA) methods are applied in data preprocessing for tackling the problems of lack of data and weak generalization ability. They include cropping, flipping, and noise injection. Secondly, Gramian Angular Fields (GAF), Markov Transition Fields (MTF), and their combination GAF-MTF are used to encode time series into images which are more proper for convolutional neural network (CNN) to extract features and classify. Then, the processed data are fed into the proposed MTL framework, and tasks for classifying the fault type and severity are trained jointly as they share some general knowledge and this saves more time. Besides, attention is applied to make the MTL adaptive, which is helpful for more balanced training. Some experiments are carried out. And the experiment results show that the proposed framework is relatively simple but more effective, classifying the fault type and severity with high accuracy (basically higher than 99%). It is shown that every step of the framework is important and essential. This paper provides a reference for future studies on bearing fault diagnosis from the perspective of feature extraction and CNN. It could also be applied to other time series classification situations which could be promising directions.http://www.sciencedirect.com/science/article/pii/S2352484723006285Bearing fault diagnosisMulti-Task Learning (MTL)Convolutional neural network (CNN)Gramian Angular Fields (GAF)Markov Transition Fields (MTF)Data augmentation (DA) |
| spellingShingle | Jierui Zhang Jianjun Chen Huiwen Deng Weihao Hu A novel framework based on adaptive Multi-Task learning for bearing fault diagnosis Energy Reports Bearing fault diagnosis Multi-Task Learning (MTL) Convolutional neural network (CNN) Gramian Angular Fields (GAF) Markov Transition Fields (MTF) Data augmentation (DA) |
| title | A novel framework based on adaptive Multi-Task learning for bearing fault diagnosis |
| title_full | A novel framework based on adaptive Multi-Task learning for bearing fault diagnosis |
| title_fullStr | A novel framework based on adaptive Multi-Task learning for bearing fault diagnosis |
| title_full_unstemmed | A novel framework based on adaptive Multi-Task learning for bearing fault diagnosis |
| title_short | A novel framework based on adaptive Multi-Task learning for bearing fault diagnosis |
| title_sort | novel framework based on adaptive multi task learning for bearing fault diagnosis |
| topic | Bearing fault diagnosis Multi-Task Learning (MTL) Convolutional neural network (CNN) Gramian Angular Fields (GAF) Markov Transition Fields (MTF) Data augmentation (DA) |
| url | http://www.sciencedirect.com/science/article/pii/S2352484723006285 |
| work_keys_str_mv | AT jieruizhang anovelframeworkbasedonadaptivemultitasklearningforbearingfaultdiagnosis AT jianjunchen anovelframeworkbasedonadaptivemultitasklearningforbearingfaultdiagnosis AT huiwendeng anovelframeworkbasedonadaptivemultitasklearningforbearingfaultdiagnosis AT weihaohu anovelframeworkbasedonadaptivemultitasklearningforbearingfaultdiagnosis AT jieruizhang novelframeworkbasedonadaptivemultitasklearningforbearingfaultdiagnosis AT jianjunchen novelframeworkbasedonadaptivemultitasklearningforbearingfaultdiagnosis AT huiwendeng novelframeworkbasedonadaptivemultitasklearningforbearingfaultdiagnosis AT weihaohu novelframeworkbasedonadaptivemultitasklearningforbearingfaultdiagnosis |