12 research outputs found
Learning Conditional Deformable Templates with Convolutional Networks
We develop a learning framework for building deformable templates, which play
a fundamental role in many image analysis and computational anatomy tasks.
Conventional methods for template creation and image alignment to the template
have undergone decades of rich technical development. In these frameworks,
templates are constructed using an iterative process of template estimation and
alignment, which is often computationally very expensive. Due in part to this
shortcoming, most methods compute a single template for the entire population
of images, or a few templates for specific sub-groups of the data. In this
work, we present a probabilistic model and efficient learning strategy that
yields either universal or conditional templates, jointly with a neural network
that provides efficient alignment of the images to these templates. We
demonstrate the usefulness of this method on a variety of domains, with a
special focus on neuroimaging. This is particularly useful for clinical
applications where a pre-existing template does not exist, or creating a new
one with traditional methods can be prohibitively expensive. Our code and
atlases are available online as part of the VoxelMorph library at
http://voxelmorph.csail.mit.edu.Comment: NeurIPS 2019: Neural Information Processing Systems. Keywords:
deformable templates, conditional atlases, diffeomorphic image registration,
probabilistic models, neuroimagin
Learning the Effect of Registration Hyperparameters with HyperMorph
We introduce HyperMorph, a framework that facilitates efficient
hyperparameter tuning in learning-based deformable image registration.
Classical registration algorithms perform an iterative pair-wise optimization
to compute a deformation field that aligns two images. Recent learning-based
approaches leverage large image datasets to learn a function that rapidly
estimates a deformation for a given image pair. In both strategies, the
accuracy of the resulting spatial correspondences is strongly influenced by the
choice of certain hyperparameter values. However, an effective hyperparameter
search consumes substantial time and human effort as it often involves training
multiple models for different fixed hyperparameter values and may lead to
suboptimal registration. We propose an amortized hyperparameter learning
strategy to alleviate this burden by learning the impact of hyperparameters on
deformation fields. We design a meta network, or hypernetwork, that predicts
the parameters of a registration network for input hyperparameters, thereby
comprising a single model that generates the optimal deformation field
corresponding to given hyperparameter values. This strategy enables fast,
high-resolution hyperparameter search at test-time, reducing the inefficiency
of traditional approaches while increasing flexibility. We also demonstrate
additional benefits of HyperMorph, including enhanced robustness to model
initialization and the ability to rapidly identify optimal hyperparameter
values specific to a dataset, image contrast, task, or even anatomical region,
all without the need to retrain models. We make our code publicly available at
http://hypermorph.voxelmorph.net.Comment: Accepted for publication at the Journal of Machine Learning for
Biomedical Imaging (MELBA) at https://www.melba-journal.or
Recommended from our members
Multimodal Investigation of Brain Network Systems: From Brain Structure and Function to Connectivity and Neuromodulation
The field of cognitive neuroscience has benefited greatly from multimodal investigations of the human brain, which integrate various tools and neuroimaging data to understand brain functions and guide treatments for brain disorders. In this dissertation, we present a series of studies that illustrate the use of multimodal approaches to investigate brain structure and function, brain connectivity, and neuromodulation effects.
Firstly, we propose a novel landmark-guided region-based spatial normalization technique to accurately quantify brain morphology, which can improve the sensitivity and specificity of functional imaging studies. Subsequently, we shift the investigation to the characteristics of functional brain activity due to visual stimulations. Our findings reveal that the task-evoked positive blood-oxygen-level dependent (BOLD) response is accompanied by sustained negative BOLD responses in the visual cortex. These negative BOLD responses are likely generated through subcortical neuromodulatory systems with distributed ascending projections to the cortex.
To further explore the cortico-subcortical relationship, we conduct a multimodal investigation that involves simultaneous data acquisition of pupillometry, electroencephalography (EEG), and functional magnetic resonance imaging (fMRI). This investigation aims to examine the connectivity of brain circuits involved in the cognitive processes of salient stimuli. Using pupillary response as a surrogate measure of activity in the locus coeruleus-norepinephrine system, we find that the pupillary response is associated with the reorganization of functional brain networks during salience processing.
In addition, we propose a cortico-subcortical integrated network reorganization model with potential implications for understanding attentional processing and network switching. Lastly, we employ a multimodal investigation that involves concurrent transcranial magnetic stimulation (TMS), EEG, and fMRI to explore network perturbations and measurements of the propagation effects. In a preliminary exploration on brain-state dependency of TMS-induced effects, we find that the propagation of left dorsolateral prefrontal cortex TMS to regions in the lateral frontoparietal network might depend on the brain-state, as indexed by the EEG prefrontal alpha phase.
Overall, the studies in this dissertation contribute to the understanding of the structural and functional characteristics of brain network systems, and may inform future investigations that use multimodal methodological approaches, such as pupillometry, brain connectivity, and neuromodulation tools. The work presented in this dissertation has potential implications for the development of efficient and personalized treatments for major depressive disorder, attention deficit hyperactivity disorder, and Alzheimer's disease
TOWARD SOLVING GROUPWISE MEDICAL IMAGE ANALYSIS PROBLEMS WITH DEEP LEARNING
Image regression, atlas building, and multi-atlas segmentation are three groupwise medical image analysis problems extended from image registration. These three problems are challenging because of the difficulty in establishing spatial correspondences and the associated high computational cost. Specifically, most previous methods are computationally costly as they are optimization-based approaches. Hence fast and accurate approaches are highly desirable. This dissertation addresses the following problems concerning the three groupwise medical im- age analysis problems: (1) fast and reliable geodesic regression for image time series; (2) joint atlas building and diffeomorphic registration learning; (3) efficient and accurate label fusion for multi-atlas segmentation; and (4) spatially localized probability calibration for semantic segmentation networks. Specifically, the contributions in this thesis are as follows: (1) A fast predictive simple geodesic regression approach is proposed to capture the frequently subtle deformation trends of longitudinal image data. (2) A new deep learning model that jointly builds an atlas and learns the diffeomorphic registrations in both the atlas-to-image and the image-to-atlas directions is developed. (3) A novel deep learning label fusion method (VoteNet) that locally identifies sets of trustworthy atlases is presented; and several ways to improve the performance under the VoteNet based multi-atlas segmentation framework are explored. (4) A learning-based local temperature scaling method that predicts a separate temperature scale for each pixel/voxel is designed. The resulting post-processing approach is accuracy preserving and is theoretically guaranteed to be effective.Doctor of Philosoph