-
Multi-Node Multi-GPU Diffeomorphic Image Registration for Large-Scale Imaging Problems
Authors:
Malte Brunn,
Naveen Himthani,
George Biros,
Miriam Mehl,
Andreas Mang
Abstract:
We present a Gauss-Newton-Krylov solver for large deformation diffeomorphic image registration. We extend the publicly available CLAIRE library to multi-node multi-graphics processing unit (GPUs) systems and introduce novel algorithmic modifications that significantly improve performance. Our contributions comprise ($i$) a new preconditioner for the reduced-space Gauss-Newton Hessian system, (…
▽ More
We present a Gauss-Newton-Krylov solver for large deformation diffeomorphic image registration. We extend the publicly available CLAIRE library to multi-node multi-graphics processing unit (GPUs) systems and introduce novel algorithmic modifications that significantly improve performance. Our contributions comprise ($i$) a new preconditioner for the reduced-space Gauss-Newton Hessian system, ($ii$) a highly-optimized multi-node multi-GPU implementation exploiting device direct communication for the main computational kernels (interpolation, high-order finite difference operators and Fast-Fourier-Transform), and ($iii$) a comparison with state-of-the-art CPU and GPU implementations. We solve a $256^3$-resolution image registration problem in five seconds on a single NVIDIA Tesla V100, with a performance speedup of 70% compared to the state-of-the-art. In our largest run, we register $2048^3$ resolution images (25 B unknowns; approximately 152$\times$ larger than the largest problem solved in state-of-the-art GPU implementations) on 64 nodes with 256 GPUs on TACC's Longhorn system.
△ Less
Submitted 28 August, 2020;
originally announced August 2020.
-
Fast GPU 3D Diffeomorphic Image Registration
Authors:
Malte Brunn,
Naveen Himthani,
George Biros,
Miriam Mehl,
Andreas Mang
Abstract:
3D image registration is one of the most fundamental and computationally expensive operations in medical image analysis. Here, we present a mixed-precision, Gauss--Newton--Krylov solver for diffeomorphic registration of two images. Our work extends the publicly available CLAIRE library to GPU architectures. Despite the importance of image registration, only a few implementations of large deformati…
▽ More
3D image registration is one of the most fundamental and computationally expensive operations in medical image analysis. Here, we present a mixed-precision, Gauss--Newton--Krylov solver for diffeomorphic registration of two images. Our work extends the publicly available CLAIRE library to GPU architectures. Despite the importance of image registration, only a few implementations of large deformation diffeomorphic registration packages support GPUs. Our contributions are new algorithms to significantly reduce the run time of the two main computational kernels in CLAIRE: calculation of derivatives and scattered-data interpolation. We deploy (i) highly-optimized, mixed-precision GPU-kernels for the evaluation of scattered-data interpolation, (ii) replace Fast-Fourier-Transform (FFT)-based first-order derivatives with optimized 8th-order finite differences, and (iii) compare with state-of-the-art CPU and GPU implementations. As a highlight, we demonstrate that we can register $256^3$ clinical images in less than 6 seconds on a single NVIDIA Tesla V100. This amounts to over 20$\times$ speed-up over the current version of CLAIRE and over 30$\times$ speed-up over existing GPU implementations.
△ Less
Submitted 19 April, 2020;
originally announced April 2020.
-
Quasi-Newton Waveform Iteration for Partitioned Fluid-Structure Interaction
Authors:
Benjamin Rüth,
Benjamin Uekermann,
Miriam Mehl,
Philipp Birken,
Azahar Monge,
Hans-Joachim Bungartz
Abstract:
We present novel coupling schemes for partitioned multi-physics simulation that combine four important aspects for strongly coupled problems: implicit coupling per time step, fast and robust acceleration of the corresponding iterative coupling, support for multi-rate time stepping, and higher-order convergence in time. To achieve this, we combine waveform relaxation -- a known method to achieve hi…
▽ More
We present novel coupling schemes for partitioned multi-physics simulation that combine four important aspects for strongly coupled problems: implicit coupling per time step, fast and robust acceleration of the corresponding iterative coupling, support for multi-rate time stepping, and higher-order convergence in time. To achieve this, we combine waveform relaxation -- a known method to achieve higher order in applications with split time stepping based on continuous representations of coupling variables in time -- with interface quasi-Newton coupling, which has been developed throughout the last decade and is generally accepted as a very robust iterative coupling method even for gluing together black-box simulation codes. We show convergence results (in terms of convergence of the iterative solver and in terms of approximation order in time) for two academic test cases -- a heat transfer scenario and a fluid-structure interaction simulation. We show that we achieve the expected approximation order and that our iterative method is competitive in terms of iteration counts with those designed for simpler first-order-in-time coupling.
△ Less
Submitted 4 February, 2020; v1 submitted 8 January, 2020;
originally announced January 2020.
-
Coupling Brain-Tumor Biophysical Models and Diffeomorphic Image Registration
Authors:
Klaudius Scheufele,
Andreas Mang,
Amir Gholami,
Christos Davatzikos,
George Biros,
Miriam Mehl
Abstract:
We present the SIBIA (Scalable Integrated Biophysics-based Image Analysis) framework for joint image registration and biophysical inversion and we apply it to analyse MR images of glioblastomas (primary brain tumors). In particular, we consider the following problem. Given the segmentation of a normal brain MRI and the segmentation of a cancer patient MRI, we wish to determine tumor growth paramet…
▽ More
We present the SIBIA (Scalable Integrated Biophysics-based Image Analysis) framework for joint image registration and biophysical inversion and we apply it to analyse MR images of glioblastomas (primary brain tumors). In particular, we consider the following problem. Given the segmentation of a normal brain MRI and the segmentation of a cancer patient MRI, we wish to determine tumor growth parameters and a registration map so that if we "grow a tumor" (using our tumor model) in the normal segmented image and then register it to the patient segmented image, then the registration mismatch is as small as possible. We call this "the coupled problem" because it two-way couples the biophysical inversion and registration problems. In the image registration step we solve a large-deformation diffeomorphic registration problem parameterized by an Eulerian velocity field. In the biophysical inversion step we estimate parameters in a reaction-diffusion tumor growth model that is formulated as a partial differential equation.
In this paper, our contributions are the introduction of a PDE-constrained optimization formulation of the coupled problem, the derivation of the optimality conditions, and the derivation of a Picard iterative scheme for the solution of the coupled problem. In addition, we perform several tests to experimentally assess the performance of our method on synthetic and clinical datasets. We demonstrate the convergence of the SIBIA optimization solver in different usage scenarios. We demonstrate that we can accurately solve the coupled problem in three dimensions ($256^3$ resolution) in a few minutes using 11 dual-x86 nodes. Also, we demonstrate that, with our coupled approach, we can successfully register normal MRI to tumor-bearing MRI while obtaining Dice coefficients that match those achieved when registering of normal-to-normal MRI.
△ Less
Submitted 2 March, 2018; v1 submitted 15 October, 2017;
originally announced October 2017.
