DOI: 10.3390/app16136401 ISSN: 2076-3417

EFIB-Net: Information Bottleneck-Guided Multi-Resolution Attention Network for Robust ECG Denoising

Minghao Ma, Chen Liu, Yulin Mu, Jingqiu Chen, Li Zhu

Wearable electrocardiogram (ECG) monitoring enables continuous cardiovascular assessment, yet signals acquired in ambulatory environments are inevitably corrupted by baseline wander, electrode motion artifacts, and muscle interference, which obscure diagnostically critical waveform features. Existing deep learning denoisers rely on heuristic attention mechanisms and time-domain-only losses, lacking principled control over what information the network retains or discards. To address this limitation, we propose EFIB-Net, an information bottleneck-guided multi-resolution network for robust ECG denoising. The framework introduces two complementary components: an efficient frequency-guided attention module that derives temporal attention weights directly from the energy distribution of parallel multi-resolution convolutional branches, requiring only four learnable parameters while providing physically interpretable feature selection that naturally highlights QRS complexes, and a variational information bottleneck constraint at the encoder–decoder bottleneck that forces the latent representation to retain only reconstruction-relevant information and discard noise, guided by a spectral–temporal composite loss. To the best of our knowledge, we are among the first to explicitly introduce the information bottleneck principle into deep-learning-based ECG signal denoising. Experiments on the MIT-BIH Arrhythmia Database show that EFIB-Net outperforms ten traditional and deep learning baselines across four standard metrics—signal-to-noise ratio (SNR), root mean square error, percentage root-mean-square difference, and correlation coefficient; at an input SNR of −5 dB it reaches 8.12 dB output SNR, surpassing the strongest attention-based competitor by 1.77 dB (p<0.01) while using only 0.45 M parameters and 10.8 ms inference latency per segment; downstream evaluation further demonstrates that the denoised signals achieve 99.18% R-peak detection sensitivity and 91.26% heartbeat classification F1-score, both within approximately one percentage point of the clean-signal upper bound, making it practical for real-time cardiac monitoring on resource-constrained wearable devices. Zero-shot cross-database evaluation on the QT Database further confirms generalizability, with only 0.54 dB degradation without retraining.

More from our Archive