Publication

Hyperautonomy Artificial Intelligence Lab

2026 ARDiff: An Adaptive Reverse-Step Diffusion Framework for Unsupervised Vibration Signal Denoising with Frequency Attention

본문

Journal
Mechanical Systems and Signal Processing
Author
Juhyun Kim, Yong Chae Kim, Jongwha Baek, Jinwook Lee, Joon Ha Jung*, and Byeng D. Youn*
Date
2026-01
Citation Index
SCIE (IF: 8.9, Rank: 3.0%)
Vol./ Page
Vol. 244, pp. 113726
Year
2026

35ad6ca63a7a63953b4d94db5ad8778c_1765756108_2056.png
Abstract
 

This paper proposes ARDiff, an adaptive reverse-step diffusion framework for unsupervised denoising of vibration signals under high-noise industrial conditions. Unlike conventional diffusion models that rely on fixed denoising trajectories or external condition inputs, ARDiff enables adaptive noise-aware reconstruction without any class labels or guidance, through the integration of three key components. First, the frequency attention module (FAM) introduces a complex-valued attention mechanism in the frequency domain, enabling the model to effectively capture fault-relevant spectral features critical for signal reconstruction. Second, the noise step prediction model (NSPM) estimates the optimal reverse diffusion timestep based on the input noise level, improving reconstruction accuracy. Third, the variance-based scaling method (VSM) aligns the scale of the input signal with the expected distribution of the diffusion model at the predicted timestep, ensuring stable and consistent restoration. Experimental validation using the open-source CWRU dataset, the experimental HAI-PUMP dataset, and the SNU Planetary Gearbox dataset demonstrated that ARDiff consistently outperformed existing denoising models, particularly under extremely low signal-to-noise ratio (SNR) conditions. Ablation studies confirmed the individual contributions of FAM, NSPM, and VSM, verifying that ARDiff is a robust and generalizable denoising framework capable of operating effectively without any external conditions or label supervision. Moreover, ARDiff consistently maintained stable denoising performance under both Gaussian and non-Gaussian noise conditions, highlighting its practical robustness in realistic industrial environments.