Cortical surface registration using unsupervised learning
A.A. Martinos Center for Biomedical Imaging, Massachusetts General Hospital, Harvard Medical School(哈佛医学院麻省总医院A.A.马丁诺斯生物医学成像中心);Computer Science and Artificial Intelligence Laboratory, Massachusetts Institute of Technology(麻省理工学院计算机科学与人工智能实验室)
Jieyu Cheng, Adrian V. Dalca, Bruce Fischl, Lilla Zollei
备注:
cortical surface registration, deep network, unsupervised learning, registration, deep learning, cortical, spherical, invertible
Non-rigid cortical registration is an important and challenging task due to the geometric complexity of the human cortex and the high degree of inter-subject variability. A conventional solution is to use a spherical representation of surface properties and perform registration by aligning cortical folding patterns in that space. This strategy produces accurate spatial alignment but often requires a high computational cost. Recently, convolutional neural networks (CNNs) have demonstrated the potential to dramatically speed up volumetric registration. However, due to distortions introduced by projecting a sphere to a 2D plane, a direct application of recent learning-based methods to surfaces yields poor results. In this study, we present SphereMorph, a diffeomorphic registration framework for cortical surfaces using deep networks that addresses these issues. SphereMorph uses a UNet-style network associated with a spherical kernel to learn the displacement field and warps the sphere using a modified spatial transformer layer. We propose a resampling weight in computing the data fitting loss to account for distortions introduced by polar projection, and demonstrate the performance of our proposed method on two tasks, including cortical parcellation and group-wise functional area alignment. The experiments show that the proposed SphereMorph is capable of modeling the geometric registration problem in a CNN framework and demonstrate superior registration accuracy and computational efficiency. The source code of SphereMorph will be released to the public upon acceptance of this manuscript at https://github.com/voxelmorph/spheremorph.
JParc: Joint cortical surface parcellation with registration
1 Athinoula A. Martinos Center for Biomedical Imaging, Department of Radiology, MGH \& HMS 2 Center for Neurotechnology;Neurorecovery, Department of Neurology, MGH \& HMS 3 Computer Science;Artificial Intelligence Laboratory, EECS, MIT
Joint cortical registration of geometry and function using semi-supervised learning
Massachusetts General Hospital(麻省总医院);Harvard Medical School(哈佛医学院);Massachusetts Institute of Technology(麻省理工学院);McGovern Institute for Brain Research(麦戈文脑科学研究所);Harvard University(哈佛大学)
CortexMorph: fast cortical thickness estimation via diffeomorphic registration using VoxelMorph
Support Center for Advanced Neuroimaging (SCAN)(高级神经影像支持中心(SCAN));University Institute of Diagnostic and Interventional Neuroradiology(诊断与介入神经放射学大学研究所);Inselspital, Bern University Hospital(伯尔尼大学医院 Inselspital)
JOSA: Joint surface-based registration and atlas construction of brain geometry and function
McGovern Institute for Brain Research, MIT(麦戈文脑科学研究所(麻省理工学院));Harvard University(哈佛大学);Computer Science and Artificial Intelligence Laboratory, MIT(计算机科学与人工智能实验室(麻省理工学院))
Uncertainty-Aware Cortical Surface Reconstruction and Spherical Registration for Heterogeneous Brain MRI
Danish Research Centre for Magnetic Resonance, Department of Radiology and Nuclear Medicine, Copenhagen University Hospital – Amager and Hvidovre(丹麦磁共振研究中心,放射科与核医学科,哥本哈根大学医院-阿马格和维多弗);Department of Health Technology, Technical University of Denmark(技术丹麦大学健康技术系);Athinoula A. Martinos Center for Biomedical Imaging, Massachusetts General Hospital(阿蒂努拉·A·马丁诺斯生物医学成像中心,麻省总医院);Computer Science and Artificial Intelligence Lab, Massachusetts Institute of Technology(麻省理工学院计算机科学与人工智能实验室);Department of Radiology, Harvard Medical School(哈佛医学院放射学系);Hawkes Institute, University College London(霍克斯研究所,伦敦大学学院)
We present cortical surface parcellation using spherical deep convolutional neural networks. Traditional multi-atlas cortical surface parcellation requires inter-subject surface registration using geometric features with high processing time on a single subject (2-3 hours). Moreover, even optimal surface registration does not necessarily produce optimal cortical parcellation as parcel boundaries are not fully matched to the geometric features. In this context, a choice of training features is important for accurate cortical parcellation. To utilize the networks efficiently, we propose cortical…
Image registration is a fundamental building block for various applications in medical image analysis. To better explore the correlation between the fixed and moving images and improve registration performance, we propose a novel deep learning network, Co-Attention guided Registration Network (CAR-Net). CAR-Net employs a co-attention block to learn a new representation of the inputs, which drives the registration of the fixed and moving images. Experiments on UK Biobank cardiac cine-magnetic resonance image data demonstrate that CAR-Net obtains higher registration accuracy and smoother deforma…
Learning Cortical Anomaly through Masked Encoding for Unsupervised Heterogeneity Mapping
Department of Psychiatry and Psychotherapy, Tübingen Center for Mental Health (TÜCMH)(图宾根心理健康中心精神病学与心理治疗系);Norwegian Centre for Mental Disorders Research(挪威精神障碍研究中心);Donders Institute for Brain, Cognition, and Behaviour, Radboud University(拉德堡德大学唐德斯脑、认知与行为研究所)
University of Cambridge(剑桥大学);Cancer Research UK Cambridge Centre(英国癌症研究中心剑桥中心);Medical University Vienna(维也纳医科大学);Cambridge University Hospitals National Health Service (NHS) Foundation Trust(剑桥大学医院国民保健服务基金会信托);Eindhoven University of Technology(埃因霍温理工大学);Leiden University(莱顿大学);University College London(伦敦大学学院);Cancer Research UK Cambridge Institute(英国癌症研究中心剑桥研究所)
This study investigates the use of the unsupervised deep learning framework VoxelMorph for deformable registration of longitudinal abdominopelvic CT images acquired in patients with bone metastases from breast cancer. The CT images were refined prior to registration by automatically removing the CT table and all other extra-corporeal components. To improve the learning capabilities of VoxelMorph when only a limited amount of training data is available, a novel incremental training strategy is proposed based on simulated deformations of consecutive CT images. In a 4-fold cross-validation scheme…
College of Physics and Information Engineering, Fuzhou University(福州大学物理与信息工程学院);School of Data Science, Fudan University(复旦大学大数据学院);School of Biomedical Engineering, Shanghai Jiao Tong University(上海交通大学生物医学工程学院)
Multi-modality medical images can provide relevant or complementary information for a target (organ, tumor or tissue). Registering multi-modality images to a common space can fuse these comprehensive information, and bring convenience for clinical application. Recently, neural networks have been widely investigated to boost registration methods. However, it is still challenging to develop a multi-modality registration network due to the lack of robust criteria for network training. In this work, we propose a multi-modality registration network (MMRegNet), which can perform registration between…
Cluster of Excellence – ML for Science(卓越集群——科学机器学习);University of Tübingen(蒂宾根大学);Faculty of Health Sciences and Medicine, University of Lucerne(卢塞恩大学健康科学与医学院);University of Lübeck(吕贝克大学)
Tsinghua University(清华大学);Shenzhen International Graduate School(深圳国际研究生院);Harvard Medical School(哈佛医学院);Brigham and Women’s Hospital(布莱根和妇女医院)
Multimodal image registration (MIR) is a fundamental procedure in many image-guided therapies. Recently, unsupervised learning-based methods have demonstrated promising performance over accuracy and efficiency in deformable image registration. However, the estimated deformation fields of the existing methods fully rely on the to-be-registered image pair. It is difficult for the networks to be aware of the mismatched boundaries, resulting in unsatisfactory organ boundary alignment. In this paper, we propose a novel multimodal registration framework, which leverages the deformation fields estima…
Korea Advanced Institute of Science and Technology(韩国科学技术院);University of Ulsan College of Medicine(蔚山大学医学院);Asan Medical Center(峨山医学中心)
Medical image registration is one of the key processing steps for biomedical image analysis such as cancer diagnosis. Recently, deep learning based supervised and unsupervised image registration methods have been extensively studied due to its excellent performance in spite of ultra-fast computational time compared to the classical approaches. In this paper, we present a novel unsupervised medical image registration method that trains deep neural network for deformable registration of 3D volumes using a cycle-consistency. Thanks to the cycle consistency, the proposed deep neural networks can t…
Shiraz University of Medical Sciences(设拉子医科大学);University of Pittsburgh School of Medicine(匹兹堡大学医学院);UPMC Hillman Cancer Center(UPMC希尔曼癌症中心);Shiraz University of Technology(设拉子理工大学);Mashhad University of Medical Science(马什哈德医科大学)
In medicine, image registration is vital in image-guided interventions and other clinical applications. However, it is a difficult subject to be addressed which by the advent of machine learning, there have been considerable progress in algorithmic performance has recently been achieved for medical image registration in this area. The implementation of deep neural networks provides an opportunity for some medical applications such as conducting image registration in less time with high accuracy, playing a key role in countering tumors during the operation. The current study presents a comprehe…
The Hong Kong University of Science and Technology(香港科技大学)
Registration of pre-operative and post-recurrence brain images is often needed to evaluate the effectiveness of brain gliomas treatment. While recent deep learning-based deformable registration methods have achieved remarkable success with healthy brain images, most of them would be unable to accurately align images with pathologies due to the absent correspondences in the reference image. In this paper, we propose a deep learning-based deformable registration method that jointly estimates regions with absent correspondence and bidirectional deformation fields. A forward-backward consistency c…
Paul C. Lauterbur Research Center for Biomedical Imaging, Shenzhen Institutes of Advanced Technology, Chinese Academy of Sciences(中国科学院深圳先进技术研究院保罗·C·劳特伯生物医学成像研究中心);Pengcheng Laboratory(鹏城实验室);Pazhou Lab(琶洲实验室);Guizhou Provincial People’s Hospital(贵州省人民医院)
Multi-contrast magnetic resonance (MR) image registration is useful in the clinic to achieve fast and accurate imaging-based disease diagnosis and treatment planning. Nevertheless, the efficiency and performance of the existing registration algorithms can still be improved. In this paper, we propose a novel unsupervised learning-based framework to achieve accurate and efficient multi-contrast MR image registrations. Specifically, an end-to-end coarse-to-fine network architecture consisting of affine and deformable transformations is designed to improve the robustness and achieve end-to-end reg…
Korea Advanced Institute of Science and Technology (KAIST)(韩国科学技术院)
Deformable image registration is one of the fundamental tasks in medical imaging. Classical registration algorithms usually require a high computational cost for iterative optimizations. Although deep-learning-based methods have been developed for fast image registration, it is still challenging to obtain realistic continuous deformations from a moving image to a fixed image with less topological folding problem. To address this, here we present a novel diffusion-model-based image registration method, called DiffuseMorph. DiffuseMorph not only generates synthetic deformed images through revers…
Johns Hopkins Medical Institutes(约翰斯·霍普金斯医学研究所);Johns Hopkins University(约翰斯·霍普金斯大学);NVIDIA Corporation(英伟达公司)
In the past, optimization-based registration models have used spatially-varying regularization to account for deformation variations in different image regions. However, deep learning-based registration models have mostly relied on spatially-invariant regularization. Here, we introduce an end-to-end framework that uses neural networks to learn a spatially-varying deformation regularizer directly from data. The hyperparameter of the proposed regularizer is conditioned into the network, enabling easy tuning of the regularization strength. The proposed method is built upon a Transformer-based mod…
Unsupervised Deformable Image Registration with Local-Global Attention and Image Decomposition
Institute of Medical Technology, Peking University Health Science Center(北京大学医学部医学技术研究所);National Institute of Health Data Science, Peking University(北京大学国家健康数据科学研究院);Department of Radiology, Peking University Third Hospital(北京大学第三医院放射科);Department of Pediatrics, Peking University First Hospital(北京大学第一医院儿科);Pediatric Epilepsy Center, Peking University First Hospital(北京大学第一医院儿童癫痫中心);School of Biological Science and Medical Engineering, Southeast University(东南大学生物科学与医学工程学院);Institute for Artificial Intelligence, Peking University(北京大学人工智能研究院)
Institute of Medical Informatics, Universität zu Lübeck(吕贝克大学医学信息学研究所)
Multimodal image registration is a very challenging problem for deep learning approaches. Most current work focuses on either supervised learning that requires labelled training scans and may yield models that bias towards annotated structures or unsupervised approaches that are based on hand-crafted similarity metrics and may therefore not outperform their classical non-trained counterparts. We believe that unsupervised domain adaptation can be beneficial in overcoming the current limitations for multimodal registration, where good metrics are hard to define. Domain adaptation has so far been…
Johns Hopkins University(约翰斯·霍普金斯大学);Johns Hopkins Medical Institutes(约翰斯·霍普金斯医学研究所)
In the last decade, convolutional neural networks (ConvNets) have dominated and achieved state-of-the-art performances in a variety of medical imaging applications. However, the performances of ConvNets are still limited by lacking the understanding of long-range spatial relations in an image. The recently proposed Vision Transformer (ViT) for image classification uses a purely self-attention-based model that learns long-range spatial relations to focus on the relevant parts of an image. Nevertheless, ViT emphasizes the low-resolution features because of the consecutive downsamplings, result i…
Australian Institute for Machine Learning(澳大利亚机器学习研究所);University of Adelaide(阿德莱德大学);South Australian Health and Medical Research Institute(南澳大利亚健康与医学研究所)
Many applications in image-guided surgery and therapy require fast and reliable non-linear, multi-modal image registration. Recently proposed unsupervised deep learning-based registration methods have demonstrated superior performance compared to iterative methods in just a fraction of the time. Most of the learning-based methods have focused on mono-modal image registration. The extension to multi-modal registration depends on the use of an appropriate similarity function, such as the mutual information (MI). We propose guiding the training of a deep learning-based registration method with MI…
Southeast University(东南大学);Tencent Healthcare Co.(腾讯医疗健康有限公司);The Chinese University of Hong Kong(香港中文大学)
Recently, deep-learning-based approaches have been widely studied for deformable image registration task. However, most efforts directly map the composite image representation to spatial transformation through the convolutional neural network, ignoring its limited ability to capture spatial correspondence. On the other hand, Transformer can better characterize the spatial relationship with attention mechanism, its long-range dependency may be harmful to the registration task, where voxels with too large distances are unlikely to be corresponding pairs. In this study, we propose a novel Deforme…
Recon-all-clinical: Cortical surface reconstruction and analysis of heterogeneous clinical brain MRI
Massachusetts General Hospital(麻省总医院);Harvard Medical School(哈佛医学院);Copenhagen University Hospital - Amager and Hvidovre(哥本哈根大学医院阿莫尔与维杜夫院区);Danish Research Centre for Magnetic Resonance(丹麦磁共振研究中心);Centre for Functional and Diagnostic Imaging and Research(功能与诊断成像及研究中心)
Medical Imaging Research Center, UZ Leuven(UZ Leuven医学成像研究中心);Department of Electrical Engineering, ESAT/PSI, KU Leuven(KU Leuven电气工程系(ESAT/PSI));Center for Craniofacial and Dental Genetics, Department of Oral and Craniofacial Sciences, University of Pittsburgh(匹兹堡大学口腔与颅面科学系颅面和牙科遗传学中心);Department of Computing, Imperial College London(帝国理工学院计算机系);IDSIA, USI Lugano(卢加诺大学IDSIA研究所);Twitter(推特公司);Department of Human Genetics, KU Leuven(KU Leuven人类遗传学系)
Registration is an essential tool in image analysis. Deep learning based alternatives have recently become popular, achieving competitive performance at a faster speed. However, many contemporary techniques are limited to volumetric representations, despite increased popularity of 3D surface and shape data in medical image analysis. We propose a one-step registration model for 3D surfaces that internalises a lower dimensional probabilistic deformation model (PDM) using conditional variational autoencoders (CVAE). The deformations are constrained to be diffeomorphic using an exponentiation laye…
Identification of Rare Cortical Folding Patterns using Unsupervised Deep Learning
Université Paris-Saclay(巴黎萨克雷大学);CEA(法国原子能和替代能源委员会);CNRS(法国国家科学研究中心);NeuroSpin(神经自旋研究所);Hôpital Raymond Poincaré(雷蒙·普安卡雷医院);APHP(巴黎公立医院集团);Hôpital Trousseau(特罗索医院);Hôpitaux Universitaires de l’Est Parisien(巴黎东部大学医院);Sorbonne Université(索邦大学)
New York University(纽约大学);NYU Abu Dhabi(纽约大学阿布扎比分校);NYU Tandon School of Engineering(纽约大学坦登工程学院)
Point cloud registration is the process of aligning a pair of point sets via searching for a geometric transformation. Recent works leverage the power of deep learning for registering a pair of point sets. However, unfortunately, deep learning models often require a large number of ground truth labels for training. Moreover, for a pair of source and target point sets, existing deep learning mechanisms require explicitly designed encoders to extract both deep spatial features from unstructured point clouds and their spatial correlation representation, which is further fed to a decoder to regres…
Korea Advanced Institute of Science and Technology(韩国科学技术院);The Catholic University of Korea(韩国加图立大学);University of Ulsan College of Medicine(蔚山大学医学院);Asan Medical Center(峨山医疗中心);Korea Automotive Technology Institute(韩国汽车技术研究院)
Image registration is a fundamental task in medical image analysis. Recently, deep learning based image registration methods have been extensively investigated due to their excellent performance despite the ultra-fast computational time. However, the existing deep learning methods still have limitation in the preservation of original topology during the deformation with registration vector fields. To address this issues, here we present a cycle-consistent deformable image registration. The cycle consistency enhances image registration performance by providing an implicit regularization to pres…
Technical University of Munich (TUM)(慕尼黑工业大学);Klinikum Rechts der Isar(伊萨尔河右岸医院);Imperial College London(伦敦帝国理工学院)
Diffeomorphic deformable multi-modal image registration is a challenging task which aims to bring images acquired by different modalities to the same coordinate space and at the same time to preserve the topology and the invertibility of the transformation. Recent research has focused on leveraging deep learning approaches for this task as these have been shown to achieve competitive registration accuracy while being computationally more efficient than traditional iterative registration methods. In this work, we propose a simple yet effective unsupervised deep learning-based {\em multi-modal} …
UDCR: Unsupervised Aortic DSA/CTA Rigid Registration Using Deep Reinforcement Learning and Overlap Degree Calculation
School of Artificial Intelligence, Beijing University of Posts and Telecommunications(北京邮电大学人工智能学院);Shanghai Changhai Hospital, Naval Medical University(海军军医大学上海长海医院);Dalian University of Technology(大连理工大学);School of Computer Science and Information Security, Guilin University of Electronic Technology(桂林电子科技大学计算机科学与信息安全学院);The Fourth Hospital of Hebei Medical University and Hebei Tumor Hospital(河北医科大学第四医院暨河北省肿瘤医院);Erasmus MC, University Medical Center Rotterdam(鹿特丹伊拉斯姆斯大学医学中心)
To deform or not: treatment-aware longitudinal registration for breast DCE-MRI during neoadjuvant chemotherapy via unsupervised keypoints detection
Radboud University Medical Centre(拉德堡德大学医学中心);The Netherlands Cancer Institute(荷兰癌症研究所);Macao Polytechnic University(澳门理工大学);Maastricht University Medical Centre(马斯特里赫特大学医学中心)
University of Edinburgh(爱丁堡大学);Imperial College London(帝国理工学院);University of Essex(埃塞克斯大学)
Magnetic Resonance Imaging (MRI) typically recruits multiple sequences (defined here as "modalities"). As each modality is designed to offer different anatomical and functional clinical information, there are evident disparities in the imaging content across modalities. Inter- and intra-modality affine and non-rigid image registration is an essential medical image analysis process in clinical imaging, as for example before imaging biomarkers need to be derived and clinically evaluated across different MRI modalities, time phases and slices. Although commonly needed in real clinical scenarios, …
Ryerson University(瑞尔森大学);University of Toronto(多伦多大学);Institute for Biomedical Engineering, Science and Technology (iBEST)(生物医学工程、科学与技术研究所(iBEST))
We propose FlowReg, a deep learning-based framework for unsupervised image registration for neuroimaging applications. The system is composed of two architectures that are trained sequentially: FlowReg-A which affinely corrects for gross differences between moving and fixed volumes in 3D followed by FlowReg-O which performs pixel-wise deformations on a slice-by-slice basis for fine tuning in 2D. The affine network regresses the 3D affine matrix based on a correlation loss function that enforces global similarity. The deformable network operates on 2D image slices based on the optical flow netw…
This study proposes an end-to-end unsupervised diffeomorphic deformable registration framework based on moving mesh parameterization. Using this parameterization, a deformation field can be modeled with its transformation Jacobian determinant and curl of end velocity field. The new model of the deformation field has three important advantages; firstly, it relaxes the need for an explicit regularization term and the corresponding weight in the cost function. The smoothness is implicitly embedded in the solution which results in a physically plausible deformation field. Secondly, it guarantees d…
Johns Hopkins Medical Institutes(约翰·霍普金斯医学院);Johns Hopkins University(约翰·霍普金斯大学);NVIDIA Corporation(英伟达公司);Duke University Medical Center(杜克大学医学中心);Massachusetts General Hospital(麻省总医院);Harvard Medical School(哈佛医学院)
In the last decade, convolutional neural networks (ConvNets) have been a major focus of research in medical image analysis. However, the performances of ConvNets may be limited by a lack of explicit consideration of the long-range spatial relationships in an image. Recently Vision Transformer architectures have been proposed to address the shortcomings of ConvNets and have produced state-of-the-art performances in many medical imaging applications. Transformers may be a strong candidate for image registration because their substantially larger receptive field enables a more precise comprehensi…
Johns Hopkins Medical Institutes(约翰霍普金斯医学研究所);Johns Hopkins University(约翰霍普金斯大学)
Objectives: Computerized phantoms play an essential role in various applications of medical imaging research. Although the existing computerized phantoms can model anatomical variations through organ and phantom scaling, this does not provide a way to fully reproduce anatomical variations seen in humans. However, having a population of phantoms that models the variations in patient anatomy and, in nuclear medicine, uptake realization is essential for comprehensive validation and training. In this work, we present a novel image registration method for creating highly anatomically detailed anthr…
Accuracy validation of cortical thickness measurement is a difficult problem due to the lack of ground truth data. To address this need, many methods have been developed to synthetically induce gray matter (GM) atrophy in an MRI via deformable registration, creating a set of images with known changes in cortical thickness. However, these methods often cause blurring in atrophied regions, and cannot simulate realistic atrophy within deep sulci where cerebrospinal fluid (CSF) is obscured or absent. In this paper, we present a solution using a self-supervised inpainting model to generate CSF in t…
Unsupervised Deformable Image Registration with Structural Nonparametric Smoothing
Cornell University(康奈尔大学);Hunan University(湖南大学);University of Pennsylvania(宾夕法尼亚大学);Johns Hopkins University(约翰霍普金斯大学);Shanghai Jiao Tong University(上海交通大学);University of Birmingham(伯明翰大学);University of Manchester(曼彻斯特大学)
Chengdu University of Information Technology(成都信息工程大学);Keya Medical(科亚医疗);University at Buffalo, SUNY(纽约州立大学布法罗分校)
Large deformations of organs, caused by diverse shapes and nonlinear shape changes, pose a significant challenge for medical image registration. Traditional registration methods need to iteratively optimize an objective function via a specific deformation model along with meticulous parameter tuning, but which have limited capabilities in registering images with large deformations. While deep learning-based methods can learn the complex mapping from input images to their respective deformation field, it is regression-based and is prone to be stuck at local minima, particularly when large defor…
University of Oxford(牛津大学);Institute of Biomedical Engineering(生物医学工程研究所)
Recent successes in deep learning based deformable image registration (DIR) methods have demonstrated that complex deformation can be learnt directly from data while reducing computation time when compared to traditional methods. However, the reliance on fully linear convolutional layers imposes a uniform sampling of pixel/voxel locations which ultimately limits their performance. To address this problem, we propose a novel approach of learning a continuous warp of the source image. Here, the required deformation vector fields are obtained from a concatenated linear and non-linear convolution …