Abstract:Background: The accuracy of a prognostic prediction model has become an essential aspect of the quality and reliability of the health-related decisions made by clinicians in modern medicine. Unfortunately, individual institutions often lack sufficient samples, which might not provide sufficient statistical power for models. One mitigation is to expand data collection from a single institution to multiple centers to collectively increase the sample size. However, sharing sensitive biomedical data for research involves complicated issues. Machine learning models such as random forests (RF), though they are commonly used and achieve good performances for prognostic prediction, usually suffer worse performance under multicenter privacy-preserving data mining scenarios compared to a centrally trained version. Methods and materials: In this study, a multicenter random forest prognosis prediction model is proposed that enables federated clinical data mining from horizontally partitioned datasets. By using a novel data enhancement approach based on a differentially private generative adversarial network customized to clinical prognosis data, the proposed model is able to provide a multicenter RF model with performances on par with-or even better than-centrally trained RF but without the need to aggregate the raw data. Moreover, our model also incorporates an importance ranking step designed for feature selection without sharing patient-level information. Result: The proposed model was evaluated on colorectal cancer datasets from the US and China. Two groups of datasets with different levels of heterogeneity within the collaborative research network were selected. First, we compare the performance of the distributed random forest model under different privacy parameters with different percentages of enhancement datasets and validate the effectiveness and plausibility of our approach. Then, we compare the discrimination and calibration ability of the proposed multicenter random forest with a centrally trained random forest model and other tree-based classifiers as well as some commonly used machine learning methods. The results show that the proposed model can provide better prediction performance in terms of discrimination and calibration ability than the centrally trained RF model or the other candidate models while following the privacy-preserving rules in both groups. Additionally, good discrimination and calibration ability are shown on the simplified model based on the feature importance ranking in the proposed approach. Conclusion: The proposed random forest model exhibits ideal prediction capability using multicenter clinical data and overcomes the performance limitation arising from privacy guarantees. It can also provide feature importance ranking across institutions without pooling the data at a central site. This study offers a practical solution for building a prognosis prediction model in the collaborative clinical research network and solves practical issues in real-world applications of medical artificial intelligence.

Predicting treatment response in multicenter non-small cell lung cancer patients based on federated learning

An Efficient Digital Twin Assisted Clustered Federated Learning Algorithm for Disease Prediction.

Predicting anticancer drug sensitivity on distributed data sources using federated deep learning

A comparison between centralized and asynchronous federated learning approaches for survival outcome prediction using clinical and PET data from non-small cell lung cancer patients

A Federated Learning Framework for Brain Tumor Segmentation Without Sharing Patient Data

A multicenter bladder cancer MRI dataset and baseline evaluation of federated learning in clinical application

A decentralized federated learning-based cancer survival prediction method with privacy protection

Federated learning survival model and potential radiotherapy decision support impact assessment for non-small cell lung cancer using real-world data

Advancing oncology with federated learning: transcending boundaries in breast, lung, and prostate cancer. A systematic review

Evaluating Federated Learning Scenarios in a Tumor Classification Application

Federated Learning: A Cross-Institutional Feasibility Study of Deep Learning Based Intracranial Tumor Delineation Framework for Stereotactic Radiosurgery

Medical Federated Model with Mixture of Personalized and Shared Components

Distributed Federated Learning-Based Deep Learning Model for Privacy MRI Brain Tumor Detection

Medical Federated Model with Mixture of Personalized and Sharing Components

Quantitative risk analysis of treatment plans for patients with tumor by mining historical similar patients from electronic health records using federated learning

A Federated Learning Approach to Breast Cancer Prediction in a Collaborative Learning Framework

Federated learning in medicine: facilitating multi-institutional collaborations without sharing patient data

Customized Federated Learning for Multi-Source Decentralized Medical Image Classification

Federated learning for predicting clinical outcomes in patients with COVID-19

Federated learning using model projection for multi-center disease diagnosis with non-IID data

A multicenter random forest model for effective prognosis prediction in collaborative clinical research network