Skip to content

This is a benckmark for domain generalization-based fault diagnosis (基于领域泛化的相关代码)

License

Notifications You must be signed in to change notification settings

CHAOZHAO-1/Domain-generalization-fault-diagnosis-benchmark

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

24 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Domain-Generalization-Fault-Diagnosis-Benchmark

Last update

Details of the benchmark can be found here.

File Description

Index File Name Description Paper
1 CCDG.py Model Conditional Contrastive Domain Generalization for Fault Diagnosis
2 CNN-C.py Model Learn Generalization Feature via Convolutional Neural Network: A Fault Diagnosis Scheme Toward Unseen Operating Conditions
3 DANN.py Model Adversarial Training Among Multiple Source Domains
4 DCORAL.py Model Reduce CORAL Among Multiple Source Domains
5 DDC.py Model Reduce MMD Among Multiple Source Domains
6 DGNIS.py Model A Domain Generalization Network Combining Invariance and Specificity Towards Real-Time Intelligent Fault Diagnosis
7 ERM.py Model Reduce Classification Loss
8 IEDGNet.py Model A Hybrid Generalization Network for Intelligent Fault Diagnosis of Rotating Machinery Under Unseen Working Conditions
9 data_loaded_1d.py Data Preparation /
10 resnet18_1d.py Network /
11 utils.py Metrics /

Dataset Preparation

You can find the data for the DG benchmark here.

Cross-Working Condition

For example:

CWRUTasksetting = {
    'dataset': 'C-CWRU', 
    'class_num': 10, 
    'src_tar': np.array([[0, 1, 2, 3], [0, 1, 3, 2], [0, 2, 3, 1], [1, 2, 3, 0]])
}

In the C-CWRU dataset, there are four working conditions (0, 1, 2, 3) with 10 types of healthy vibration data. Four tasks are created by selecting one condition as the target domain and the other three as the source domains.

Cross-Machine Condition

For example:

datasetlist = [['M_CWRU', 'M_IMS', 'M_JNU', 'M_HUST']]

This indicates using data from CWRU, IMS, and JNU as source domains (combining all working conditions) and the HUST dataset as the target domain (also combining all working conditions).

Contact

If you have any questions, please feel free to contact me:

BibTeX Citation

If you find this paper and repository useful, please cite our paper 😊.

@article{Zhao2024domain,
  title={Domain Generalization for Cross-Domain Fault Diagnosis: an Application-oriented Perspective and a Benchmark Study},
  author={Zhao, Chao and Zio, Enrico and Shen, Weiming},
  journal={Reliability Engineering & System Safety},
  pages={109964},
  year={2024}
}

About

This is a benckmark for domain generalization-based fault diagnosis (基于领域泛化的相关代码)

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages