Preserving Patient Privacy while Training a Predictive Model of In-hospital Mortality
arXiv:1912.00354
Abstract
Machine learning models can be used for pattern recognition in medical data in order to improve patient outcomes, such as the prediction of in-hospital mortality. Deep learning models, in particular, require large amounts of data for model training. However, the data is often collected at different hospitals and sharing is restricted due to patient privacy concerns. In this paper, we aimed to demonstrate the potential of distributed training in achieving state-of-the-art performance while maintaining data privacy. Our results show that training the model in the federated learning framework leads to comparable performance to the traditional centralised setting. We also suggest several considerations for the success of such frameworks in future work.
arXiv commentAI for Social Good Workshop, Neurips 2019, Vancouver, Canada
References in corpus (6)
- Very Deep Convolutional Networks for Large-Scale Image Recognition
- Communication-Efficient Learning of Deep Networks from Decentralized Data
- Deep EHR: A Survey of Recent Advances in Deep Learning Techniques for Electronic Health Record (EHR) Analysis
- Federated Optimization: Distributed Machine Learning for On-Device Intelligence
- Towards Federated Learning at Scale: System Design
- Federated Optimization:Distributed Optimization Beyond the Datacenter