A Unified Benchmark of Federated Learning with Kolmogorov-Arnold Networks for Medical Imaging

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Lee, Youngjoon, Gong, Jinu, Kang, Joonhyuk
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866908544091029504
author Lee, Youngjoon
Gong, Jinu
Kang, Joonhyuk
author_facet Lee, Youngjoon
Gong, Jinu
Kang, Joonhyuk
contents Federated Learning (FL) enables model training across decentralized devices without sharing raw data, thereby preserving privacy in sensitive domains like healthcare. In this paper, we evaluate Kolmogorov-Arnold Networks (KAN) architectures against traditional MLP across six state-of-the-art FL algorithms on a blood cell classification dataset. Notably, our experiments demonstrate that KAN can effectively replace MLP in federated environments, achieving superior performance with simpler architectures. Furthermore, we analyze the impact of key hyperparameters-grid size and network architecture-on KAN performance under varying degrees of Non-IID data distribution. In addition, our ablation studies reveal that optimizing KAN width while maintaining minimal depth yields the best performance in federated settings. As a result, these findings establish KAN as a promising alternative for privacy-preserving medical imaging applications in distributed healthcare. To the best of our knowledge, this is the first comprehensive benchmark of KAN in FL settings for medical imaging task.
format Preprint
id arxiv_https___arxiv_org_abs_2504_19639
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle A Unified Benchmark of Federated Learning with Kolmogorov-Arnold Networks for Medical Imaging
Lee, Youngjoon
Gong, Jinu
Kang, Joonhyuk
Machine Learning
Signal Processing
Federated Learning (FL) enables model training across decentralized devices without sharing raw data, thereby preserving privacy in sensitive domains like healthcare. In this paper, we evaluate Kolmogorov-Arnold Networks (KAN) architectures against traditional MLP across six state-of-the-art FL algorithms on a blood cell classification dataset. Notably, our experiments demonstrate that KAN can effectively replace MLP in federated environments, achieving superior performance with simpler architectures. Furthermore, we analyze the impact of key hyperparameters-grid size and network architecture-on KAN performance under varying degrees of Non-IID data distribution. In addition, our ablation studies reveal that optimizing KAN width while maintaining minimal depth yields the best performance in federated settings. As a result, these findings establish KAN as a promising alternative for privacy-preserving medical imaging applications in distributed healthcare. To the best of our knowledge, this is the first comprehensive benchmark of KAN in FL settings for medical imaging task.
title A Unified Benchmark of Federated Learning with Kolmogorov-Arnold Networks for Medical Imaging
topic Machine Learning
Signal Processing
url https://arxiv.org/abs/2504.19639