Deep evidential uncertainty estimation in joint synthesis and registration of MRI and cone-beam CT images

M. Yi; R. Duan; Z. Li; J. H. Siewerdsen; A. Uneri; J. Lee; C. K. Jones

doi:10.1117/12.3008899

29 March 2024 Deep evidential uncertainty estimation in joint synthesis and registration of MRI and cone-beam CT images

M. Yi, R. Duan, Z. Li, J. H. Siewerdsen, A. Uneri, J. Lee, C. K. Jones

Proceedings Volume 12928, Medical Imaging 2024: Image-Guided Procedures, Robotic Interventions, and Modeling; 129280X (2024) https://doi.org/10.1117/12.3008899
Event: SPIE Medical Imaging, 2024, San Diego, California, United States

Abstract

Purpose. The purpose of this work is to produce uncertainty estimation for the registration task between MR and CBCT based on the joint synthesis and registration network. This novel framework aims to equip surgeons with robust tools for assessing the fidelity of registration, enabling informed decisions regarding the volume alignment, particularly in regions where uncertainty is prevalent. Methods. The prediction of deformation fields is framed as a regression problem, and therefore allows Evidential Deep Learning (EDL) for uncertainty estimation. By utilizing a loss function grounded in likelihood, the network is compelled to simultaneously learn the parameters of EDL and generate precise predictions. This adaptive approach is tailored specifically for the registration task, rendering the estimation method notably flexible for unsupervised registration. Results. The visualization of the uncertainty map demonstrates areas where the registered ventricle deviates from the ground truth alignment. Additionally, the incorporation of uncertainty assessment enhances registration performance, as evidenced by quantitative evaluations across seven distinct brain regions. Comparative analysis between networks trained with and without uncertainty underscores the performance of the uncertainty-aware model in achieving more accurate registrations. Conclusions. The preliminary results show that incorporating uncertainty estimation into registration model leads to improved registration performance. Furthermore, the estimated uncertainty map is also useful for surgeons to evaluate the reliability of registered images.

Conference Presentation

(2024) Published by SPIE. Downloading of the abstract is permitted for personal use only.

Citation Download Citation

M. Yi, R. Duan, Z. Li, J. H. Siewerdsen, A. Uneri, J. Lee, and C. K. Jones "Deep evidential uncertainty estimation in joint synthesis and registration of MRI and cone-beam CT images", Proc. SPIE 12928, Medical Imaging 2024: Image-Guided Procedures, Robotic Interventions, and Modeling, 129280X (29 March 2024); https://doi.org/10.1117/12.3008899

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