Abstract:Federated learning (FL) holds great potential for medical applications. However, statistical heterogeneity across healthcare institutions poses a major challenge for FL, as the global model struggles both to generalize across unseen patient populations and to adapt to the unique data distributions of individual hospitals. This heterogeneity also exacerbates forgetting at both the global and local level, resulting in previous learned patient patterns to be misclassified after model updates. While prior work has largely treated generalization and personalization as separate challenges, we show that a better balance between the two can be achieved through selective alignment with the global model and a modified aggregation scheme, which together mitigate the effects of statistical heterogeneity. Specifically, we introduce FedKPer, which introduces knowledge personalization into the training stage of each local device. Afterwards, generalization is considered via the global model aggregation process, where local updates that are reliable and label-diverse are emphasized. We evaluate the performance of FedKPer, devising additional metrics that relate to common consequences of forgetting. Overall, we demonstrate FedKPer improves the generalization-personalization trade-off without sacrificing retention.
| Comments: | Accepted to IEEE International Conference on Image Processing (ICIP) |
| Subjects: | Image and Video Processing (eess.IV); Machine Learning (cs.LG) |
| Cite as: | arXiv:2605.00698 [eess.IV] |
| (or arXiv:2605.00698v1 [eess.IV] for this version) | |
| https://doi.org/10.48550/arXiv.2605.00698 arXiv-issued DOI via DataCite (pending registration) |
From: Zoe Fowler [view email]
[v1]
Fri, 1 May 2026 14:36:28 UTC (372 KB)
此内容由惯性聚合(RSS阅读器)自动聚合整理，仅供阅读参考。 原文来自 — 版权归原作者所有。