/Deep-discriminative-transfer-learning-network-for-cross-machine-fault-diagnosis

Deep discriminative transfer learning network for cross-machine fault diagnosis

Primary LanguagePython

DDTLN

I am just reproducing the code and only accept emails with errors about the source code. For more detailed content such as article algorithms, please contact the original first author: Quan Qian: qian_1998#cqu.edu.cn

DownLoad Dataset：

Official Materials

Cited:

@article{QIAN2023109884,
title = {Deep discriminative transfer learning network for cross-machine fault diagnosis},
journal = {Mechanical Systems and Signal Processing},
volume = {186},
pages = {109884},
year = {2023},
issn = {0888-3270},
doi = {https://doi.org/10.1016/j.ymssp.2022.109884},
url = {https://www.sciencedirect.com/science/article/pii/S0888327022009529},
author = {Quan Qian and Yi Qin and Jun Luo and Yi Wang and Fei Wu},
keywords = {Discriminative feature learning, Joint domain adaptation, Distribution alignment, Classification loss, Fault transfer diagnosis},
abstract = {Many domain adaptation methods have been presented to deal with the distribution alignment and knowledge transfer between the target domain and the source domain. However, most of them only pay attention to marginal distribution alignment while neglecting the discriminative feature learning in two domains. Thus, they still cannot satisfy the diagnosis requirement in some cases. To enhance the distribution alignment and match the marginal distributions as well as conditional distributions of two domains, an improved joint distribution adaptation (IJDA) mechanism is proposed. In IJDA, to enhance domain confusion, maximum mean discrepancy and CORrelation Alignment (CORAL) are combined as a new distribution discrepancy metric. Furthermore, an improved conditional distribution alignment mechanism is constructed. To contribute to feature learning and learn more separable features, a new I-Softmax loss that can be optimized like the original Softmax loss and possesses a stronger classification ability is proposed. With the IJDA mechanism and I-Softmax loss, the deep discriminative transfer learning network (DDTLN) is built to implement fault transfer diagnosis. Under the unlabeled target-domain samples, the experimental results on six cross-machine diagnostic tasks verify that the proposed DDTLN has a higher performance of transfer fault diagnosis than other typical domain adaptation methods.}
}

Contact

  • Chao He
  • 22110435#bjtu.edu.cn (please replace # by @)