Discriminative fault diagnosis transfer learning network under joint mechanism

Abstract Unsupervised fault diagnosis methods for rotating machinery are gaining attention but face challenges such as feature extraction from vibration signals, aligning distributions between source and target domains, and managing domain shifts. This paper proposes a novel unsupervised transfer le...

Full description

Saved in:
Bibliographic Details
Main Authors: Yuxuan Yang, Jiarui Jing, Jian Zhang, Ziyu Liu, Xueyi Li
Format: Article
Language:English
Published: Nature Portfolio 2025-03-01
Series:Scientific Reports
Subjects:
Online Access:https://doi.org/10.1038/s41598-025-93996-1
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1850039913651634176
author Yuxuan Yang
Jiarui Jing
Jian Zhang
Ziyu Liu
Xueyi Li
author_facet Yuxuan Yang
Jiarui Jing
Jian Zhang
Ziyu Liu
Xueyi Li
author_sort Yuxuan Yang
collection DOAJ
description Abstract Unsupervised fault diagnosis methods for rotating machinery are gaining attention but face challenges such as feature extraction from vibration signals, aligning distributions between source and target domains, and managing domain shifts. This paper proposes a novel unsupervised transfer learning method that integrates the Squeeze-and-Excitation (SE) attention mechanism to enhance useful features while suppressing redundant ones. An Integrated Distribution Alignment Framework (IDAF) is introduced, which employs the Joint Adaptation Network (JAN) approach to construct a local maximum mean discrepancy in conjunction with Correlation Alignment (CORAL) to improve distribution alignment between domains. Moreover, to enhance feature learning and obtain more distinct features, the authors utilize a novel discriminative feature learning method called I-Softmax loss. This method can be optimized in a manner similar to the traditional Softmax loss while providing improved classification performance. Finally, deep adversarial training is applied between the source and target domains to adaptively optimize the target domain network parameters, reducing domain shift and improving fault classification accuracy. Experimental validation using four sets of bearing faults and six sets of gear faults demonstrates the superior performance of the proposed method in unsupervised fault diagnosis tasks.
format Article
id doaj-art-8eaf4c718347459faadb2653fe7409ed
institution DOAJ
issn 2045-2322
language English
publishDate 2025-03-01
publisher Nature Portfolio
record_format Article
series Scientific Reports
spelling doaj-art-8eaf4c718347459faadb2653fe7409ed2025-08-20T02:56:12ZengNature PortfolioScientific Reports2045-23222025-03-0115111710.1038/s41598-025-93996-1Discriminative fault diagnosis transfer learning network under joint mechanismYuxuan Yang0Jiarui Jing1Jian Zhang2Ziyu Liu3Xueyi Li4College of Mechanical and Electrical Engineering, Northeast Forestry UniversityCollege of Mechanical and Electrical Engineering, Northeast Forestry UniversityCollege of Mechanical and Electrical Engineering, Northeast Forestry UniversityCollege of Mechanical and Electrical Engineering, Northeast Forestry UniversityCollege of Mechanical and Electrical Engineering, Northeast Forestry UniversityAbstract Unsupervised fault diagnosis methods for rotating machinery are gaining attention but face challenges such as feature extraction from vibration signals, aligning distributions between source and target domains, and managing domain shifts. This paper proposes a novel unsupervised transfer learning method that integrates the Squeeze-and-Excitation (SE) attention mechanism to enhance useful features while suppressing redundant ones. An Integrated Distribution Alignment Framework (IDAF) is introduced, which employs the Joint Adaptation Network (JAN) approach to construct a local maximum mean discrepancy in conjunction with Correlation Alignment (CORAL) to improve distribution alignment between domains. Moreover, to enhance feature learning and obtain more distinct features, the authors utilize a novel discriminative feature learning method called I-Softmax loss. This method can be optimized in a manner similar to the traditional Softmax loss while providing improved classification performance. Finally, deep adversarial training is applied between the source and target domains to adaptively optimize the target domain network parameters, reducing domain shift and improving fault classification accuracy. Experimental validation using four sets of bearing faults and six sets of gear faults demonstrates the superior performance of the proposed method in unsupervised fault diagnosis tasks.https://doi.org/10.1038/s41598-025-93996-1SE attention mechanismDomain adaptationClassification lossConditional adversarial network
spellingShingle Yuxuan Yang
Jiarui Jing
Jian Zhang
Ziyu Liu
Xueyi Li
Discriminative fault diagnosis transfer learning network under joint mechanism
Scientific Reports
SE attention mechanism
Domain adaptation
Classification loss
Conditional adversarial network
title Discriminative fault diagnosis transfer learning network under joint mechanism
title_full Discriminative fault diagnosis transfer learning network under joint mechanism
title_fullStr Discriminative fault diagnosis transfer learning network under joint mechanism
title_full_unstemmed Discriminative fault diagnosis transfer learning network under joint mechanism
title_short Discriminative fault diagnosis transfer learning network under joint mechanism
title_sort discriminative fault diagnosis transfer learning network under joint mechanism
topic SE attention mechanism
Domain adaptation
Classification loss
Conditional adversarial network
url https://doi.org/10.1038/s41598-025-93996-1
work_keys_str_mv AT yuxuanyang discriminativefaultdiagnosistransferlearningnetworkunderjointmechanism
AT jiaruijing discriminativefaultdiagnosistransferlearningnetworkunderjointmechanism
AT jianzhang discriminativefaultdiagnosistransferlearningnetworkunderjointmechanism
AT ziyuliu discriminativefaultdiagnosistransferlearningnetworkunderjointmechanism
AT xueyili discriminativefaultdiagnosistransferlearningnetworkunderjointmechanism