Personalized Federated Learning with Residual Fisher Information for Medical Image Segmentation
About
Federated learning enables multiple clients (institutions) to collaboratively train machine learning models without sharing their private data. To address the challenge of data heterogeneity across clients, personalized federated learning (pFL) aims to learn customized models for each client. In this work, we propose pFL-ResFIM, a novel pFL framework that achieves client-adaptive personalization at the parameter level. Specifically, we introduce a new metric, Residual Fisher Information Matrix (ResFIM), to quantify the sensitivity of model parameters to domain discrepancies. To estimate ResFIM for each client model under privacy constraints, we employ a spectral transfer strategy that generates simulated data reflecting the domain styles of different clients. Based on the estimated ResFIM, we partition model parameters into domain-sensitive and domain-invariant components. A personalized model for each client is then constructed by aggregating only the domain-invariant parameters on the server. Extensive experiments on public datasets demonstrate that pFL-ResFIM consistently outperforms state-of-the-art methods, validating its effectiveness.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Retinal Vessel Segmentation | HRF (test) | Dice73.31 | 17 | |
| Gland Segmentation | Gland (LFSC) | Dice (%)76.58 | 5 | |
| Gland Segmentation | Gland MGD-1K | Dice Coefficient82.32 | 5 | |
| Gland Segmentation | Gland (Average) | Dice Score (%)79.45 | 5 | |
| Prostate Segmentation | Prostate (test) | Dice Score (BIDMC Center)78.07 | 5 | |
| Vessel segmentation | CHASEDB (test) | Dice Score72.74 | 5 | |
| Vessel segmentation | DR-Hagis (test) | Dice Score71.63 | 5 | |
| Vessel segmentation | LES-AV (test) | Dice Score74.58 | 5 | |
| Vessel segmentation | ORVS (test) | Dice Score71.5 | 5 | |
| Vessel segmentation | DRIVE (test) | -- | 5 |