TinyCardioUNet: IMU-to-ECG Translation with Graph-Encoded Inter-Axis Dependencies and Tensor Decomposition-Based Parameter Reduction
In the authors' words
Estimating electrocardiography (ECG) from a chest-worn inertial measurement unit (IMU) enables continuous heart rate (HR) monitoring without the discomfort of electrodes. We propose TinyCardioUNet, a lightweight UNet that uses all six IMU axes without prior channel selection, refines its bottleneck with a graph neural network that encodes inter-axis dependencies, and employs tensor decomposition with automatic variational Bayesian rank selection for parameter reduction. On a public dataset, TinyCardioUNet achieves an RMSE of and a Pearson correlation coefficient of with only k parameters and remains comparatively robust to additive noise, demonstrating accurate ECG reconstruction with a compact model.
Appeared: Friday, September 25. arXiv. Preprint, not yet peer-reviewed.
Authors' comment: The source code and pretrained models are available at https://github.com/ttlabtuat/TinyCardioUNet