Privacy Preserving and Communication Efficient Information Enhancement for Imbalanced Medical Image Classification

Abstract

AbstractDeep learning methods, especially convolutional neural networks, have become more and more popular in medical image classifications. However, training a deep neural network from scratch can be a luxury for many medical image datasets as the process requires a large and well-balanced sample to output satisfactory results. Unlike natural image datasets, medical images are expensive to collect owing to labor and equipment costs. Besides, the class labels in medical image datasets are usually severely imbalanced subject to the availability of patients. Further, aggregating medical images from multiple sources can be challenging due to policy restrictions, privacy concerns, communication costs, and data heterogeneity caused by equipment differences and labeling discrepancies. In this paper, we propose to address these issues with the help of transfer learning and artificial samples created by generative models. Instead of requesting medical images from source data, our method only needs a parsimonious supplement of model parameters pre-trained on the source data. The proposed method preserves the data privacy in the source data and significantly reduces the communication cost. Our study shows transfer learning together with artificial samples can improve the pneumonia classification accuracy on a small but heavily imbalanced chest X-ray image dataset by \(11.53\%\) which performs even better than directly augmenting that source data into the training process.KeywordsDeep learningGenerative modelsMedical image classificationPrivacy preservationTransfer learning