Ontology highlight
ABSTRACT: Purpose
This study aims to explore training strategies to improve convolutional neural network-based image-to-image deformable registration for abdominal imaging.Methods
Different training strategies, loss functions, and transfer learning schemes were considered. Furthermore, an augmentation layer which generates artificial training image pairs on-the-fly was proposed, in addition to a loss layer that enables dynamic loss weighting.Results
Guiding registration using segmentations in the training step proved beneficial for deep-learning-based image registration. Finetuning the pretrained model from the brain MRI dataset to the abdominal CT dataset further improved performance on the latter application, removing the need for a large dataset to yield satisfactory performance. Dynamic loss weighting also marginally improved performance, all without impacting inference runtime.Conclusion
Using simple concepts, we improved the performance of a commonly used deep image registration architecture, VoxelMorph. In future work, our framework, DDMR, should be validated on different datasets to further assess its value.
SUBMITTER: Perez de Frutos J
PROVIDER: S-EPMC9956065 | biostudies-literature | 2023
REPOSITORIES: biostudies-literature
Pérez de Frutos Javier J Pedersen André A Pelanis Egidijus E Bouget David D Survarachakan Shanmugapriya S Langø Thomas T Elle Ole-Jakob OJ Lindseth Frank F
PloS one 20230224 2
<h4>Purpose</h4>This study aims to explore training strategies to improve convolutional neural network-based image-to-image deformable registration for abdominal imaging.<h4>Methods</h4>Different training strategies, loss functions, and transfer learning schemes were considered. Furthermore, an augmentation layer which generates artificial training image pairs on-the-fly was proposed, in addition to a loss layer that enables dynamic loss weighting.<h4>Results</h4>Guiding registration using segme ...[more]