Similarity-aware VAE with wavelet-convolutional 1D-CNN for rolling bearing fault diagnosis

Wei Xiong, Na Xiao and Ruili Wang

PLOS ONE, 2026, vol. 21, issue 1, 1-15

Abstract: To address the uneven distribution of fault categories in data sets for deep learning-based fault diagnosis, we propose a fault diagnosis framework combining an improved Variational Autoencoder (Similarity-Aware VAE) with a Wavelet-Convolutional 1D-CNN. The Similarity-Aware VAE employs a novel similarity loss function for data augmentation, measuring feature distances in high-dimensional space while automatically adjusting training parameters and weights through an enhanced attention mechanism to balance the dataset.The Wavelet-Convolutional 1D-CNN replaces the first convolutional layer of CNN with a Wavelet-Convolutional layer based on continuous wavelet transform, enabling multi-scale feature extraction for fault data analysis. Experimental validation using public datasets demonstrates that this method effectively enhances data quality while maintaining robust diagnostic performance, offering practical implications for industrial fault diagnosis.

Date: 2026
References: Add references at CitEc
Citations:

Downloads: (external link)
https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0338388 (text/html)
https://journals.plos.org/plosone/article/file?id= ... 38388&type=printable (application/pdf)

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:plo:pone00:0338388

DOI: 10.1371/journal.pone.0338388

Access Statistics for this article

More articles in PLOS ONE from Public Library of Science
Bibliographic data for series maintained by plosone ().