Deep-learning-based deformable image registration of head CT and MRI scans

Abstract

This work is motivated by the lack of publications on the direct application of multimodal image registration with deep-learning techniques for the enhancement of treatment planning in particle therapy. An unsupervised workflow, which seeks to improve image alignment, was developed and evaluated for computed tomography and magnetic resonance imaging scans of the head. The scans of 39 paediatric patients with brain tumours were available. The focus of the two-step workflow, including preprocessing of the scans for normalisation, is deformable image registration (DIR) with a deep neural network, which generates deformation vector fields (DVFs). To obtain a suitable configuration of the network, parameter tuning is performed by varying its parameters, e.g., layer size, regularisation (λ) of the DVF and learning rate (α). Image similarity was determined with the Dice similarity coefficient, mDSC, using segmented images and the mutual-information metric, mMI. The performance of the deep-learning models was assessed with the inverse consistency, mIC, and the Jacobian determinant, mJD. Inverse consistency is obtained for mIC = 0 mm, while the determinant of a deformed image is expected to be unity. The deep-learning models passed both performance checks, indicated by the mean values m̄IC=(0.57±1.00)mm and m̄JD=(1.00±0.07). Models with λ ≥ 1 yielded higher mDSC values than models with lower λ values. A small-architecture model with α = 10–4 was found to be most suitable for DIR, as improvement in image similarity of up to 12% was obtained in terms of mMI. The direct application of deep-learning models produced registered images improving image alignment between scans of different modalities.