自动调参的稀疏非负矩阵分解,提升噪声中周期信号提取精度
Sparse Hyperparametric Itakura-Saito Nonnegative Matrix Factorization via Bi-Level Optimization
- 通过双层优化自动调节每行的正则化参数
- 在合成与真实数据上均实现更优的频谱分离效果
- 适合高噪声环境下轴承故障的非侵入式检测
非负矩阵分解(NMF)中惩罚超参数的选择至关重要,它们决定了重建精度与约束满足之间的权衡。本文聚焦于基于Itakura-Saito(IS)散度的NMF问题,该方法对从混合信号的频谱图中提取低谱密度成分尤为有效,并可通过引入稀疏性约束进一步增强性能。提出的新算法SHINBO采用双层优化框架,自动适应性地调整行相关的惩罚超参数,显著提升了IS-NMF在噪声环境中分离稀疏周期信号的能力。实验结果表明,SHINBO在合成数据和真实场景中均实现了精确的频谱分解。在真实应用中,该方法特别适用于滚动轴承的非侵入式振动故障检测,其中目标信号成分常位于高频子带,但被更强、谱宽更大的噪声所掩盖。通过解决超参数选择这一关键难题,SHINBO在复杂噪声环境下的信号恢复性能达到当前最优水平。
原文摘要 · Abstract (English)
The selection of penalty hyperparameters is a critical aspect in Nonnegative Matrix Factorization (NMF), since these values control the trade-off between reconstruction accuracy and adherence to desired constraints. In this work, we focus on an NMF problem involving the Itakura-Saito (IS) divergence, which is particularly effective for extracting low spectral density components from spectrograms of mixed signals, and benefits from the introduction of sparsity constraints. We propose a new algorithm called SHINBO, which introduces a bi-level optimization framework to automatically and adaptively tune the row-dependent penalty hyperparameters, enhancing the ability of IS-NMF to isolate sparse, periodic signals in noisy environments. Experimental results demonstrate that SHINBO achieves accurate spectral decompositions and demonstrates superior performance in both synthetic and real-world applications. In the latter case, SHINBO is particularly useful for noninvasive vibration-based fault detection in rolling bearings, where the desired signal components often reside in high-frequency subbands but are obscured by stronger, spectrally broader noise. By addressing the critical issue of hyperparameter selection, SHINBO improves the state-of-the-art in signal recovery for complex, noise-dominated environments.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。