Share Email Print

Proceedings Paper

Impact of data augmentation techniques on a deep learning based medical imaging task
Author(s): Sandeep Dutta; Prakhar Prakash; Christopher G. Matthews
Format Member Price Non-Member Price
PDF $17.00 $21.00

Paper Abstract

For any deep learning (DL) based task, model generalization and prediction performance improve as a function of training data set size and variety. However, its application to medical imaging is still challenging because of the limited availability of high-quality and sufficiently diverse annotated data. Data augmentation techniques can improve the model performance when the available dataset size is limited. Anatomy region localization from medical images can be automated with deep learning and is important for tasks such as organ segmentation and lesion detection. Different data augmentation methods were compared for DL based anatomy region localization with computed tomography images. The impact of different neural network architectures was also explored. The prediction accuracy on an independent test set improved from 88% to 97% with optimal selection of data augmentation and architecture while using the same training dataset. Data augmentation steps such as zoom, translation and flips had incremental effect on classifier performance whereas samplewise mean shift appeared to degrade the classifier performance. Global average pooling improved classifier accuracy compared to fully-connected layer when limited data augmentation was used. All model architectures converged to an optimal performance with the right combination of augmentation steps. Prediction inaccuracies were mostly observed in the boundary regions between anatomies. The networks also successfully localized anatomy for Positron Emission Tomography studies reaching an accuracy of up to 97%. Similar impact of data augmentation and pooling layer was also observed.

Paper Details

Date Published: 2 March 2020
PDF: 10 pages
Proc. SPIE 11318, Medical Imaging 2020: Imaging Informatics for Healthcare, Research, and Applications, 113180M (2 March 2020); doi: 10.1117/12.2549806
Show Author Affiliations
Sandeep Dutta, GE Healthcare (United States)
Prakhar Prakash, GE Healthcare (United States)
Christopher G. Matthews, GE Healthcare (United States)

Published in SPIE Proceedings Vol. 11318:
Medical Imaging 2020: Imaging Informatics for Healthcare, Research, and Applications
Po-Hao Chen; Thomas M. Deserno, 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?