Deep Feature Learning from a Hospital-Scale Chest X-ray Dataset with Application to TB Detection on a Small-Scale Dataset

Ophir Gozes, Hayit Greenspan

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

The use of ImageNet pre-trained networks is becoming widespread in the medical imaging community. It enables training on small datasets, commonly available in medical imaging tasks. The recent emergence of a large Chest X-ray dataset opened the possibility for learning features that are specific to the X-ray analysis task. In this work, we demonstrate that the features learned allow for better classification results for the problem of Tuberculosis detection and enable generalization to an unseen dataset.To accomplish the task of feature learning, we train a DenseNet-121 CNN on 112K images from the ChestXray14 dataset which includes labels of 14 common thoracic pathologies. In addition to the pathology labels, we incorporate meta-data which is available in the dataset: Patient Positioning, Gender and Patient Age. We term this architecture MetaChexNet. As a by-product of the feature learning, we demonstrate state of the art performance on the task of patient Age Gender estimation using CNN's. Finally, we show the features learned using ChestXray14 allow for better transfer learning on small-scale datasets for Tuberculosis.

Original languageEnglish
Title of host publication2019 41st Annual International Conference of the IEEE Engineering in Medicine and Biology Society, EMBC 2019
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages4076-4079
Number of pages4
ISBN (Electronic)9781538613115
DOIs
StatePublished - Jul 2019
Event41st Annual International Conference of the IEEE Engineering in Medicine and Biology Society, EMBC 2019 - Berlin, Germany
Duration: 23 Jul 201927 Jul 2019

Publication series

NameProceedings of the Annual International Conference of the IEEE Engineering in Medicine and Biology Society, EMBS
ISSN (Print)1557-170X

Conference

Conference41st Annual International Conference of the IEEE Engineering in Medicine and Biology Society, EMBC 2019
Country/TerritoryGermany
CityBerlin
Period23/07/1927/07/19

Fingerprint

Dive into the research topics of 'Deep Feature Learning from a Hospital-Scale Chest X-ray Dataset with Application to TB Detection on a Small-Scale Dataset'. Together they form a unique fingerprint.

Cite this