Giter VIP home page Giter VIP logo

few-shot-transfer-learning's Introduction

Few-shot Transfer Learning for Intelligent Fault Diagnosis of Machine

PyTorch code for paper: Few-shot Transfer Learning for Intelligent Fault Diagnosis of Machine

Data

For PU dataset experiments, please download PU dataset and replace the 'root' in 'finetune_generator' amd 'transfer_generator' with your local data path

Run

meta transfer learning 1 shot:

python main_meta_transfer.py -s 1

feature transfer 1 shot:

python main_finetune.py -s 1 -u 0

unfrozen 1 fine-tune 1 shot:

python main_finetune.py -s 1 -u 1

unfrozen 2 fine-tune 1 shot:

python main_finetune.py -s 1 -u 2

unfrozen 3 fine-tune 1 shot:

python main_finetune.py -s 1 -u 3

all fine-tune 1 shot:

python main_finetune.py -s 1 -u 4

direct training 1 shot:

python main_direct.py -s 1

you can change -b parameter based on your GPU memory.

Citing

If you use this code in your research, please use the following BibTeX entry.

@article{WU2020108202,
title = "Few-shot Transfer Learning for Intelligent Fault Diagnosis of Machine",
journal = "Measurement",
pages = "108202",
year = "2020",
issn = "0263-2241",
doi = "https://doi.org/10.1016/j.measurement.2020.108202",
url = "http://www.sciencedirect.com/science/article/pii/S0263224120307405",
author = "Jingyao Wu and Zhibin Zhao and Chuang Sun and Ruqiang Yan and Xuefeng Chen",
keywords = "few-shot learning, intelligent diagnosis, transfer learning, meta-learning, rotating machinery",
abstract = "Rotating machinery intelligent diagnosis with large data has been researched comprehensively, while there is still a gap between the existing diagnostic model and the practical application, due to the variability of working conditions and the scarcity of fault samples. To address this problem, few-shot transfer learning method is constructed utilizing meta-learning for few-shot samples diagnosis in variable conditions in this paper. We consider two transfer situations of rotating machinery intelligent diagnosis named conditions transfer and artificial-to-natural transfer, and construct seven few-shot transfer learning methods based on a unified 1D convolution network for few-shot diagnosis of three datasets. Baseline accuracy under different sample capacity and transfer situations are provided for comprehensive comparison and guidelines. What is more, data dependency, transferability, and task plasticity of various methods in the few-shot scenario are discussed in detail, the data analysis result shows meta-learning holds the advantage for machine fault diagnosis with extremely few-shot instances on the relatively simple transfer task. Our code is available at https://github.com."
}

Reference

LearningToCompare_FSL

few-shot-transfer-learning's People

Contributors

a1018680161 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.