Distributed Deep Learning Across Multi-site Datasets for Generalized CT Hemorrhage Segmentation.

Researchers

Journal

Modalities

Models

Abstract

As deep neural networks achieve more success in the wide field of computer vision, greater emphasis is being placed on the generalizations of these models for production deployment. With sufficiently large training datasets, models can typically avoid overfitting their data; however, for medical imaging it is often difficult to obtain enough data from a single site. Sharing data between institutions is also frequently nonviable or prohibited due to security measures and research compliance constraints, enforced to guard protected health information (PHI) and patient anonymity.
In this paper, we implement cyclic weight transfer with independent datasets from multiple geographically disparate sites without compromising PHI. We compare results between single-site learning (SSL) and multi-site learning (MSL) models on testing data drawn from each of the training sites as well as two other institutions.
The MSL model attains an average Dice Similarity Coefficient (DSC) of 0:690 on the holdout institution datasets with a volume correlation of 0:914, respectively corresponding to a 7% and 5% statistically significant improvement over the average of both SSL models, which attained an average DSC of 0:646 and average correlation of 0:871.
We show that a neural network can be efficiently trained on data from two physically remote sites without consolidating patient data to a single location. The resulting network improves model generalization and achieves higher average DSCs on external datasets than neural networks trained on data from a single source.
© 2019 American Association of Physicists in Medicine.