Share Email Print

Proceedings Paper

Fast few-shot transfer learning for disease identification from chest x-ray images using autoencoder ensemble
Author(s): Angshuman Paul; Yu-Xing Tang; Ronald M. Summers
Format Member Price Non-Member Price
PDF $17.00 $21.00

Paper 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.

Paper Details

Date Published: 16 March 2020
PDF: 6 pages
Proc. SPIE 11314, Medical Imaging 2020: Computer-Aided Diagnosis, 1131407 (16 March 2020); doi: 10.1117/12.2549060
Show Author Affiliations
Angshuman Paul, National Institutes of Health Clinical Ctr. (United States)
Yu-Xing Tang, National Institutes of Health Clinical Ctr. (United States)
Ronald M. Summers, National Institutes of Health Clinical Ctr. (United States)

Published in SPIE Proceedings Vol. 11314:
Medical Imaging 2020: Computer-Aided Diagnosis
Horst K. Hahn; Maciej A. Mazurowski, Editor(s)

© SPIE. Terms of Use
Back to Top
Sign in to read the full article
Create a free SPIE account to get access to
premium articles and original research
Forgot your username?