Federated learning model for predicting major postoperative complications

Overview

• This study explores the use of federated learning, a machine learning technique, to predict postoperative complications in surgical patients using electronic health records (EHR) data from multiple hospitals.
• Federated learning allows for the training of AI models without directly sharing sensitive patient data across institutions, addressing challenges around medical data privacy.
• The researchers compared the performance of federated learning models with local models trained on data from a single hospital and central models trained on pooled data from multiple hospitals.

Plain English Explanation

Predicting the risk of complications after surgery is an important task for doctors. Federated learning is a technique that allows different hospitals to train an AI model together without having to share their patients' private medical information. In this study, the researchers used federated learning to develop models that can predict nine major complications, such as prolonged stay in the intensive care unit (ICU) or the need for mechanical ventilation, based on data from patients' medical records.

The researchers compared the performance of these federated learning models to models trained using data from a single hospital (local models) and models trained on data pooled from multiple hospitals (central models). They found that the federated learning models achieved similar or even slightly better performance than the central models in some cases, demonstrating that federated learning can be a useful tool for training robust and generalizable models while protecting patient privacy.

Technical Explanation

This retrospective cohort study used EHR data from two University of Florida Health (UFH) hospitals in Gainesville (GNV) and Jacksonville (JAX) to develop federated learning models to predict nine major postoperative complications. The researchers trained these federated learning models using perioperative and intraoperative features from a total of 108,486 adult patients who underwent inpatient surgical procedures.

The performance of the federated learning models was compared to local learning models trained on a single site and central learning models trained on the pooled dataset from both centers. The federated learning models achieved AUROC (area under the receiver operating characteristics curve) values ranging from 0.81 for wound complications to 0.92 for prolonged ICU stay at the GNV center, and 0.73-0.74 for wound complications to 0.92-0.93 for hospital mortality at the JAX center.

The federated learning models performed comparably to the central learning models, except for prolonged ICU stay, where the federated learning models slightly outperformed the central models at the GNV center but slightly underperformed at the JAX center. Additionally, the federated learning models achieved similar performance to the best local learning models at each center, demonstrating strong generalizability.

Critical Analysis

The researchers acknowledge that their study has several limitations. First, the data used in the study was limited to two UFH hospitals, and the performance of the federated learning models may vary when applied to data from other healthcare systems. Further research involving a more diverse set of hospitals and medical centers would be necessary to assess the broader applicability of the federated learning approach.

Additionally, the study did not explore the impact of different federated learning strategies, such as federated multimodal learning or global contrastive training, on the model performance. Investigating the use of more advanced federated learning techniques could potentially lead to further improvements in the accuracy and robustness of the models.

Conclusion

This study demonstrates the potential of federated learning as a valuable tool for training accurate and generalizable AI models for predicting postoperative complications using EHR data, while maintaining patient privacy. The performance of the federated learning models was comparable to or better than models trained on pooled data from multiple hospitals, showcasing the feasibility of this approach in real-world clinical settings where data sharing may be challenging. Further research is needed to explore the scalability and broader applicability of federated learning in healthcare, but this study provides a promising foundation for the continued development of privacy-preserving AI solutions in the medical domain.