IDEAS home Printed from https://ideas.repec.org/a/eee/reensy/v243y2024ics0951832023007469.html

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

    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. 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).
    3. 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).
    4. 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).
    5. 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).
    6. 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).
    7. 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.
    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. Li, Duowei & Wong, Yiik Diew & Chen, Tianyi & Wang, Nanxi & Yuen, Kum Fai, 2024. "An ensemble method for investigating maritime casualties resulting in pollution occurrence: Data augmentation and feature analysis," Reliability Engineering and System Safety, Elsevier, vol. 251(C).
    2. Wang, Haoyu & Li, Chuanjiang & Ding, Peng & Li, Shaobo & Li, Tandong & Liu, Chenyu & Zhang, Xiangjie & Hong, Zejian, 2024. "A novel transformer-based few-shot learning method for intelligent fault diagnosis with noisy labels under varying working conditions," Reliability Engineering and System Safety, Elsevier, vol. 251(C).
    3. 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).
    4. Dong, Haoyan & Peng, Chuang & Chen, Lei & Hao, Kuangrong, 2025. "Dynamic ensemble fault diagnosis framework with adaptive hierarchical sampling strategy for industrial imbalanced and overlapping data," Reliability Engineering and System Safety, Elsevier, vol. 260(C).
    5. Zhou, Tao & Yao, Dechen & Yang, Jianwei & Meng, Chang & Li, Ankang & Li, Xi, 2024. "DRSwin-ST: An intelligent fault diagnosis framework based on dynamic threshold noise reduction and sparse transformer with Shifted Windows," Reliability Engineering and System Safety, Elsevier, vol. 250(C).
    6. Mu, Guoqing & Liu, Ching-Lien & Chen, Junghui, 2025. "Integrating physical knowledge for generative-based zero-shot learning models in process fault diagnosis," Reliability Engineering and System Safety, Elsevier, vol. 258(C).
    7. Li, Ying & Zhang, Lijie & Liang, Pengfei & Wang, Xiangfeng & Wang, Bin & Xu, Leitao, 2024. "Semi-supervised meta-path space extended graph convolution network for intelligent fault diagnosis of rotating machinery under time-varying speeds," Reliability Engineering and System Safety, Elsevier, vol. 251(C).
    8. Zhou, Yue & Fu, Chuanyun & Wei, Lin & Zhou, Wengang & Li, Xiuyi & You, Yi, 2026. "An integrated approach for addressing data imbalance in predicting fatality of helicopter accident," Reliability Engineering and System Safety, Elsevier, vol. 267(PB).
    9. Zhang, Xiao & Huang, Weiguo & Huangfu, Yifan & Wang, Jun & Ding, Chuancang & Du, Guifu, 2026. "EnergyMatch: A long-tailed semi-supervised learning framework for cross-domain fault diagnosis," Reliability Engineering and System Safety, Elsevier, vol. 269(C).
    10. Yu, Tian & Li, Chaoshun & Huang, Jie & Xiao, Xiangqu & Zhang, Xiaoyuan & Li, Yuhong & Fu, Bitao, 2024. "ReF-DDPM: A novel DDPM-based data augmentation method for imbalanced rolling bearing fault diagnosis," Reliability Engineering and System Safety, Elsevier, vol. 251(C).
    11. Huang, Yaodi & Song, Yunpeng & Cai, Zhongmin, 2025. "A supervised contrastive learning method with novel data augmentation for transient stability assessment considering sample imbalance," Reliability Engineering and System Safety, Elsevier, vol. 256(C).
    12. Zhang, Xiao & Huang, Weiguo & Wang, Jun & Zhu, Zhongkui & Shen, Changqing & Chen, Kai & Zhong, Xingli & He, Li, 2025. "Adaptive variational sampling-embedded domain generalization network for fault diagnosis with intra-inter-domain class imbalance," Reliability Engineering and System Safety, Elsevier, vol. 256(C).
    13. Cai, Li & Deng, Xuanhong & Yin, Hongpeng & Lin, Jingdong & Qin, Yan, 2025. "Generalized zero-sample industrial fault diagnosis with domain bias," Reliability Engineering and System Safety, Elsevier, vol. 253(C).
    14. Zhang, Guowei & Kong, Xianguang & Wang, Qibin & Du, Jingli & Wang, Jinrui & Ma, Hongbo, 2024. "Single domain generalization method based on anti-causal learning for rotating machinery fault diagnosis," Reliability Engineering and System Safety, Elsevier, vol. 250(C).
    15. Ye, Xin & Zhao, Xiaojie & Dong, Lu-an, 2026. "DK-TGAN-ERRA: A data-knowledge-driven tabular GAN for synthesising cases applied to emergency rescue risk analysis," Reliability Engineering and System Safety, Elsevier, vol. 267(PA).
    16. Lin, Cuiying & Kong, Yun & Huang, Guoyu & Han, Qinkai & Dong, Mingming & Liu, Hui & Chu, Fulei, 2025. "Generalization classification regularization generative adversarial network for machinery fault diagnostics under data imbalance," Reliability Engineering and System Safety, Elsevier, vol. 256(C).
    17. Xing, Zhikai & Liu, Yongbao & Wang, Qiang & Li, Mo, 2026. "Fault diagnosis of inter-shaft bearings in small sample scenarios using multi-sensor spatio-temporal causal fusion and graph conditional variational adversarial networks," Reliability Engineering and System Safety, Elsevier, vol. 268(C).
    18. Wei, Yuan & Xiao, Zhijun & Chen, Xiangyan & Gu, Xiaohui & Schröder, Kai-Uwe, 2025. "A bearing fault data augmentation method based on hybrid-diversity loss diffusion model and parameter transfer," Reliability Engineering and System Safety, Elsevier, vol. 253(C).
    19. Cheng, Pu & Cai, Yuchen & Zhong, Jie & Xu, Zhiqiang & Miao, Qiang, 2026. "Parallel time-series mixer network enhanced by slicing procedure and attention mechanism for remaining useful life prediction," Reliability Engineering and System Safety, Elsevier, vol. 266(PA).
    20. Wu, Zhangjun & Xu, Renli & Luo, Yuansheng & Shao, Haidong, 2024. "A holistic semi-supervised method for imbalanced fault diagnosis of rotational machinery with out-of-distribution samples," Reliability Engineering and System Safety, Elsevier, vol. 250(C).
    21. Liu, Ruonan & Zhang, Quanhu & Lin, Di & Zhang, Weidong & Ding, Steven X., 2024. "Causal intervention graph neural network for fault diagnosis of complex industrial processes," Reliability Engineering and System Safety, Elsevier, vol. 251(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. Yu, Tian & Li, Chaoshun & Huang, Jie & Xiao, Xiangqu & Zhang, Xiaoyuan & Li, Yuhong & Fu, Bitao, 2024. "ReF-DDPM: A novel DDPM-based data augmentation method for imbalanced rolling bearing fault diagnosis," Reliability Engineering and System Safety, Elsevier, vol. 251(C).
    2. Liu, Rui & Ding, Xiaoxi & Liu, Shenglan & Zheng, Hebin & Xu, Yuanyaun & Shao, Yimin, 2025. "Knowledge-informed FIR-based cross-category filtering framework for interpretable machinery fault diagnosis under small samples," Reliability Engineering and System Safety, Elsevier, vol. 254(PB).
    3. Zhao, Hao & Pan, Rong, 2025. "Gaussian Derivative Change-point Detection for early warnings of industrial system failures," Reliability Engineering and System Safety, Elsevier, vol. 256(C).
    4. Xing, Zhikai & Liu, Yongbao & Wang, Qiang & Li, Mo, 2026. "Fault diagnosis of inter-shaft bearings in small sample scenarios using multi-sensor spatio-temporal causal fusion and graph conditional variational adversarial networks," Reliability Engineering and System Safety, Elsevier, vol. 268(C).
    5. 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).
    6. 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.
    7. 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).
    8. He, Yuxuan & Qiao, Lingyun & Zio, Enrico & Su, Huai & Zhang, Li & Yang, Zhaoming & Peng, Shiliang & Zhang, Jinjun, 2026. "A framework based on temporal causal inference graph neural networks for the probabilistic estimation of the remaining useful life of proton exchange membrane fuel cells," Reliability Engineering and System Safety, Elsevier, vol. 265(PB).
    9. Yiping Gao & Liang Gao & Xinyu Li & Yuwei Zheng, 2020. "A zero-shot learning method for fault diagnosis under unknown working loads," Journal of Intelligent Manufacturing, Springer, vol. 31(4), pages 899-909, April.
    10. Zhou, Haoxuan & Wang, Bingsen & Zio, Enrico & Wen, Guangrui & Liu, Zimin & Su, Yu & Chen, Xuefeng, 2023. "Hybrid system response model for condition monitoring of bearings under time-varying operating conditions," Reliability Engineering and System Safety, Elsevier, vol. 239(C).
    11. Zhang, Dingyang & Zhang, Yiming & Li, Pei & Zhang, Shuyou, 2025. "Kernel Reinforcement Learning for sampling-efficient risk management of large-scale engineering systems," Reliability Engineering and System Safety, Elsevier, vol. 260(C).
    12. 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).
    13. Wang, Shi-ao & Wang, Shibin & Wang, Wei & Zheng, Yiming & Ding, Baoqing & Yan, Ruqiang & Chen, Xuefeng, 2025. "Unrolled sparse coding via lifting wavelet for interpretable intelligent fault diagnosis," Reliability Engineering and System Safety, Elsevier, vol. 264(PB).
    14. Pang, Zhenan & Li, Tianmei & Pei, Hong & Si, Xiaosheng, 2023. "A condition-based prognostic approach for age- and state-dependent partially observable nonlinear degrading system," Reliability Engineering and System Safety, Elsevier, vol. 230(C).
    15. Zhang, Mingyuan & He, Chen & Huang, Chengxuan & Yang, Jianhong, 2024. "A weighted time embedding transformer network for remaining useful life prediction of rolling bearing," Reliability Engineering and System Safety, Elsevier, vol. 251(C).
    16. Sandeep S. Udmale & Aneesh G. Nath & Durgesh Singh & Sanjay Kumar Singh, 2026. "Data-driven approach under varying operating conditions for bearing fault diagnosis," Annals of Operations Research, Springer, vol. 358(1), pages 495-528, March.
    17. 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).
    18. Peng, Shiliang & Fan, Lin & Zhang, Li & Su, Huai & He, Yuxuan & He, Qian & Wang, Xiao & Yu, Dejun & Zhang, Jinjun, 2024. "Spatio-temporal prediction of total energy consumption in multiple regions using explainable deep neural network," Energy, Elsevier, vol. 301(C).
    19. 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).
    20. Wang, Chu & Dou, Manfeng & Li, Zhongliang & Outbib, Rachid & Zhao, Dongdong & Zuo, Jian & Wang, Yuanlin & Liang, Bin & Wang, Peng, 2023. "Data-driven prognostics based on time-frequency analysis and symbolic recurrent neural network for fuel cells under dynamic load," Reliability Engineering and System Safety, Elsevier, vol. 233(C).

    More about this item

    Keywords

    ;
    ;
    ;
    ;
    ;

    Statistics

    Access and download statistics

    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.