Journal
DIAGNOSTICS
Volume 12, Issue 6, Pages -Publisher
MDPI
DOI: 10.3390/diagnostics12061457
Keywords
convolutional neural network; deep learning; source data set; supervised classification
Categories
Funding
- Ministry of Science and Technology, Taiwan [MOST 107-2118-M-009-005-MY2, MOST 109-2118-M-009-004-MY2]
Ask authors/readers for more resources
In this study, deep convolutional neural network models were used to extract features and identify diseases in CXR images. Transfer learning combined with large open-source image data sets was utilized to overcome insufficient training data and optimize the classification model. The results demonstrated the effectiveness of transfer learning in enhancing prediction capabilities and reducing computing costs for CXR images.
Chest X-ray (CXR) is widely used to diagnose conditions affecting the chest, its contents, and its nearby structures. In this study, we used a private data set containing 1630 CXR images with disease labels; most of the images were disease-free, but the others contained multiple sites of abnormalities. Here, we used deep convolutional neural network (CNN) models to extract feature representations and to identify possible diseases in these images. We also used transfer learning combined with large open-source image data sets to resolve the problems of insufficient training data and optimize the classification model. The effects of different approaches of reusing pretrained weights (model finetuning and layer transfer), source data sets of different sizes and similarity levels to the target data (ImageNet, ChestX-ray, and CheXpert), methods integrating source data sets into transfer learning (initiating, concatenating, and co-training), and backbone CNN models (ResNet50 and DenseNet121) on transfer learning were also assessed. The results demonstrated that transfer learning applied with the model finetuning approach typically afforded better prediction models. When only one source data set was adopted, ChestX-ray performed better than CheXpert; however, after ImageNet initials were attached, CheXpert performed better. ResNet50 performed better in initiating transfer learning, whereas DenseNet121 performed better in concatenating and co-training transfer learning. Transfer learning with multiple source data sets was preferable to that with a source data set. Overall, transfer learning can further enhance prediction capabilities and reduce computing costs for CXR images.
Authors
I am an author on this paper
Click your name to claim this paper and add it to your profile.
Reviews
Recommended
No Data Available