Federated learning model for predicting major postoperative complications

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Park, Yonggi, Ren, Yuanfang, Shickel, Benjamin, Guan, Ziyuan, Patela, Ayush, Ma, Yingbo, Hu, Zhenhong, Loftus, Tyler J., Rashidi, Parisa, Ozrazgat-Baslanti, Tezcan, Bihorac, Azra
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866917636005167104
author Park, Yonggi
Ren, Yuanfang
Shickel, Benjamin
Guan, Ziyuan
Patela, Ayush
Ma, Yingbo
Hu, Zhenhong
Loftus, Tyler J.
Rashidi, Parisa
Ozrazgat-Baslanti, Tezcan
Bihorac, Azra
author_facet Park, Yonggi
Ren, Yuanfang
Shickel, Benjamin
Guan, Ziyuan
Patela, Ayush
Ma, Yingbo
Hu, Zhenhong
Loftus, Tyler J.
Rashidi, Parisa
Ozrazgat-Baslanti, Tezcan
Bihorac, Azra
contents Background: The accurate prediction of postoperative complication risk using Electronic Health Records (EHR) and artificial intelligence shows great potential. Training a robust artificial intelligence model typically requires large-scale and diverse datasets. In reality, collecting medical data often encounters challenges surrounding privacy protection. Methods: This retrospective cohort study includes adult patients who were admitted to UFH Gainesville (GNV) (n = 79,850) and Jacksonville (JAX) (n = 28,636) for any type of inpatient surgical procedure. Using perioperative and intraoperative features, we developed federated learning models to predict nine major postoperative complications (i.e., prolonged intensive care unit stay and mechanical ventilation). We compared federated learning models with local learning models trained on a single site and central learning models trained on pooled dataset from two centers. Results: Our federated learning models achieved the area under the receiver operating characteristics curve (AUROC) values ranged from 0.81 for wound complications to 0.92 for prolonged ICU stay at UFH GNV center. At UFH JAX center, these values ranged from 0.73-0.74 for wound complications to 0.92-0.93 for hospital mortality. Federated learning models achieved comparable AUROC performance to central learning models, except for prolonged ICU stay, where the performance of federated learning models was slightly higher than central learning models at UFH GNV center, but slightly lower at UFH JAX center. In addition, our federated learning model obtained comparable performance to the best local learning model at each center, demonstrating strong generalizability. Conclusion: Federated learning is shown to be a useful tool to train robust and generalizable models from large scale data across multiple institutions where data protection barriers are high.
format Preprint
id arxiv_https___arxiv_org_abs_2404_06641
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Federated learning model for predicting major postoperative complications
Park, Yonggi
Ren, Yuanfang
Shickel, Benjamin
Guan, Ziyuan
Patela, Ayush
Ma, Yingbo
Hu, Zhenhong
Loftus, Tyler J.
Rashidi, Parisa
Ozrazgat-Baslanti, Tezcan
Bihorac, Azra
Machine Learning
Artificial Intelligence
Computers and Society
Background: The accurate prediction of postoperative complication risk using Electronic Health Records (EHR) and artificial intelligence shows great potential. Training a robust artificial intelligence model typically requires large-scale and diverse datasets. In reality, collecting medical data often encounters challenges surrounding privacy protection. Methods: This retrospective cohort study includes adult patients who were admitted to UFH Gainesville (GNV) (n = 79,850) and Jacksonville (JAX) (n = 28,636) for any type of inpatient surgical procedure. Using perioperative and intraoperative features, we developed federated learning models to predict nine major postoperative complications (i.e., prolonged intensive care unit stay and mechanical ventilation). We compared federated learning models with local learning models trained on a single site and central learning models trained on pooled dataset from two centers. Results: Our federated learning models achieved the area under the receiver operating characteristics curve (AUROC) values ranged from 0.81 for wound complications to 0.92 for prolonged ICU stay at UFH GNV center. At UFH JAX center, these values ranged from 0.73-0.74 for wound complications to 0.92-0.93 for hospital mortality. Federated learning models achieved comparable AUROC performance to central learning models, except for prolonged ICU stay, where the performance of federated learning models was slightly higher than central learning models at UFH GNV center, but slightly lower at UFH JAX center. In addition, our federated learning model obtained comparable performance to the best local learning model at each center, demonstrating strong generalizability. Conclusion: Federated learning is shown to be a useful tool to train robust and generalizable models from large scale data across multiple institutions where data protection barriers are high.
title Federated learning model for predicting major postoperative complications
topic Machine Learning
Artificial Intelligence
Computers and Society
url https://arxiv.org/abs/2404.06641