Fast few-shot transfer learning for disease identification from chest x-ray images using autoencoder ensemble

Angshuman Paul; Yu-Xing Tang; Ronald M. Summers

doi:10.1117/12.2549060

16 March 2020 Fast few-shot transfer learning for disease identification from chest x-ray images using autoencoder ensemble

Angshuman Paul, Yu-Xing Tang, Ronald M. Summers

Author Affiliations +

Proceedings Volume 11314, Medical Imaging 2020: Computer-Aided Diagnosis; 1131407 (2020) https://doi.org/10.1117/12.2549060
Event: SPIE Medical Imaging, 2020, Houston, Texas, United States

Abstract

We propose a fast few-shot learning framework that uses transfer learning to identify different lung and chest diseases and conditions from chest x-rays. Our model can be trained with as few as five training examples, making it potentially applicable for diagnosis of rare diseases. In this work, we divide different chest diseases into two disjoint categories: (i) base classes (with large training set) and (ii) novel classes (with a few training examples per class). Our method consists of two steps, namely feature extraction and classification. For the feature extraction, we employ a deep convolutional neural network, customized for chest x-rays. We train the feature extractor with data only from base classes. So the novel classes are unseen to the feature extractor during training. However, we use the feature extractor for extracting features from the data of novel classes resulting in transfer learning. Our classifier, on the other hand, uses the data only from the novel classes for training. We introduce the idea of autoencoder ensemble to design the classifier. Only a few feature vectors from each of the novel classes are used for training the classifier making it a few-shot learner. Incorporating new novel classes require training only the classifier which makes the entire process extremely fast. The performance of the classifier is evaluated on the test data from the novel classes. Experiments show _ 18% improvement in the F1 score compared to the baseline on identifying the novel diseases from publicly available chest x-ray dataset.

Conference Presentation

Citation Download Citation

Angshuman Paul, Yu-Xing Tang, and Ronald M. Summers "Fast few-shot transfer learning for disease identification from chest x-ray images using autoencoder ensemble", Proc. SPIE 11314, Medical Imaging 2020: Computer-Aided Diagnosis, 1131407 (16 March 2020); https://doi.org/10.1117/12.2549060

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