Hybrid Quantum-inspired Kolmogorov-Arnold Networks for Privacy-Aware Federated Biosignal Learning
Authors: Chun-Hua Lin, Samuel Yen-Chi Chen, Yu-Chao Hsu, Kuo-Chung Peng, Jiun-Cheng Jiang, Chi-Sheng Chen, Tai-Yue Li, Nan-Yow Chen, +2 more
Organizations: Department of Physics and Center for Theoretical Physics, National Taiwan University, Taipei, Taiwan · National Center for High-Performance Computing, National Institutes of Applied Research, Hsinchu, Taiwan · Brookhaven National Laboratory, Upton, NY, USA · School of Electrical Engineering, Korea Advanced Institute of Science and Technology, Daejeon, Korea · Center for Quantum Science and Engineering, National Taiwan University, Taipei, Taiwan · Beth Israel Deaconess Medical Center, Harvard University, Boston, MA, USA · Department of Electrophysics, National Yang Ming Chiao Tung University, Hsinchu, Taiwan · Physics Division, National Center for Theoretical Sciences, Taipei, Taiwan
Electrocardiogram (ECG) recordings are sensitive biomedical data, limiting the ability of hospitals and wearable devices to share raw signals for centralized model training. Federated learning addresses this practical privacy constraint by enabling collaborative model training while keeping raw biosignal data at their respective sources. However, federated ECG classification remains challenging due to limited client-side samples, imbalanced arrhythmia labels, and non-independent and identically distributed (non-IID) data across clients. These constraints require classifiers that are both communication-efficient and robust to cross-client distribution shifts. In this work, we evaluate a hybrid quantum-inspired Kolmogorov-Arnold network (HQKAN) against a multilayer perceptron (MLP) for five-class arrhythmia classification on the MIT-BIH dataset and three-class classification on the INCART dataset under federated averaging (FedAvg). Across multiple client configurations, HQKAN improves most aggregate and minority-class metrics while using 37.35% fewer trainable parameters and reducing communication cost by 24.89% on MIT-BIH; on INCART, it achieves corresponding reductions of 44.81% and 36.41%. These results indicate that HQKAN offers a compact, communication-efficient and robust alternative to the MLP baseline for privacy-aware federated learning on biosignal data.
Figures & tables
Fig. 1: ROC curves (a) and precision-recall curves (b) for the three minority classes (S, V, F) under IID and non-IID federated partitions on MIT-BIH, comparing the MLP baseline against HQKAN. Curves are averaged over five random seeds.
Fig. 2: ROC curves (a) and precision-recall curves (b) for the majority class N and minority classes S and V under IID and non-IID federated partitions on INCART, comparing the MLP baseline with HQKAN. Curves are averaged over five random seeds.
Setting
Clients
Model
macro-F1 ↑
κ↑
Brier ↓
IID
8
MLP
0.726 ± 0.008
0.764 ± 0.011
0.127 ± 0.007
HQKAN
0.739 ± 0.006
0.776 ± 0.008
0.119 ± 0.005
16
MLP
0.672 ± 0.007
0.686 ± 0.013
0.178 ± 0.009
HQKAN
0.716 ± 0.018
0.744 ± 0.025
0.137 ± 0.017
32
MLP
0.633 ± 0.008
0.626 ± 0.013
0.232 ± 0.011
HQKAN
0.687 ± 0.012
0.701 ± 0.017
0.164 ± 0.011
TABLE I: Performance of MIT-BIH dataset with 5 local epochs (mean ± std over 5 seeds). Non-IID uses α=0.3 .
Setting
Clients
Model
macro-F1 ↑
κ↑
Brier ↓
IID
8
MLP
0.815 ± 0.015
0.882 ± 0.013
0.047 ± 0.007
HQKAN
0.838 ± 0.023
0.898 ± 0.017
0.040 ± 0.009
16
MLP
0.764 ± 0.014
0.824 ± 0.017
0.071 ± 0.007
HQKAN
0.806 ± 0.036
0.864 ± 0.036
0.054 ± 0.011
32
MLP
0.743 ± 0.014
0.793 ± 0.026
0.090 ± 0.011
HQKAN
0.773 ± 0.032
0.827 ± 0.039
0.068 ± 0.014
TABLE II: Performance of INCART dataset with 5 local epochs (mean ± std over 5 seeds). Non-IID uses α=0.3 .
Setting
Class
Model
PPV
Sen
Spe
IID
S
MLP
0.282 ± 0.009
0.739 ± 0.018
0.951 ± 0.003
HQKAN
0.334 ± 0.030
0.747 ± 0.014
0.961 ± 0.006
V
MLP
0.669 ± 0.017
0.881 ± 0.003
0.969 ± 0.002
HQKAN
0.705 ± 0.019
0.908 ± 0.012
0.973 ± 0.003
F
MLP
0.147 ± 0.015
0.868 ± 0.007
0.962 ± 0.004
HQKAN
0.239 ± 0.022
0.837 ± 0.014
0.980 ± 0.003
TABLE III: Per-class performance at 32 clients and 5 local epochs of MIT-BIH dataset (mean ± std over 5 seeds). Non-IID uses α=0.3 .
Setting
Class
Model
PPV
Sen
Spe
IID
N
MLP
0.994 ± 0.000
0.945 ± 0.010
0.961 ± 0.003
HQKAN
0.995 ± 0.001
0.956 ± 0.013
0.966 ± 0.004
S
MLP
0.199 ± 0.025
0.838 ± 0.009
0.961 ± 0.008
HQKAN
0.278 ± 0.076
0.833 ± 0.013
0.974 ± 0.009
V
MLP
0.916 ± 0.014
0.966 ± 0.004
0.989 ± 0.002
HQKAN
0.895 ± 0.038
0.974 ± 0.005
0.985 ± 0.006
TABLE IV: Per-class performance at 32 clients and 5 local epochs of INCART dataset (mean ± std over 5 seeds). Non-IID uses α=0.3 .
Fig. 3: Robustness to client label heterogeneity at 32 clients and 5 local epochs of MIT-BIH dataset. The x-axis uses Hellinger distance to summarize label skew, with larger values indicating stronger non-IID partitions. The y-axis reports AUROC or AUPRC of different models. Each point corresponds to one Dirichlet concentration α .
Fig. 4: Robustness to client label heterogeneity at 32 clients and 5 local epochs of INCART dataset. The x-axis uses Hellinger distance to summarize label skew, with larger values indicating stronger non-IID partitions. The y-axis reports AUROC or AUPRC of different models. Each point corresponds to one Dirichlet concentration α .
Cardiovascular disease remains the leading cause of death worldwide, and early detection of arrhythmias through continuous ECG monitoring on wearable devices can prevent life-threatening events. Federated Learning (FL) enables privacy-preserving collaborative training by keeping raw ECG data on device, yet standard FL incurs prohibitive communication overhead and standard deep learning models cannot fit on ultra-low-power microcontrollers. We propose Family-Grouped Hierarchical Federated Learning (Family-FL), a three-tier architecture that uses the family as a natural privacy boundary for intra-family aggregation before global synchronization. We further design a hardware-constrained Tiny CNN-LSTM architecture with only 669 parameters, INT8-quantized to occupy merely 4.65KB Flash and 2.95KB RAM, meeting the constraints of STC32G12K128-class microcontrollers. Experiments on the MIT-BIH Arrhythmia Database (mean of 5 independent runs with different seeds) demonstrate that Family-FL reduces communication volume by 76.7% compared to FedAvg while maintaining comparable accuracy. Family-FL-Tiny achieves 91.9 +/- 1.2% accuracy with macro-F1 of 0.483 +/- 0.031, reducing total communication to 0.31% of FedAvg. The model achieves reliable ventricular arrhythmia detection (per-class F1 = 0.80), the most clinically critical abnormality for home-based preliminary screening. These results demonstrate the technical feasibility of privacy-preserving federated learning on ultra-resource-constrained microcontrollers through simulation-based evaluation. We honestly discuss limitations: no hardware deployment, single-dataset validation (MIT-BIH, 47 subjects), reduced rare-class sensitivity, and absence of formal differential privacy guarantees.
Hangyu Wu
Shenzhen Coddie Technology co.,ltd, Shenzhen, 518000, China
Electrocardiogram (ECG) monitoring in Internet of Medical Things (IoMT) networks is constrained by strict data-sharing regulations and privacy concerns. Federated learning (FL) enables collaborative learning by keeping raw ECG data on devices, but frequent transmissions of high-dimensional model updates incur heavy per-round traffic over bandwidth-limited links. To alleviate this bottleneck, federated distillation (FD) replaces parameter exchange with logit-based knowledge transfer. However, the performance of FD often degrades under the non-independent and identically distributed (non-IID) and long-tailed label distributions in ECG deployments. To address these challenges, we propose a bidirectional federated knowledge distillation (BiFedKD) framework that employs an aggregation-by-distillation pipeline with temperature scaling to produce a stable global distillation signal for cross-client alignment. Experiments on the MIT-BIH Arrhythmia dataset show that BiFedKD improves accuracy and Macro-F1 over the baseline by 3.52% and 9.93%, respectively. Moreover, to reach the same Macro-F1, BiFedKD reduces communication overhead by 40% and computation cost by 71.7% compared with the baseline.
Zixuan Shu, Tiancheng Cao, Hen-Wei Huang
School of Electrical and Electronic Engineering, Nanyang Technological University, Republic of Singapore · Lee Kong Chian School of Medicine, Nanyang Technological University, Republic of Singapore
Continuous electrocardiography (ECG) monitoring could surface rhythm abnormalities before they escalate into cardiovascular events. However, a deployable system must satisfy three requirements simultaneously: legal-grade privacy (GDPR, HIPAA), real-time inference on constrained edge hardware, and detection quality under non-IID cross-hospital data. We design and evaluate an end-to-end federated system addressing all three for unsupervised 12-lead ECG anomaly detection on PTB-XL dataset, combining three autoencoder families (VanillaAE, ConvAE, VAE), Flower-based federated averaging (FedAvg) across ten simulated hospitals, client-side differentially private SGD (DP-SGD) with a Rényi-DP accountant, and 8-bit integer (INT8) post-training quantization with Raspberry Pi 4 benchmarking. Our main contributions are: an empirical characterization of how these mechanisms compose, practical DP-specific recommendations, and technical and security insights for a clinically sensitive setting. Federated learning matches or exceeds the centralized baseline across all architectures (ConvAE federated area under the ROC curve, AUROC, 0.782), and an ε sweep identifies ε=4 as the recommended clinical operating point. INT8 quantization roughly halves model size and cuts Pi 4 latency by up to 44 with <0.12 AUROC loss. Crucially, DP and quantization penalties are empirically independent, so practitioners need not trade a strong privacy guarantee for a compact edge footprint. To our knowledge, this is the first system combining federated learning, formal (ε,δ)-DP, unsupervised reconstruction-based detection, and quantized AArch64 deployment.
Kaan Arda Akyol, Jakub Kacper Szeląg, Aydin Abadi +9
School of Computing, Newcastle University, Newcastle upon Tyne, United Kingdom