SDEMG: Score-based Diffusion Model for Surface Electromyographic Signal Denoising
Yu-Tung Liu, Kuan-Chen Wang, Kai-Chun Liu, Sheng-Yu Peng, Yu Tsao
TL;DR
This work addresses ECG contamination in sEMG recordings, which compromises clinical analysis and prosthetic control. It introduces SDEMG, a conditional score-based diffusion model that denoises sEMG by conditioning the reverse diffusion process on ECG-polluted inputs, with a two-stage training and sampling framework. The model uses a two-stream architecture with Half Normalized Filter blocks and FiLM-conditioned bridges, trained with a cosine schedule and noise-scale conditioning, and evaluated on the NINAPro DB2 dataset with MIT-BIH NSRD ECG interference. Results show that SDEMG yields higher signal quality and more faithful feature preservation than HP, TS, and FCN baselines across a range of SNRs and mismatch conditions, suggesting strong practical potential for clinical sEMG analysis.
Abstract
Surface electromyography (sEMG) recordings can be influenced by electrocardiogram (ECG) signals when the muscle being monitored is close to the heart. Several existing methods use signal-processing-based approaches, such as high-pass filter and template subtraction, while some derive mapping functions to restore clean sEMG signals from noisy sEMG (sEMG with ECG interference). Recently, the score-based diffusion model, a renowned generative model, has been introduced to generate high-quality and accurate samples with noisy input data. In this study, we proposed a novel approach, termed SDEMG, as a score-based diffusion model for sEMG signal denoising. To evaluate the proposed SDEMG approach, we conduct experiments to reduce noise in sEMG signals, employing data from an openly accessible source, the Non-Invasive Adaptive Prosthetics database, along with ECG signals from the MIT-BIH Normal Sinus Rhythm Database. The experiment result indicates that SDEMG outperformed comparative methods and produced high-quality sEMG samples. The source code of SDEMG the framework is available at: https://github.com/tonyliu0910/SDEMG
