Impact of Initialization on Intra-subject Pediatric Brain MR Image Registration: A Comparative Analysis between SyN ANTs and Deep Learning-Based Approaches

Andjela Dimitrijevic1,20000-0002-8799-6300, Vincent Noblet30000-0002-3655-3163, Benjamin De Leener1,2,40000-0002-1378-2756
1: NeuroPoly Lab, Institute of Biomedical Engineering, Polytechnique Montréal, Montréal, QC, Canada, 2: Research Center, Ste-Justine Hospital University Centre, Montréal, QC, Canada, 3: ICube-UMR 7357, Université de Strasbourg, CNRS, Strasbourg, France, 4: Computer Engineering and Software Engineering, Polytechnique Montréal, Montréal, QC, Canada
Publication date: 2024/06/28
https://doi.org/10.59275/j.melba.2024-151b
PDF · Code

Abstract

This study evaluates the performance of conventional SyN ANTs and learning-based reg- istration methods in the context of pediatric neuroimaging, specifically focusing on intra- subject deformable registration. The comparison involves three approaches—without (NR), with rigid (RR), and with rigid and affine (RAR) initializations. In addition to initial- ization, performances are evaluated in terms of accuracy, speed, and the impact of age intervals and sex per pair. Data consists of the publicly available MRI scans from the Calgary Preschool dataset, which includes 63 children aged 2-7 years, allowing for 431 registration pairs. We implemented the unsupervised deep learning (DL) framework with a U-Net architecture using DeepReg and it was 5-fold cross-validated. The evaluation includes Dice scores for tissue segmentation from 18 smaller regions obtained by SynthSeg, analysis of log Jacobian determinants, and registration pro-rated training and inference times. Learning-based approaches, with or without linear initializations, exhibit slight superiority over SyN ANTs in terms of Dice scores. Specifically, DL-based implementations with RR and RAR initializations significantly outperform SyN ANTs. The lower Dice scores of SyN ANTs are likely due to its lack of population-based optimization, unlike the DL methods which learn optimal parameters through training. Both SyN ANTs and DL-based registration involve parameter optimization, but the choice between these methods depends on the scale of registration—network-based for broader coverage or SyN ANTs for specific structures. Learning-based registration offers fast inference times but needs training, whereas SyN ANTs requires manual fine-tuning, with less clear guidelines, particularly for younger cohorts. Both methods face challenges with larger age intervals due to greater growth changes. Future work will extend the framework to younger populations and explore models that better separate different levels of transformations for improved local brain region registration. The main takeaway is that while DL-based methods show promise with faster and more accurate registrations, SyN ANTs remains robust and generalizable without the need for extensive training, highlighting the importance of method selection based on specific registration needs in the pediatric context. Our code is available at https://github.com/neuropoly/pediatric-DL-registration

Keywords

Deep Learning · MRI · Pediatric · Image Registration · Learning-based Registration

Bibtex @article{melba:2024:013:dimitrijevic, title = "Impact of Initialization on Intra-subject Pediatric Brain MR Image Registration: A Comparative Analysis between SyN ANTs and Deep Learning-Based Approaches ", author = "Dimitrijevic, Andjela and Noblet, Vincent and De Leener, Benjamin", journal = "Machine Learning for Biomedical Imaging", volume = "2", issue = "Special Issue on Image Registration", year = "2024", pages = "916--955", issn = "2766-905X", doi = "https://doi.org/10.59275/j.melba.2024-151b", url = "https://melba-journal.org/2024:013" }
RISTY - JOUR AU - Dimitrijevic, Andjela AU - Noblet, Vincent AU - De Leener, Benjamin PY - 2024 TI - Impact of Initialization on Intra-subject Pediatric Brain MR Image Registration: A Comparative Analysis between SyN ANTs and Deep Learning-Based Approaches T2 - Machine Learning for Biomedical Imaging VL - 2 IS - Special Issue on Image Registration SP - 916 EP - 955 SN - 2766-905X DO - https://doi.org/10.59275/j.melba.2024-151b UR - https://melba-journal.org/2024:013 ER -

2024:013 cover