IDEAS home Printed from https://ideas.repec.org/a/eee/reensy/v243y2024ics0951832023007469.html
   My bibliography  Save this article

A novel data augmentation approach to fault diagnosis with class-imbalance problem

Author

Listed:
  • Tian, Jilun
  • Jiang, Yuchen
  • Zhang, Jiusi
  • Luo, Hao
  • Yin, Shen

Abstract

Data-driven fault diagnosis techniques are frequently applied to ensure the reliability and safety of industrial systems. However, as a common challenge, the class-imbalance problem reduces the performance of data-driven methods due to the lack of data information. We propose a weighted modified conditional variational auto-encoder (WM-CVAE) as a novel data augmentation technique to tackle the issue. The modified structure can alleviate the existing Kullback–Leibler (KL) divergence vanishing by an adaptive loss. Meanwhile, kernel mean matching (KMM) is proposed on weight computation to reduce the negative effect of dissimilar generated samples. Constructing the WM-CVAE data augmentation framework can effectively improve the data quality and learning capability in class-imbalance fault diagnosis. To validate the proposed WM-CVAE model, three real-world industrial datasets are used as study objects, and the random forest is used as the base learner in the fault classification tasks. The diagnostic results demonstrate that the proposed WM-CVAE data augmentation framework can improve learning results in class-imbalance fault diagnosis.

Suggested Citation

  • Tian, Jilun & Jiang, Yuchen & Zhang, Jiusi & Luo, Hao & Yin, Shen, 2024. "A novel data augmentation approach to fault diagnosis with class-imbalance problem," Reliability Engineering and System Safety, Elsevier, vol. 243(C).
    • Handle: RePEc:eee:reensy:v:243:y:2024:i:c:s0951832023007469
    DOI: 10.1016/j.ress.2023.109832
    as

    Download full text from publisher

    File URL: http://www.sciencedirect.com/science/article/pii/S0951832023007469
    Download Restriction: Full text for ScienceDirect subscribers only
    File URL: https://libkey.io/10.1016/j.ress.2023.109832?utm_source=ideas
    LibKey link: if access is restricted and if your library uses this service, LibKey will redirect you to where you can use your library subscription to access this item
    ---><---

    As the access to this document is restricted, you may want to search for a different version of it.

    References listed on IDEAS

    as
    1. Zio, Enrico, 2022. "Prognostics and Health Management (PHM): Where are we and where do we (need to) go in theory and practice," Reliability Engineering and System Safety, Elsevier, vol. 218(PA).
    2. Leoni, Leonardo & De Carlo, Filippo & Abaei, Mohammad Mahdi & BahooToroody, Ahmad & Tucci, Mario, 2023. "Failure diagnosis of a compressor subjected to surge events: A data-driven framework," Reliability Engineering and System Safety, Elsevier, vol. 233(C).
    3. Pedro Santos & Jesús Maudes & Andres Bustillo, 2018. "Identifying maximum imbalance in datasets for fault diagnosis of gearboxes," Journal of Intelligent Manufacturing, Springer, vol. 29(2), pages 333-351, February.
    4. Shi, Peiming & Wu, Shuping & Xu, Xuefang & Zhang, Bofei & Liang, Pengfei & Qiao, Zijian, 2023. "TSN: A novel intelligent fault diagnosis method for bearing with small samples under variable working conditions," Reliability Engineering and System Safety, Elsevier, vol. 240(C).
    5. Chaleshtori, Amir Eshaghi & Aghaie, Abdollah, 2024. "A novel bearing fault diagnosis approach using the Gaussian mixture model and the weighted principal component analysis," Reliability Engineering and System Safety, Elsevier, vol. 242(C).
    6. Wu, Jingyao & Zhao, Zhibin & Sun, Chuang & Yan, Ruqiang & Chen, Xuefeng, 2021. "Learning from Class-imbalanced Data with a Model-Agnostic Framework for Machine Intelligent Diagnosis," Reliability Engineering and System Safety, Elsevier, vol. 216(C).
    7. Melani, Arthur Henrique de Andrade & Michalski, Miguel Angelo de Carvalho & da Silva, Renan Favarão & de Souza, Gilberto Francisco Martha, 2021. "A framework to automate fault detection and diagnosis based on moving window principal component analysis and Bayesian network," Reliability Engineering and System Safety, Elsevier, vol. 215(C).
    Full references (including those not matched with items on IDEAS)

    Citations

    Citations are extracted by the CitEc Project, subscribe to its RSS feed for this item.
    as


    Cited by:

    1. Zhang, Jiusi & Tian, Jilun & Yan, Pengfei & Wu, Shimeng & Luo, Hao & Yin, Shen, 2024. "Multi-hop graph pooling adversarial network for cross-domain remaining useful life prediction: A distributed federated learning perspective," Reliability Engineering and System Safety, Elsevier, vol. 244(C).

    Most related items

    These are the items that most often cite the same works as this one and are cited by the same works as this one.
    1. Chen, Zhen & Zhou, Di & Zio, Enrico & Xia, Tangbin & Pan, Ershun, 2023. "Adaptive transfer learning for multimode process monitoring and unsupervised anomaly detection in steam turbines," Reliability Engineering and System Safety, Elsevier, vol. 234(C).
    2. Mingfei Li & Jiajian Wu & Zhengpeng Chen & Jiangbo Dong & Zhiping Peng & Kai Xiong & Mumin Rao & Chuangting Chen & Xi Li, 2022. "Data-Driven Voltage Prognostic for Solid Oxide Fuel Cell System Based on Deep Learning," Energies, MDPI, vol. 15(17), pages 1-20, August.
    3. Liu, Jiale & Wang, Huan, 2024. "A brain-inspired energy-efficient Wide Spiking Residual Attention Framework for intelligent fault diagnosis," Reliability Engineering and System Safety, Elsevier, vol. 243(C).
    4. Kamei, Sayaka & Taghipour, Sharareh, 2023. "A comparison study of centralized and decentralized federated learning approaches utilizing the transformer architecture for estimating remaining useful life," Reliability Engineering and System Safety, Elsevier, vol. 233(C).
    5. Zuo, Jian & Cadet, Catherine & Li, Zhongliang & Bérenguer, Christophe & Outbib, Rachid, 2024. "A deterioration-aware energy management strategy for the lifetime improvement of a multi-stack fuel cell system subject to a random dynamic load," Reliability Engineering and System Safety, Elsevier, vol. 241(C).
    6. Costa, Nahuel & Sánchez, Luciano, 2022. "Variational encoding approach for interpretable assessment of remaining useful life estimation," Reliability Engineering and System Safety, Elsevier, vol. 222(C).
    7. Jorge Maldonado-Correa & Marcelo Valdiviezo-Condolo & Estefanía Artigao & Sergio Martín-Martínez & Emilio Gómez-Lázaro, 2024. "Classification of Highly Imbalanced Supervisory Control and Data Acquisition Data for Fault Detection of Wind Turbine Generators," Energies, MDPI, vol. 17(7), pages 1-20, March.
    8. Lewis, Austin D. & Groth, Katrina M., 2022. "Metrics for evaluating the performance of complex engineering system health monitoring models," Reliability Engineering and System Safety, Elsevier, vol. 223(C).
    9. Li, Xiao Yan & Cheng, De Jun & Fang, Xi Feng & Zhang, Chun Yan & Wang, Yu Feng, 2024. "A novel data augmentation strategy for aeroengine multitask prognosis based on degradation behavior extrapolation and diversity-usability trade-off," Reliability Engineering and System Safety, Elsevier, vol. 249(C).
    10. Zhao, Chao & Shen, Weiming, 2022. "Adaptive open set domain generalization network: Learning to diagnose unknown faults under unknown working conditions," Reliability Engineering and System Safety, Elsevier, vol. 226(C).
    11. Zheng, Shuwen & Wang, Chong & Zio, Enrico & Liu, Jie, 2024. "Fault detection in complex mechatronic systems by a hierarchical graph convolution attention network based on causal paths," Reliability Engineering and System Safety, Elsevier, vol. 243(C).
    12. Zaitseva, Elena & Levashenko, Vitaly & Rabcan, Jan, 2023. "A new method for analysis of Multi-State systems based on Multi-valued decision diagram under epistemic uncertainty," Reliability Engineering and System Safety, Elsevier, vol. 229(C).
    13. Lyu, Dongzhen & Niu, Guangxing & Liu, Enhui & Zhang, Bin & Chen, Gang & Yang, Tao & Zio, Enrico, 2022. "Time space modelling for fault diagnosis and prognosis with uncertainty management: A general theoretical formulation," Reliability Engineering and System Safety, Elsevier, vol. 226(C).
    14. Liu, Jie & Xu, Yubo & Wang, Lisong, 2022. "Fault information mining with causal network for railway transportation system," Reliability Engineering and System Safety, Elsevier, vol. 220(C).
    15. Hao, Zhaojun & Di Maio, Francesco & Zio, Enrico, 2023. "A sequential decision problem formulation and deep reinforcement learning solution of the optimization of O&M of cyber-physical energy systems (CPESs) for reliable and safe power production and supply," Reliability Engineering and System Safety, Elsevier, vol. 235(C).
    16. Han, Te & Li, Yan-Fu, 2022. "Out-of-distribution detection-assisted trustworthy machinery fault diagnosis approach with uncertainty-aware deep ensembles," Reliability Engineering and System Safety, Elsevier, vol. 226(C).
    17. Chen, Rentong & Zhang, Chao & Wang, Shaoping & Zio, Enrico & Dui, Hongyan & Zhang, Yadong, 2023. "Importance measures for critical components in complex system based on Copula Hierarchical Bayesian Network," Reliability Engineering and System Safety, Elsevier, vol. 230(C).
    18. Li, Jimeng & Cheng, Xing & Peng, Junling & Meng, Zong, 2022. "A new adaptive parallel resonance system based on cascaded feedback model of vibrational resonance and stochastic resonance and its application in fault detection of rolling bearings," Chaos, Solitons & Fractals, Elsevier, vol. 164(C).
    19. Zou, Xinyu & Tao, Laifa & Sun, Lulu & Wang, Chao & Ma, Jian & Lu, Chen, 2023. "A case-learning-based paradigm for quantitative recommendation of fault diagnosis algorithms: A case study of gearbox," Reliability Engineering and System Safety, Elsevier, vol. 237(C).
    20. Zhao, Shuaiyu & Duan, Yiling & Roy, Nitin & Zhang, Bin, 2024. "A deep learning methodology based on adaptive multiscale CNN and enhanced highway LSTM for industrial process fault diagnosis," Reliability Engineering and System Safety, Elsevier, vol. 249(C).

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:eee:reensy:v:243:y:2024:i:c:s0951832023007469. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    If CitEc recognized a bibliographic reference but did not link an item in RePEc to it, you can help with this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: Catherine Liu (email available below). General contact details of provider: https://www.journals.elsevier.com/reliability-engineering-and-system-safety .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.