DR-net-Mamba: Selective State-Space Modeling for Long-Range ECG Time-Series Denoising
Organizations: Department of Computer Science at ETH Zurich, Universitätstrasse 6, 8092 Zürich, Switzerland
Abstract
Electrocardiogram (ECG) recordings are corrupted by non-stationary noise sources that degrade diagnostic reliability, particularly in ambulatory and long-duration recordings. Deep learning denoisers exist, but convolutional architectures are limited by their receptive field, transformer-based models scale quadratically with sequence length, and diffusion-based approaches incur prohibitive inference cost. We propose a Mamba-augmented model that inserts selective state-space blocks at the convolutional bottleneck, combining local feature extraction with long-range temporal modeling at linear complexity. We comprehensively evaluate the proposed model with respect to reconstruction fidelity, noise robustness, recording-length scaling, and downstream diagnostic classification across over 40 pathology classes. On synthetic and real datasets, our model achieves the highest SNR and lowest RMSE, with the Mamba advantage increasing with sequence length and in low-SNR regimes. On classification with two independent classifiers, the proposed Mamba-based models achieve the best macro AUROC among all denoisers and improve over their convolutional base models. Calibration is more nuanced and classifier-dependent: denoising improves Binary Cross-Entropy and Brier score on Inception1D but often fails to beat the noisy input on ResNet1D-Wang, and the lead-specific Mamba variant is the only denoiser to improve both calibration metrics over the noisy baseline on both classifiers. Per-class analysis reveals a morphology-dependent benefit: Mamba substantially improves ST/T-change diagnoses, which depend on broad, context-sensitive waveforms.
Figures & tables
| Classifier trained on | Test input | AUC |
| Clean | Clean | |
| Clean | Noisy | |
| Noisy | Noisy | |
| Clean | Denoised (Mamba1-3B Lead Aware) |
| CD | HYP | MI | NORM | STTC | Overall | |||
| AUROC | Sensitivity | Specificity | F1 | |||||
| UNet-Mamba1-3B | .925 .01 | .883 .02 | .909 .02 | .926 .01 | .880 .02 | .789 .02 | .840 .01 | .703 .02 |
| DR-net-Mamba1-3B | .921 .01 | .887 .02 | .909 .02 | .926 .01 | .881 .02 | .794 .02 | .841 .01 | .706 .02 |
| UNet-Mamba1-3B (LS) | .920 .01 | .887 .02 | .910 .01 | .928 .01 | .887 .02 | .808 .02 | .839 .01 | .715 .02 |
| DRNN | .900 .02 | .885 .02 | .895 .02 | .910 .02 | .824 .02 | .783 .02 | .780 .01 | .651 .02 |
| DAE | .860 .02 | .854 .03 | .874 .01 | .907 .01 | .848 .02 | .709 .02 | .819 .01 | .638 .02 |
Appendix figures & tables29 assets
Supplementary material from the paper’s appendix.
Appendix
| Layer Type | |||||
| 1 | Conv | 1 | 25 | 1 | 24 |
| 2 | Conv | 1 | 25 | 1 | 24 |
| 3 | Conv | 1 | 25 | 1 | 24 |
| 4 | AvgPool | 5 | 5 | 1 | 4 |
| 5 | Conv | 1 | 15 | 5 | 70 |
| 6 | Conv | 1 | 15 | 5 | 70 |
| UNet / IMUNet / DAE / | MECGE | UNet-Mamba | |
| DRNN / DR-net-S2 | (ours) | ||
| Optimizer | Adam | AdamW | Adam |
| Learning rate | |||
| LR scheduler | ReduceLROnPlateau | Exp. decay | Cosine + warmup |
| Warmup epochs | — | — | 10 |
| (cosine) | — | — | 400 |
| P | Q | R | S | T | |
| 1.2 | 30.0 | 0.75 | |||
| 0.6 | 0.2 | 0.0 | 1.0 | 0.35 | |
| 0.25 | 0.1 | 0.1 | 0.1 | 0.4 | |
| 0.1 | 0.1 | 0.0 | 0.0 | 0.0 |
| Config | BW | MA | EM | AWGN | Combined SNR |
| Light | 5.0 | 10.0 | 15.0 | 25.0 | 3.46 |
| Medium | 2.5 | 7.5 | 12.5 | 22.5 | 0.96 |
| Strong | 0.0 | 5.0 | 10.0 | 20.0 |
| Method | Split Length | Epochs | Runtime |
| UNet-Mamba1-3B | 14400 | 180 | 41m 45s |
| UNet-Mamba1-3B | 7200 | 256 | 1h 4m 52s |
| UNet-Mamba1-3B | 3600 | 250 | 1h 50m 39s |
| UNet-Mamba1-3B | 1800 | 286 | 7h 18m 38s |
| UNet | 14400 | 102 | 12m 33s |
| UNet | 7200 | 105 | 21m 22s |
| Method | Split Length | Epochs | Runtime |
| UNet-Mamba1-3B | 14400 | 263 | 32m 13s |
| UNet-Mamba1-3B | 7200 | 285 | 39m 44s |
| UNet-Mamba1-3B | 3600 | 274 | 1h 7m 7s |
| UNet-Mamba1-3B | 1800 | 113 | 1h 34m 27s |
| UNet | 14400 | 257 | 17m 13s |
| UNet | 7200 | 206 | 21m 56s |
| Baseline | UNet Mamba1-3B | DR-net Mamba1-3B |
| DRNN | ||
| DAE | ||
| UNet | ||
| DR-net-UNet | ||
| IMUNet | ||
| DR-net-IMUNet |
| Baseline | UNet Mamba1-3B | DR-net Mamba1-3B |
| DRNN | ||
| DAE | ||
| UNet | ||
| DR-net-UNet | ||
| IMUNet | ||
| DR-net-IMUNet |
| Baseline | UNet Mamba1-3B | DR-net Mamba1-3B |
| DRNN | ||
| DAE | ||
| UNet | ||
| DR-net-UNet | ||
| IMUNet | ||
| DR-net-IMUNet |
| Baseline | UNet | DR-net- UNet | UNet Mamba1-3B (No Comp) | DR-net Mamba1-3B (No Comp) | UNet Mamba1-3B | DR-net Mamba1-3B |
| UNet | — | |||||
| DR-net- UNet | — | |||||
| UNet Mamba1-3B (No Comp) | — | |||||
| DR-net Mamba1-3B (No Comp) | — | |||||
| UNet Mamba1-3B | — |
| BW | MA | EM | AWGN |
| 2.5 | 7.5 | 12.5 | 22.5 |
| CD | HYP | MI | NORM | STTC | |
| UNet Mamba1-3B | .926 .02 | .888 .02 | .916 .01 | .931 .01 | .888 .02 |
| DRNET Mamba1-3B | .927 .02 | .895 .02 | .915 .02 | .933 .01 | .889 .02 |
| UNet Mamba1-3B (LS) | .924 .01 | .894 .02 | .918 .02 | .935 .01 | .899 .02 |
| DRNN | .911 .01 | .887 .02 | .903 .01 | .919 .01 | .857 .02 |
| DAE | .863 .02 | .853 .02 | .886 .02 | .909 .01 | .859 .02 |
| UNet | .919 .01 | .877 .02 | .913 .01 | .926 .01 | .881 .02 |
| Sensitivity | Specificity | F1 | |
| UNet Mamba1-3B (ours) | .785 .02 | .862 .01 | .723 .01 |
| DRNET Mamba1-3B (ours) | .788 .02 | .865 .01 | .727 .01 |
| UNet Mamba1-3B (Lead aware) (ours) | .791 .02 | .861 .01 | .725 .01 |
| DAE | .698 .02 | .846 .01 | .653 .02 |
| DRNN | .784 .02 | .819 .01 | .682 .02 |
| MECGE | .760 .02 | .846 .01 | .693 .02 |
| Sensitivity | Specificity | F1 | |
| UNet Mamba1-3B (ours) | .797 .02 | .853 .01 | .722 .01 |
| DRNET Mamba1-3B (ours) | .800 .02 | .858 .01 | .727 .02 |
| UNet Mamba1-3B (Lead Specific) (ours) | .798 .02 | .858 .01 | .728 .02 |
| DAE | .733 .02 | .831 .01 | .668 .02 |
| DRNN | .787 .02 | .808 .01 | .679 .02 |
| MECGE | .786 .02 | .826 .01 | .691 .02 |
| CD | HYP | MI | NORM | STTC | |
| Sensitivity | |||||
| UNet-Mamba1-3B | .822 .03 | .526 .05 | .865 .03 | .918 .02 | .792 .03 |
| DR-net-Mamba1-3B | .834 .04 | .534 .05 | .854 .04 | .917 .02 | .801 .04 |
| UNet-Mamba1-3B (LS) | .816 .03 | .556 .05 | .815 .03 | .943 .01 | .824 .04 |
| DAE | .640 .05 | .299 .07 | .889 .03 | .891 .02 | .769 .04 |
| DRNN | .822 .03 | .489 .06 | .887 .03 | .863 .03 | .860 .03 |
| CD | HYP | MI | NORM | STTC | |
| Sensitivity | |||||
| UNet-Mamba1-3B | .784 .03 | .515 .06 | .898 .02 | .942 .02 | .786 .03 |
| DR-net-Mamba1-3B | .784 .03 | .522 .07 | .870 .03 | .931 .02 | .794 .04 |
| UNet-Mamba1-3B (LS) | .780 .03 | .541 .06 | .841 .03 | .961 .01 | .803 .04 |
| DAE | .667 .04 | .347 .06 | .894 .03 | .915 .02 | .756 .04 |
| DRNN | .762 .03 | .507 .06 | .902 .02 | .898 .03 | .850 .03 |
| Abbreviation | Description |
| NDT | non-diagnostic T abnormalities |
| NST_ | non-specific ST changes |
| DIG | digitalis-effect |
| LNGQT | long QT-interval |
| NORM | normal ECG |
| IMI | inferior myocardial infarction |
| STTC | CD | |
| Temporal scale | Broad (200–400 ms) | Narrow (R-peak time 45–60 ms) |
| Bottleneck steps spanned | –7 | |
| Frequency content | Low | High |
| Context needed | Global (baseline-relative) | Local (QRS-internal) |
| Mamba effect on denoising | Better noise/signal separation | Over-smoothing of sharp features |