Federated Kolmogorov-Arnold Networks for Health Data Analysis: A Study Using ECG Signal
With the increasing adoption of predictive healthcare systems, Federated Learning (FL) has emerged as a privacy-preserving paradigm for collaborative model training, which is crucial for sensitive health data. This study investigates the integration of Kolmogorov-Arnold Network (KAN) with FL to enhance predictive healthcare applications. KAN’s innovative implementation of spline-based activation functions offers enhanced flexibility, interoperability, and efficiency in capturing complex nonlinear data patterns with fewer parameters than traditional neural networks. When benchmarked against standard Multilayer Perceptrons (MLPs), KANs show significant performance improvements in both real-world and synthetic electrocardiogram (ECG) datasets. Notably, KAN’s adaptive activation functions, particularly quadratic splines with a grid size of 30, effectively capture the complexities of time-series data, achieving a test accuracy of 93.73% and an F1-score of 92.90% on the MIT-BIH arrhythmia dataset. Furthermore, KAN outperformed MLP on synthetic ECG data, highlighting its potential for broader medical applications. Despite its higher computational overhead, federated KAN shows considerable promise in healthcare settings that require both privacy and high accuracy in modeling complex data patterns. The findings underscore the need for optimization to address computational inefficiencies in KAN-based FL environments while establishing a foundation for future applications of KAN in time-series-based medical diagnostics.
Authors
- Mario A. Bochicchio (ORCID: https://orcid.org/0000-0002-9122-6317)
- Sileshi Nibret Zeleke (ORCID: https://orcid.org/0009-0006-8172-9646)
Institutions
- University of Bari Aldo Moro (IT)
Publication Details
- Published
- 2024-12-15
- DOI
- https://doi.org/10.1109/bigdata62323.2024.10825188
- Citations
- 6
- Primary Topic
- Wireless Body Area Networks
- Type
- article
- Field-Weighted Citation Impact
- 1.11