Learning Across the Divide: Personalised Federated Learning for Robust Clinical Modelling Under Data-View Heterogeneity

Federated Learning (FL) enables collaborative clinical modelling across distributed electronic health records (EHRs) without sharing sensitive patient data. However, variations in medical practice, documentation standards, and data collection across institutions create data-view heterogeneity, where clients possess different or only partially overlapping clinical feature sets. This misalignment hinders the use of standard FL methods. Existing approaches rely on complex preprocessing and manual harmonisation, which can cause information loss, reduce data utility, limit scalability, and restrict client-specific personalisation. To address these limitations, we propose Personalised Attention-based Federated Graph Network (PAFNet), a scalable FL framework that enables meaningful parameter exchange across heterogeneous clients by mapping their distinct data-views into a shared latent space through client-specific projection layers. It then applies a personalised adaptation mechanism using trainable parameter masks, allowing each client to selectively incorporate global model parameters relevant to its own feature set. This design preserves local specificity, improves generalisation, and removes the need for heavy manual preprocessing common in existing approaches. Across CURIAL, eICU, and MIMIC-III datasets, PAFNet consistently outperformed state-of-the-art data-view heterogeneity FL baselines, demonstrating strong generalisation under substantial differences in client feature sets. By enabling effective personalisation and cross-institutional knowledge sharing without extensive harmonisation, PAFNet offers a robust and scalable solution for the federated training of clinical models in data-view heterogeneous environments.

Authors

Institutions

Publication Details

Journal
IEEE Journal of Biomedical and Health Informatics
Published
2025-09-18
DOI
https://doi.org/10.1109/jbhi.2025.3611069
Primary Topic
Machine Learning in Healthcare
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Learning Across the Divide: Personalised Federated Learning for Robust Clinical Modelling Under Data-View Heterogeneity

Danielle Belgrave, Andrew A. S. Soltan, Patrick Schwab, Anshul Thakur et al.
IEEE Journal of Biomedical and Health Informatics
Machine Learning in Healthcare
article

Learning Across the Divide: Personalised Federated Learning for Robust Clinical Modelling Under Data-View Heterogeneity

Danielle Belgrave, Andrew A. S. Soltan, Patrick Schwab, Anshul Thakur, David A. Clifton, Lei Clifton, Soheila Molaei, Kim Branson
article en

Abstract

Federated Learning (FL) enables collaborative clinical modelling across distributed electronic health records (EHRs) without sharing sensitive patient data. However, variations in medical practice, documentation standards, and data collection across institutions create data-view heterogeneity, where clients possess different or only partially overlapping clinical feature sets. This misalignment hinders the use of standard FL methods. Existing approaches rely on complex preprocessing and manual harmonisation, which can cause information loss, reduce data utility, limit scalability, and restrict client-specific personalisation. To address these limitations, we propose Personalised Attention-based Federated Graph Network (PAFNet), a scalable FL framework that enables meaningful parameter exchange across heterogeneous clients by mapping their distinct data-views into a shared latent space through client-specific projection layers. It then applies a personalised adaptation mechanism using trainable parameter masks, allowing each client to selectively incorporate global model parameters relevant to its own feature set. This design preserves local specificity, improves generalisation, and removes the need for heavy manual preprocessing common in existing approaches. Across CURIAL, eICU, and MIMIC-III datasets, PAFNet consistently outperformed state-of-the-art data-view heterogeneity FL baselines, demonstrating strong generalisation under substantial differences in client feature sets. By enabling effective personalisation and cross-institutional knowledge sharing without extensive harmonisation, PAFNet offers a robust and scalable solution for the federated training of clinical models in data-view heterogeneous environments.

IEEE Journal of Biomedical and Health InformaticsVol. 30(3)
GlaxoSmithKline (United Kingdom) (GB), Nuffield Orthopaedic Centre (GB), University of Oxford (GB), Nuffield Health (GB), Primary Health Care (QA)
Openalex Percentile: Top 87%
Machine Learning in Healthcare
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.