Git Product home page Git Product logo

mldso's Introduction

MLDSO

github-profile-readme-generator license github-profile-readme-generator forks github-profile-readme-generator stars github-profile-readme-generator issues github-profile-readme-generator pull-requests

This is the condensed source codes for the paper Few-shot bearing fault diagnosis based on meta-learning with discriminant space optimization which published in Measurement Science and Technology.

Thanks for the Case Western University Bearing Data Center for providing the experimental data.

The repository works with Python 3.8.5 and PyTorch 1.8.1.

Citation

If you want to use the code, please cite the following paper:

@article{Zhang_2022,
	doi = {10.1088/1361-6501/ac8303},
	url = {https://doi.org/10.1088/1361-6501/ac8303},
	year = 2022,
	month = {aug},
	publisher = {{IOP} Publishing},
	volume = {33},
	number = {11},
	pages = {115024},
	author = {Dengming Zhang and Kai Zheng and Yin Bai and Dengke Yao and Dewei Yang and Shaowang Wang},
	title = {Few-shot bearing fault diagnosis based on meta-learning with discriminant space optimization},
	journal = {Measurement Science and Technology},
	abstract = {In practical industrial applications, the collected fault data are usually insufficient due to the sudden occurrence of faults. However, the current deep-learning-based fault diagnosis methods often rely on a large number of samples to achieve satisfactory performance. Moreover, the heavy background noise and the variability of working conditions also degrade the performance of existing fault diagnostic approaches. To address these challenges, a new fault diagnosis method for few-shot bearing fault diagnosis based on meta-learning with discriminant space optimization (MLDSO) is proposed in this research. First, the fault feature of the rolling bearing is extracted through the tailored networks. Then, the feature extractor is optimized by the discriminant space loss proposed in this paper, to promote the clustering of the extracted fault features of the same category and to distinguish between different types of fault features. Next, the feature extractor and discriminant space optimizer are constructed to optimize the feature discriminant space; thus, a high fault-tolerant discriminant space is obtained for meta-learning. Eventually, the faults in the new task can be accurately classified with the assistance of previously learned meta-knowledge and a few known samples when dealing with new tasks under different working conditions. The effectiveness and superiority of the proposed MLDSO method are verified via the datasets collected from our self-designed experimental platform and the Case Western Reserve University test platform. The experimental results show superior performance over the advanced methods. This indicates that the proposed method is a promising approach under small sample situations, heavy noise, and variable working conditions.}
}

Usage

You can use the following command to run the code:

# Clone the repository
git clone [email protected]:Littleor/MLDSO.git
# Change the directory
cd MLDSO
# Install the requirements
pip install -r requirements.txt
# Run the test
python train.py

If you want change the parameters, you can modify the config.py file.

mldso's People

Contributors

littleor avatar

Stargazers

 avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.