Impact of data augmentation techniques on a deep learning based medical imaging task

Sandeep Dutta; Prakhar Prakash; Christopher G. Matthews

doi:10.1117/12.2549806

2 March 2020 Impact of data augmentation techniques on a deep learning based medical imaging task

Sandeep Dutta, Prakhar Prakash, Christopher G. Matthews

Proceedings Volume 11318, Medical Imaging 2020: Imaging Informatics for Healthcare, Research, and Applications; 113180M (2020) https://doi.org/10.1117/12.2549806
Event: SPIE Medical Imaging, 2020, Houston, Texas, United States

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.

Conference Presentation

Citation Download Citation

Sandeep Dutta, Prakhar Prakash, and Christopher G. Matthews "Impact of data augmentation techniques on a deep learning based medical imaging task", Proc. SPIE 11318, Medical Imaging 2020: Imaging Informatics for Healthcare, Research, and Applications, 113180M (2 March 2020); https://doi.org/10.1117/12.2549806

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