# Federated Uncertainty-Aware Learning for Distributed Hospital EHR Data

arXiv: https://arxiv.org/abs/1910.12191 · HTML: https://mlsanity.com/paper/1910.12191/

## Key facts

- Cited by 17 papers in the map (data through 14 Oct 2026).
- First posted on arXiv 27 Oct 2019.

## Built on by (most cited first, 17 of 17)

- [Federated Learning for Smart Healthcare: A Survey](https://mlsanity.com/paper/2111.08834/) (2021)
- [Federated Learning for Medical Applications: A Taxonomy, Current Trends, Challenges, and Future Research Directions](https://mlsanity.com/paper/2208.03392/) (2022)
- [Federated Multi-View Learning for Private Medical Data Integration and Analysis](https://mlsanity.com/paper/2105.01603/) (2021)
- [Failure Prediction in Production Line Based on Federated Learning: An Empirical Study](https://mlsanity.com/paper/2101.11715/) (2021)
- [Federated learning: Applications, challenges and future directions](https://mlsanity.com/paper/2205.09513/) (2022)
- [Federated Learning of Medical Concepts Embedding using BEHRT](https://mlsanity.com/paper/2305.13052/) (2023)
- [Towards the Practical Utility of Federated Learning in the Medical Domain](https://mlsanity.com/paper/2207.03075/) (2022)
- [Explainable, Domain-Adaptive, and Federated Artificial Intelligence in Medicine](https://mlsanity.com/paper/2211.09317/) (2022)
- [Unimodal Training-Multimodal Prediction: Cross-modal Federated Learning with Hierarchical Aggregation](https://mlsanity.com/paper/2303.15486/) (2023)
- [Federated Learning for Cyber Physical Systems: A Comprehensive Survey](https://mlsanity.com/paper/2505.04873/) (2025)
- [FUNAvg: Federated Uncertainty Weighted Averaging for Datasets with Diverse Labels](https://mlsanity.com/paper/2407.07488/) (2024)
- [PMFL: Partial Meta-Federated Learning for heterogeneous tasks and its applications on real-world medical records](https://mlsanity.com/paper/2112.05321/) (2021)
- [A review of federated learning in renewable energy applications: Potential, challenges, and future directions](https://mlsanity.com/paper/2312.11220/) (2023)
- [Federated attention consistent learning models for prostate cancer diagnosis and Gleason grading](https://mlsanity.com/paper/2302.06089/) (2023)
- [Deep Learning Frameworks for Cognitive Radio Networks: Review and Open Research Challenges](https://mlsanity.com/paper/2410.23949/) (2024)
- [Feature-context driven Federated Meta-Learning for Rare Disease Prediction](https://mlsanity.com/paper/2112.14364/) (2021)
- [FedCVD++: Communication-Efficient Federated Learning for Cardiovascular Risk Prediction with Parametric and Non-Parametric Model Optimization](https://mlsanity.com/paper/2507.22963/) (2025)

## Abstract (arXiv)

Recent works have shown that applying Machine Learning to Electronic Health Records (EHR) can strongly accelerate precision medicine. This requires developing models based on diverse EHR sources. Federated Learning (FL) has enabled predictive modeling using distributed training which lifted the need of sharing data and compromising privacy. Since models are distributed in FL, it is attractive to devise ensembles of Deep Neural Networks that also assess model uncertainty. We propose a new FL model called Federated Uncertainty-Aware Learning Algorithm (FUALA) that improves on Federated Averaging (FedAvg) in the context of EHR. FUALA embeds uncertainty information in two ways: It reduces the contribution of models with high uncertainty in the aggregated model. It also introduces model ensembling at prediction time by keeping the last layers of each hospital from the final round. In FUALA, the Federator (central node) sends at each round the average model to all hospitals as well as a randomly assigned hospital model update to estimate its generalization on that hospital own data. Each hospital sends back its model update as well a generalization estimation of the assigned model. At prediction time, the model outputs C predictions for each sample where C is the number of hospital models. The experimental analysis conducted on a cohort of 87K deliveries for the task of preterm-birth prediction showed that the proposed approach outperforms FedAvg when evaluated on out-of-distribution data. We illustrated how uncertainty could be measured using the proposed approach.