Interaction-Gated Time-Frequency Learning for Parameter-Efficient PPG-Based Heart Rate Estimation
Organizations: International University, Vietnam National University HCMC, Ho Chi Minh City, Vietnam · University of New Brunswick, Fredericton, Canada · Ho Chi Minh City University of Technology, Ho Chi Minh City, Vietnam · Austrian Institute of Technology, Vienna, Austria
Abstract
Photoplethysmography (PPG) enables low-cost and non-invasive heart rate (HR) monitoring but remains sensitive to waveform distortion, physiological variability, and measurement interference. Temporal waveform characteristics and frequency-domain periodicity provide complementary information for HR estimation; however, effectively exploring and integrating these representations within a compact model remains challenging. This paper proposes TF-IG, a parameter-efficient interaction-gated time-frequency learning framework for PPG-based HR estimation. TF-IG employs a dual-branch architecture in which a temporal branch captures pulse morphology via dilated depthwise-separable convolutions, while a frequency branch extracts cardiac periodicity information from bounded log-spectrogram representations. To combine the two domains, we introduce an interaction-gated fusion module that explicitly models feature agreement and discrepancy, while also learning feature-wise weights to balance temporal and spectral information. A parameter-shared regression head further maps the temporal, spectral, and fused representations into the final HR prediction space. Experiments on the benchmark BIDMC PPG and Respiration Dataset using a subject-exclusive training, validation, and test protocol show that TF-IG achieves a mean absolute error (MAE) of 1.227 BPM, a root mean square error (RMSE) of 2.060 BPM, a Pearson correlation coefficient of 0.9934, and 92.28 percent of estimates within 3 BPM, while remaining parameter-efficient with only 84.6K parameters. Ablation results further demonstrate the complementary contributions of the temporal and frequency branches, interaction-gated fusion, and shared regression head. These results indicate that explicitly modeling cross-domain interactions provides an effective accuracy-complexity trade-off for lightweight PPG-based HR estimation.
Figures & tables
| Model | Params | FLOPs/sample | MAE (BPM) | RMSE (BPM) | MSE | Pearson | (%) | (%) |
|---|---|---|---|---|---|---|---|---|
| TF-IG (Ours) | 84,591 | 13.46 M | 1.2270 | 2.0601 | 4.2440 | 0.9934 | 95.98 | 92.28 |
| Welch PSD [ 11 ] | 0 | N/A | 1.0823 | 3.2015 | 10.2494 | 0.9810 | 93.24 | 90.79 |
| FCN_Residual [ 16 ] | 53,713 | 99.52 M | 1.5584 | 2.1180 | 4.4860 | 0.9918 | 97.79 | 87.45 |
| FCN [ 16 ] | 8,945 | 8.47 M | 2.9695 | 3.7681 | 14.1986 | 0.9824 | 85.72 | 61.10 |
| FCN_DCT [ 16 ] | 3,266 | 0.573 M | 5.8389 | 8.0721 | 65.1582 | 0.9122 | 57.68 | 36.85 |
| Time Branch | Freq Branch | Int-Gated Fusion | Shared Head | MAE | RMSE | MSE | Pearson | (%) | (%) |
|---|---|---|---|---|---|---|---|---|---|
| ✓ | ✓ | 2.1677 | 3.3984 | 11.5492 | 0.9862 | 78.52 | 87.77 | ||
| ✓ | ✓ | 2.347 | 4.5638 | 20.8283 | 0.9713 | 83.31 | 89.02 | ||
| ✓ | ✓ | ✓ | 1.5878 | 2.464 | 6.0713 | 0.9925 | 87.17 | 92.24 | |
| ✓ | ✓ | ✓ | 1.502 | 2.4475 | 5.9904 | 0.9926 | 84.75 | 93.85 | |
| ✓ | ✓ | ✓ | ✓ | 1.227 | 2.0601 | 4.244 | 0.9934 | 92.28 | 95.98 |
| Raw Window | MAE (BPM) | RMSE (BPM) | MSE | Pearson | (%) | (%) |
|---|---|---|---|---|---|---|
| seconds | 1.9331 | 3.2854 | 10.7938 | 0.9856 | 82.56 | 89.58 |
| seconds | 1.2270 | 2.0601 | 4.2440 | 0.9934 | 92.28 | 95.98 |
Heart Rate Estimation
Heart rate estimation recovers beats per minute from wearable, contact, or remote physiological signals.
See all 31 papers