Few-shot-Learning-for-Fault-Diagnosis

KO | EN

gitlite — search

#javascript #llm #android #rust #linux #docker #macos #nodejs #windows #awesome #cli #pytorch #ios #openai

Few-shot-Learning-for-Fault-Diagnosis

(★ 104)

This repository is for the Few-shot Learning for the fault diagnosis of large industrial equipment.

File Explorer

Download Latest Version (.zip)

- - backbone.py
  - Dataload.py
  - main.py
  - models.py
  - options.py
  - report.py
- - backbone.py
  - Dataload.py
  - main.py
  - models.py
  - options.py
  - report.py
- - backbone.py
  - Dataload.py
  - main.py
  - models.py
  - options.py
  - report.py
- - backbone.py
  - Dataload.py
  - main.py
  - models.py
  - options.py
  - report.py
- Readme.md
README.md

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# Few-shot-Learning-for-Fault-Diagnosis
# 小样本学习，深度学习，故障诊断
# Metric-based Meta-learning, Few-shot Learning, Feature Space,  Fault Diagnosis, Limited Data Conditions

This repository is for the few-shot learning with fault diagnosis.

The paper is as follow:

[Metric-based meta-learning model for few-shot fault diagnosis under multiple limited data conditions](https://doi.org/10.1016/j.ymssp.2020.107510)

About how to use the FSM3, one can see following the [FSM3 link](https://github.com/mingzhangPHD/Few-shot-Learning-for-Fault-Diagnosis/tree/main/FSM3)

# Citation

If you use this code and datasets for your research, please consider citing:

```
@article{WANG2021107510,
title = "Metric-based meta-learning model for few-shot fault diagnosis under multiple limited data conditions",
journal = "Mechanical Systems and Signal Processing",
volume = "155",
pages = "107510",
year = "2021",
issn = "0888-3270",
doi = "https://doi.org/10.1016/j.ymssp.2020.107510",
url = "http://www.sciencedirect.com/science/article/pii/S0888327020308967",
author = "Duo Wang and Ming Zhang and Yuchun Xu and Weining Lu and Jun Yang and Tao Zhang",
keywords = "Metric-based meta-learning, Few-shot learning, Feature space, Fault diagnosis, Limited data conditions",
abstract = "The real-world large industry has gradually become a data-rich environment with the development of information and sensor technology, making the technology of data-driven fault diagnosis acquire a thriving development and application. The success of these advanced methods depends on the assumption that enough labeled samples for each fault type are available. However, in some practical situations, it is extremely difficult to collect enough data, e.g., when the sudden catastrophic failure happens, only a few samples can be acquired before the system shuts down. This phenomenon leads to the few-shot fault diagnosis aiming at distinguishing the failure attribution accurately under very limited data conditions. In this paper, we propose a new approach, called Feature Space Metric-based Meta-learning Model (FSM3), to overcome the challenge of the few-shot fault diagnosis under multiple limited data conditions. Our method is a mixture of general supervised learning and episodic metric meta-learning, which will exploit both the attribute information from individual samples and the similarity information from sample groups. The experiment results demonstrate that our method outperforms a series of baseline methods on the 1-shot and 5-shot learning tasks of bearing and gearbox fault diagnosis across various limited data conditions. The time complexity and implementation difficulty have been analyzed to show that our method has relatively high feasibility. The feature embedding is visualized by t-SNE to investigate the effectiveness of our proposed model."
}
```

# Contact
If you have any problem about our code, feel free to contact:

zhangming_0706@163.com or m.zhang21@aston.ac.uk

or describe your problem in issues

Few-shot-Learning-for-Fault-Diagnosis

File Explorer

Today's picks

Was this content helpful?

Recent Feedback

Few-shot-Learning-for-Fault-Diagnosis

File Explorer

Today's picks

Was this content helpful?

Recent Feedback

Download README