Ambulatory ECG noise reduction algorithm for conditional diffusion model based on multi-kernel convolutional transformer

Abstract

Ambulatory electrocardiogram (ECG) testing plays a crucial role in the early detection, diagnosis, treatment evaluation, and prevention of cardiovascular diseases. Clear ECG signals are essential for the subsequent analysis of these conditions. However, ECG signals obtained during exercise are susceptible to various noise interferences, including electrode motion artifact, baseline wander, and muscle artifact. These interferences can blur the characteristic ECG waveforms, potentially leading to misjudgment by physicians. To suppress noise in ECG signals more effectively, this paper proposes a novel deep learning-based noise reduction method. This method enhances the diffusion model network by introducing conditional noise, designing a multi-kernel convolutional transformer network structure based on noise prediction, and integrating the diffusion model inverse process to achieve noise reduction. Experiments were conducted on the QT database and MIT-BIH Noise Stress Test Database and compared with the algorithms in other papers to verify the effectiveness of the present method. The results indicate that the proposed method achieves optimal noise reduction performance across both statistical and distance-based evaluation metrics as well as waveform visualization, surpassing eight other state-of-the-art methods. The network proposed in this paper demonstrates stable performance in addressing electrode motion artifact, baseline wander, muscle artifact, and the mixed complex noise of these three types, and it is anticipated to be applied in future noise reduction analysis of clinical dynamic ECG signals.