A Novel Approach for Multi-Label Chest X-Ray Classification of Common Thorax Diseases

Chest X-ray (CXR) is one of the most common types of radiology examination for the diagnosis of thorax diseases. Computer-aided diagnosis (CAD) was developed to help radiologists to achieve diagnostic excellence in a short period of time and to enhance patient healthcare. In this paper, we seek to improve the performance of the CAD system in the task of thorax diseases diagnosis by providing a new method that combines the advantages of CNN models in image feature extraction with those of the problem transformation methods in the multi-label classification task. The experimental study is tested on two publicly available CXR datasets ChestX-ray14 (frontal view) and CheXpert (frontal and lateral views). The results show that our proposed method outperformed the current state of the art.

[1]  Zhi-Hua Zhou,et al.  A k-nearest neighbor based algorithm for multi-label classification , 2005, 2005 IEEE International Conference on Granular Computing.

[2]  Kilian Q. Weinberger,et al.  Densely Connected Convolutional Networks , 2016, 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[3]  Jude Shavlik,et al.  Chapter 11 Transfer Learning , 2009 .

[4]  Amanda Clare,et al.  Knowledge Discovery in Multi-label Phenotype Data , 2001, PKDD.

[5]  Yoshua Bengio,et al.  Gradient-based learning applied to document recognition , 1998, Proc. IEEE.

[6]  Saso Dzeroski,et al.  An extensive experimental comparison of methods for multi-label learning , 2012, Pattern Recognit..

[7]  Ronald M. Summers,et al.  ChestX-ray: Hospital-Scale Chest X-ray Database and Benchmarks on Weakly Supervised Classification and Localization of Common Thorax Diseases , 2019, Deep Learning and Convolutional Neural Networks for Medical Imaging and Clinical Informatics.

[8]  Muktabh Mayank Srivastava,et al.  Boosted Cascaded Convnets for Multilabel Classification of Thoracic Diseases in Chest Radiographs , 2017, ICIAR.

[9]  Ashequl Qadir,et al.  Large Scale Automated Reading of Frontal and Lateral Chest X-Rays using Dual Convolutional Neural Networks , 2018, ArXiv.

[10]  Yi Yang,et al.  Diagnose like a Radiologist: Attention Guided Convolutional Neural Network for Thorax Disease Classification , 2018, ArXiv.

[11]  Yifan Yu,et al.  CheXpert: A Large Chest Radiograph Dataset with Uncertainty Labels and Expert Comparison , 2019, AAAI.

[12]  Li Yao,et al.  Learning to diagnose from scratch by exploiting dependencies among labels , 2017, ArXiv.

[13]  Jian Sun,et al.  Deep Residual Learning for Image Recognition , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[14]  Yuanping Zhu,et al.  Calibrated Rank-SVM for multi-label image categorization , 2008, 2008 IEEE International Joint Conference on Neural Networks (IEEE World Congress on Computational Intelligence).

[15]  Andrew Zisserman,et al.  Very Deep Convolutional Networks for Large-Scale Image Recognition , 2014, ICLR.

[16]  Andrew Y. Ng,et al.  CheXNet: Radiologist-Level Pneumonia Detection on Chest X-Rays with Deep Learning , 2017, ArXiv.

[17]  Grigorios Tsoumakas,et al.  Random k -Labelsets: An Ensemble Method for Multilabel Classification , 2007, ECML.

[18]  Geoff Holmes,et al.  Classifier chains for multi-label classification , 2009, Machine Learning.

[19]  Kristian Kersting,et al.  How is a data-driven approach better than random choice in label space division for multi-label classification? , 2016, Entropy.

[20]  Ronan McDermott,et al.  Discrepancy and Error in Radiology: Concepts, Causes and Consequences , 2012, The Ulster medical journal.

[21]  Jiebo Luo,et al.  Learning multi-label scene classification , 2004, Pattern Recognit..

[22]  Zenghui Wang,et al.  Deep Convolutional Neural Networks for Image Classification: A Comprehensive Review , 2017, Neural Computation.

[23]  Fei-Fei Li,et al.  ImageNet: A large-scale hierarchical image database , 2009, 2009 IEEE Conference on Computer Vision and Pattern Recognition.

[24]  Grigorios Tsoumakas,et al.  Random K-labelsets for Multilabel Classification , 2022 .

[25]  Jesse Read,et al.  A Pruned Problem Transformation Method for Multi-label Classification , 2008 .