TY - JOUR
T1 - Training-Aware Wavelet-Domain Controlled-Noise Augmentation for Residual Network-Based Bearing Fault Diagnosis
AU - Li, Yifan
AU - Cheng, Jingtao
AU - Zhao, Yue
AU - Song, Ping
N1 - Publisher Copyright:
© 2026 by the authors.
PY - 2026/8
Y1 - 2026/8
N2 - 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.
AB - 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.
KW - 1D-ResNet
KW - bearing fault diagnosis
KW - low-SNR vibration signal
KW - particle swarm optimization
KW - proxy-based selection
KW - training augmentation
KW - wavelet-domain controlled noise
UR - https://www.scopus.com/pages/publications/105048293487
U2 - 10.3390/machines14080929
DO - 10.3390/machines14080929
M3 - Article
AN - SCOPUS:105048293487
SN - 2075-1702
VL - 14
JO - Machines
JF - Machines
IS - 8
M1 - 929
ER -