Learning from Class-imbalanced Data with a Model-Agnostic Framework for Machine Intelligent Diagnosis

Jingyao Wu, Zhibin Zhao, Chuang Sun, Ruqiang Yan and Xuefeng Chen

Reliability Engineering and System Safety, 2021, vol. 216, issue C

Abstract: Considering the difficulty of data acquisition in industry, especially for failure data of large-scale equipment, classification with these class-imbalanced datasets can lead to the problems of minority categories overfitting and majority categories domination. A model-agnostic framework towards class-imbalanced fault diagnosis requirement is proposed to systematically alleviate these problems. Four sub-modules, including Time-series Data Augmentation, Data-Rebalanced sampler, Balanced Margin Loss, and classifier with Dynamic Decision Boundary Balancing are performed to improve recognition accuracy of minority categories without performance degradation on majority categories. Meanwhile, the framework is compatible with general neural networks and provides flexible model candidates to meet the need of feature extraction for different data types. Three case studies on public datasets demonstrate that proposed framework outperformed various state-of-the-art methods.

Keywords: Class imbalance; Fault diagnosis; Deep learning; Neural networks; Imbalanced data; Rotating machinery (search for similar items in EconPapers)
Date: 2021
References: View references in EconPapers View complete reference list from CitEc
Citations: View citations in EconPapers (9)

Downloads: (external link)
http://www.sciencedirect.com/science/article/pii/S0951832021004506
Full text for ScienceDirect subscribers only

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:eee:reensy:v:216:y:2021:i:c:s0951832021004506

DOI: 10.1016/j.ress.2021.107934

Access Statistics for this article

Reliability Engineering and System Safety is currently edited by Carlos Guedes Soares

More articles in Reliability Engineering and System Safety from Elsevier
Bibliographic data for series maintained by Catherine Liu ().