Classification of chest CT using case-level weak supervision

Ruixiang Tang; Fakrul Islam Tushar; Songyue Han; Rui Hou; Geoffrey D. Rubin; Joseph Y. Lo

doi:10.1117/12.2513576

13 March 2019 Classification of chest CT using case-level weak supervision

Ruixiang Tang, Fakrul Islam Tushar, Songyue Han, Rui Hou, Geoffrey D. Rubin, Joseph Y. Lo

Proceedings Volume 10950, Medical Imaging 2019: Computer-Aided Diagnosis; 1095017 (2019) https://doi.org/10.1117/12.2513576
Event: SPIE Medical Imaging, 2019, San Diego, California, United States

Abstract

Our goal is to investigate using only case-level labels extracted automatically from radiology reports to construct a multi-disease classifier for CT scans with deep learning method. We chose four lung diseases as a start: atelectasis, pulmonary edema, nodule and pneumonia. From a dataset of approximately 5,000 chest CT cases from our institution, we used a rule-based model to analyze those radiologist reports, labeling disease by text mining to identify cases with those diseases. From those results, we randomly selected the following mix of cases: 275 normal, 170 atelectasis, 175 nodule, 195 pulmonary edema, and 208 pneumonia. As a key feature of this study, each chest CT scan was represented by only 10 axial slices (taken at regular intervals through the lungs), and furthermore all slices shared the same label based on the radiology report. So the label was weak, because often disease will not appear in all slices. We used ResNet-50 as our classification model, with 4-fold cross-validation. Each slice was analyzed separately to yield a slice-level performance. For each case, we chose the 5 slices with highest probability and used their mean probability as the final patient-level probability. Performance was evaluated using the receiver operating characteristic (ROC) area under the curve (AUC). For the 4 diseases separately, the slice-based AUCs were 0.71 for nodule, 0.79 for atelectasis, 0.96 for edema, and 0.90 for pneumonia. The patient-based AUC were 0.74 for nodule, 0.83 for atelectasis, 0.97 for edema, and 0.91 for pneumonia. We backprojected the activations of last convolution layer and the weights from prediction layer to synthesize a heat map. This heat map could be an approximate disease detector, also could tell us feature patterns which ResNet-50 focus on.

Conference Presentation

Citation Download Citation

Ruixiang Tang, Fakrul Islam Tushar, Songyue Han, Rui Hou, Geoffrey D. Rubin, and Joseph Y. Lo "Classification of chest CT using case-level weak supervision", Proc. SPIE 10950, Medical Imaging 2019: Computer-Aided Diagnosis, 1095017 (13 March 2019); https://doi.org/10.1117/12.2513576

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available