Skip to main navigation Skip to search Skip to main content

Training-Aware Wavelet-Domain Controlled-Noise Augmentation for Residual Network-Based Bearing Fault Diagnosis

  • Yifan Li
  • , Jingtao Cheng
  • , Yue Zhao
  • , Ping Song*
  • *Corresponding author for this work
  • Beijing Institute of Technology

Research output: Contribution to journal › Article › peer-review

Abstract

Strong broadband noise can mask the weak impact responses produced by bearing faults, while wavelet reconstructions selected only by signal-domain scores may not provide useful inputs for diagnosis. To address this problem, this paper presents WPMSR-1D-ResNet, a training-aware wavelet-domain controlled-noise augmentation framework. PKPM-guided particle swarm optimization first generates scale-specific perturbation candidates in the DWT detail coefficients. Signal-fidelity constraints remove distorted reconstructions, and a proxy CNN with identity fallback selects a global candidate according to validation Macro-F1. The final 1D-ResNet is trained jointly with the measured waveform and the selected augmented view, whereas inference uses only the raw signal. Under one fixed data construction and a common fixed 20-epoch budget, WPMSR-1D-ResNet achieved (Formula presented.) Macro-F1 at the predefined CWRU low-SNR endpoint and ranked third among eleven methods, placing it within the leading statistical group. Its paired mean exceeded raw-signal 1D-ResNet and per-slice PKPM replacement by 0.05582 and 0.05227, respectively, with gains in nine of ten paired computational seeds. Mixed-SNR training increased mean Macro-F1 across eight mismatched-noise conditions from (Formula presented.) to (Formula presented.). The method ranked second on PU and third on acquisition-held-out AT data; on AT, it reduced the normal-state false-alarm rate from 0.2854 to 0.1646 while maintaining 0.9708 fault sensitivity. Raw-only inference required 0.266 ms per slice. WPMSR-1D-ResNet, therefore, aligns wavelet augmentation with diagnostic performance, preserves useful waveform information, and removes wavelet reconstruction and PSO from the deployment path.

Original languageEnglish
Article number929
JournalMachines
Volume14
Issue number8
DOIs
Publication statusPublished - Aug 2026
Externally publishedYes

Keywords

  • 1D-ResNet
  • bearing fault diagnosis
  • low-SNR vibration signal
  • particle swarm optimization
  • proxy-based selection
  • training augmentation
  • wavelet-domain controlled noise

Fingerprint

Dive into the research topics of 'Training-Aware Wavelet-Domain Controlled-Noise Augmentation for Residual Network-Based Bearing Fault Diagnosis'. Together they form a unique fingerprint.

Cite this