Deep Transfer Learning for the Multilabel Classification of Chest X-ray Images

Guan Hua Huang*, Qi Jia Fu, Ming Zhang Gu, Nan Han Lu, Kuo Ying Liu, Tai Been Chen

*此作品的通信作者

研究成果: Article同行評審

15 引文 斯高帕斯(Scopus)

摘要

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.

原文English
文章編號1457
期刊Diagnostics
12
發行號6
DOIs
出版狀態Published - 6月 2022

指紋

深入研究「Deep Transfer Learning for the Multilabel Classification of Chest X-ray Images」主題。共同形成了獨特的指紋。

引用此