GPU-based stochastic-gradient optimization for non-rigid medical image registration in time-critical applications

Parag Bhosale; Marius Staring; Zaid Al-Ars; Floris F. Berendsen

doi:10.1117/12.2293098

2 March 2018 GPU-based stochastic-gradient optimization for non-rigid medical image registration in time-critical applications

Parag Bhosale, Marius Staring, Zaid Al-Ars, Floris F. Berendsen

Author Affiliations +

Proceedings Volume 10574, Medical Imaging 2018: Image Processing; 105740R (2018) https://doi.org/10.1117/12.2293098
Event: SPIE Medical Imaging, 2018, Houston, Texas, United States

Abstract

Currently, non-rigid image registration algorithms are too computationally intensive to use in time-critical applications. Existing implementations that focus on speed typically address this by either parallelization on GPU-hardware, or by introducing methodically novel techniques into CPU-oriented algorithms. Stochastic gradient descent (SGD) optimization and variations thereof have proven to drastically reduce the computational burden for CPU-based image registration, but have not been successfully applied in GPU hardware due to its stochastic nature. This paper proposes 1) NiftyRegSGD, a SGD optimization for the GPU-based image registration tool NiftyReg, 2) random chunk sampler, a new random sampling strategy that better utilizes the memory bandwidth of GPU hardware. Experiments have been performed on 3D lung CT data of 19 patients, which compared NiftyRegSGD (with and without random chunk sampler) with CPU-based elastix Fast Adaptive SGD (FASGD) and NiftyReg. The registration runtime was 21.5s, 4.4s and 2.8s for elastix-FASGD, NiftyRegSGD without, and NiftyRegSGD with random chunk sampling, respectively, while similar accuracy was obtained. Our method is publicly available at https://github.com/SuperElastix/NiftyRegSGD.

Conference Presentation

Citation Download Citation

Parag Bhosale, Marius Staring, Zaid Al-Ars, and Floris F. Berendsen "GPU-based stochastic-gradient optimization for non-rigid medical image registration in time-critical applications", Proc. SPIE 10574, Medical Imaging 2018: Image Processing, 105740R (2 March 2018); https://doi.org/10.1117/12.2293098

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