Abstract
Image segmentation and registration are two of the most challenging tasks in medical imaging. They are closely related because both tasks are often required simultaneously. In this article, we present an improved variational model for a joint segmentation and registration based on active contour without edges and the linear curvature model. The proposed model allows large deformation to occur by solving in this way the difficulties other jointly performed segmentation and registration models have in case of encountering multiple objects into an image or their highly dependence on the initialisation or the need for a pre-registration step, which has an impact on the segmentation results. Through different numerical results, we show that the proposed model gives correct registration results when there are different features inside the object to be segmented or features that have clear boundaries but without fine details in which the old model would not be able to cope.
Keywords
Introduction
Image segmentation aims to separate objects or features in the image that have similar characteristics into different classes or sub-regions, via detection and visualisation of the contours of the objects in the images. Meanwhile, image registration is the process of finding a geometric transformation between images such that the template (target) images are aligned with the reference (source) images. In a wide range of fields, such as medical image processing, pattern recognition, geophysics, comparison of data to a common reference frame, comparison of images taken at different times, shape tracking or similar problems are challenging issues that are encountered. In those cases, image registration and segmentation depend on each other and should be treated simultaneously in a joint framework. One important applications of such a combination can be found in Gooya et al. 1 and similar article where atlases are constructed from magnetic resonance (MR) scans to analyse and understand brain tumour development. The task of construction of the atlases requires alignment of the brain tumour MR scans to a common coordinate system and the automatic segmentation of the scans. According to Erdt et al., 2 25% of published works in medical imaging literature are joint segmentation and registration methods. Many of the methods developed in this context used shape prior models in an energy minimisation framework. The first work on variational model for joint region based segmentation and registration was proposed for rigid registration by Yezzi et al. 3 Later, other publications extended the work on segmentation and rigid registration, see literature.4–8 Those approach however involves a pre-segmentation step, using different criteria for segmentation and rigid registration in a sequence of images, hence is not a joint segmentation and registration approach and will fail for shapes, which non-rigidly deforms in different images. On the other hand, it is worth mentioning approaches developed for the purpose of non-rigid registration.9–11 These techniques globally register images and estimate the deformation field over the whole image and work for non-rigid deformations, such as registration for CT and MR images. These models have difficulties with multiple objects or they do highly depend on the initialisation, which has an impact on the segmentation results. In difference with literature,3,11 Wang and Vemuri 12 propose a registration and segmentation model for multi-modality images using cross cumulative residual entropy as a distance measure for registration. To model the deformation, Wang and Vemuri 12 used the parametric model based on cubic B-spline and for segmentation the piecewise constant Chan–Vese (CV) model. 13 However, the model requires segmentation of the reference image and the work can be considered as registration driven by segmentation.
On the other hand, it is worth mentioning the work of Le Guyader and Vese (GV-JSR), 14 which presents a non-rigid coupled segmentation and registration using the non-linear elastic model to register the segmented template and reference images. The model manages to produce topology-preserving segmentation where the initial contour from the template image is deformed to the contour of the reference image without merging or breaking and allows large deformations to occur. However, the model is limited to the well-defined objects or features that have clear boundaries but without fine details.
The contribution of this article is twofold. First, to improve the GV-JSR model, for cases where the objects are with fine details, by adding a weighted Heaviside sum of the squared difference (SSD) term in the GV-JSR model. Second for a better registration, invariant to the affine registration and which allows large deformation, we use the linear curvature model15,16 to replace the nonlinear elastic term in the GV-JSR model. In this way, there is no need for a pre-registration step to cater for affine linear transformation. 15 Beside the ability to recover affine linear transformation, the linear curvature model for registration also produces more smooth transformation than a nonlinear elastic model. It is well known that low-order regularisation terms, such as nonlinear elasticity are less effective than high-order ones such as linear curvature in producing smooth transformations.17,18 To the best of our knowledge, only diffusion, linear and nonlinear elastic model for non-parametric image registration have been used in the task of joining segmentation and registration.
The outline of this article is as follows: In “Relation to previous work: The GV-JSR model” section, we review the task of joining segmentation and registration. In “The proposed NJSR model” section, we introduce our proposed new joint segmentation and registration (NJSR) model, which improves the original GV-JSR model. We show in “Numerical results” section, some numerical tests including comparisons. Finally, we present our conclusion and future work in the “Conclusion” section.
Relation to previous work: The GV-JSR model
The idea of joining the tasks of segmentation and registration utilised by Le Guyader and Vese 14 using level set representation, which aligns the contour of the template image and simultaneously segment the reference image demonstrate a state of art work with a potential of large deformation of displacement field guided by a segmentation process. The method relates both problems using an active contour based segmentation idea, 13 which is solved in terms of the displacement field. In this section, we provide a brief review of the variational formulation of GV-JSR model for joint segmentation and registration. Before we proceed, we introduce some notation.
Let
The GV-JSR model uses the initial given segmentation of the template image to find the geometric transformation of the template image and the segmentation of the reference image. The segmentation of the template image is represented by the zero level line
The joint functional for segmentation and registration
19
is given by
The variable
The regularisation term in (2), denoted by
The GV-JSR model14,19 is incorporated with the regridding step, thus it manages to recover large deformation. The idea of regridding is proposed by Christensen et al. 23 to model large deformation. The regridding step is as follows. The determinant of the Jacobian matrix of the transformation is calculated during the registration process to make sure there is no folding or cracking in the deformation field. If the minimum value of the determinant falls below a certain threshold, the last displacement field is stored and the template image is initialised using the last displacement field. Then, the displacement field is set to zero and the process is continues until convergence. In Cahill et al., 24 the authors extend the regridding concept and show how the method can be applied in the case of other regularisation terms such as diffusion, linear curvature and linear elastic with several types of boundary conditions. For example, to solve the famous large deformation problem, where we want to align a letter C with a dot (refer to Modersitzki 18 for more details), the model requires two regridding step. So, it is natural to any regularisation based models to recover large deformation as long as the regridding step is incorporated in the model.
One of the main advantages of the GV-JSR model is the ability to produce topology-preserving segmentation where the initial contour from the template image is deformed to the contour of the reference image without merging or breaking. The contour of the reference image is the deformed version of the contour of the template image using the found smooth transformation. It is deformed without separation of the initial contour from the template image, which is difficult to achieve with the standard level set implementation of the active contour. 14 Topology preservation is important for several applications in medical imaging such as in computational brain anatomy. The GV-JSR model manages to preserve the topology of the initial contour without corporation of soft or hard constraint in the model. Based on our experiments, however, we found that the model is only suitable to single object in a well-defined image with relatively large structures. Registration process is only drive by the forces on the boundary of the outer structures of the objects and does produce an incomplete deformation field for the inner structures of the objects.
The proposed NJSR model
Since in GV-JSR model, the registration process is only drive by the forces on the boundary of the outer structures of the objects, it produces an incomplete deformation field for the inner structures of the objects. To deal with the two cases where the GV-JSR model fails to register, we propose to include two new terms in the functional (2). The first term is a SSD term of the form
As
For any given parameter set
The grid points are located at the centre of the cell
and
The discretised form of the functional in (6), by a finite difference method is
Here, we are using homogeneous Neumann boundary conditions where
Starting with zero initial guess,
The NJSR model for joint segmentation and registration.
Initialisation:
For Solve registration problem on this level using Quasi-Newton method (Algorithm 2),
If End for. The NJSR model on one fixed level.
1. For Update Solve equation (15) for Check convergence criterion, if satisfied exit, else continue. End for.
Algorithm 1
where the multilevel of images of the reference and template images denoted by Algorithm 2
Numerical results
We use four sets of images for testing the GV-JSR model and the NJSR model (Algorithm 1) on a variety of images and deformation. To judge the quality of the registration, we calculate the relative reduction of the similarity measure
In all of the experiments, we do not use the regridding step for fair comparison and the value of the regularisation parameters are chosen such that the minimum value of the determinant of the Jacobian matrix Experiment 1 (Comparison between GV-JSR and NJSR Models for One Feature Object) Experiment 1 consists of two X-ray images of a human hand from Modersitzki
17
to illustrate the type of images where the GV-JSR and NJSR models are able to segment and register. The images in Experiment 1 consist of one object with relatively large structure. Experiment 2 (Brain MRI with GV-JSR and NJSR Models) Experiment 2 is used to illustrate that the GV-JSR and NJSR models are capable to solve registration problem using real medical images. We use brain MRI from IBSR
9
(https://www.nitrc.org/project/ibsr) database to test the models. We choose a pair of brain images from different individuals to test the models. Experiment 3 (Global Deformation with GV-JSR and NJSR Models using Synthetic Images) The images for the Experiment 3 come from Hömke
31
where the GV-JSR and NJSR models manage to deliver good results because the features inside the objects in the template image pose the same deformation with the boundary of the object to be segmented. Experiment 4 (Local Deformation with GV-JSR and NJSR Models) Experiment 4 is used to illustrate images where the GV-JSR model fails to provide the deformation field between the reference and template images where the data set is from Henn.
32
In this experiment, the features inside the contour pose different kinds of deformation with the contour. Since the GV-JSR model is based on the boundary mapping, we obtain no alignment for the features inside the contour Γ. Note that the outer structure is nicely registered whereas the inner structure is poorly registered. We show that our proposed model, NJSR, is able to solve the existing problem Experiment 3, which involves different kinds of deformation for the boundary (contour) of the object and the features inside the contour.
In all experiments, we use
Experiment 1: One feature with GV-JSR and NJSR models
Images for Experiment 1 are the same as Modersitzki
17
where X-ray images of two hands of different individuals need to be aligned. The size of the images is 128 × 128 and the recovered transformation is expected to be smooth. For this experiment, we take Experiment 1: GV-JSR model. Illustration of the type of images where the GV-JSR model delivers good results where the object to be segmented in the template image is relatively large. The results obtained in this experiment are for Experiment 1: NJSR model with 

Experiment 2: Brain MRI with GV-JSR and NJSR models
In Experiment 2, we use the images in Figure 3 to illustrate that the proposed model NJSR is capable to solve real medical images. Here, the size of images are 128 × 128. However, the model is applicable for larger size of images using parallel computing (Figure 4).
Experiment 2: GV-JSR model. The results obtained in this experiment are for Experiment 2: NJSR model. We have better results using the NJSR model for Experiment 2. Here, we are using 

Experiment 3: Global deformation with GV-JSR and NJSR models
Synthetic images for Experiment 2 from Hömke
31
are used to illustrate cases where the features inside the object have the same deformation as the boundary of the object. The results of Experiment 3 using the GV-JSR model with Experiment 3: GV-JSR model. Illustration of the second class of problem where the GV-JSR model manages to provide good results where the deformation of the features inside the object to be segmented pose the same deformation with the object itself. (a) 
Our new model, NJSR with Experiment 3: NJSR model with 
Experiment 4: Local deformation with GV-JSR and NJSR models
In Experiment 4, we use the images in Figure 7 to illustrate where the GV-JSR model with Experiment 4: GV-JSR model. Illustration of the type of image, which has different deformation for the boundary Γ and the features inside Γ. The GV-JSR model fails to align the features inside Γ but manages to align the outer most square in the template image. In this experiment, we are using 
We resolve the issues in Experiment 3 by using the NJSR model, and the resulting images are depicted in Figure 8. In this figure, we obtain the segmentation of the reference image as shown in Figure 8(b). Since the NJSR model uses the linear curvature model for registration which contains affine linear transformation, it manages to recover the rotation part of the deformation without affine pre-registration step as shown in Figure 8(a) with Experiment 4: NJSR model. We have better results using the NJSR model for Experiment 3 where the circles in 
Conclusion
We have present an improved model for joint segmentation and registration in a variational formulation. The proposed model consists of two new terms, which extend the original Le Guyader and Vese (GV-JSR) model’s applicability. The first term is a weighted SSD with a regularised Heaviside of the zero level set function to quantify the different deformations exhibited by the features inside of the contour of the template image. The second term is the linear curvature term to control the smoothness of the deformation field, which is superior than the non-linear elastic term in the old GV-JSR model.
Future work involves developing an efficient multigrid method to solve the model, analytical justification for the model, and automatic selection of regularisation parameters. While there has been work in parameter selection in registration, 33 further work is required to develop a method for the selection of optimal parameters for regularisation term in the joint segmentation and registration model. In addition, we can further extend the work for selective segmentation method or shape prior segmentation models.
Footnotes
Declaration of conflicting interests
The author(s) declared no potential conflicts of interest with respect to the research, authorship, and/or publication of this article.
Funding
The author(s) disclosed receipt of the following financial support for the research, authorship, and/or publication of this article: The first author acknowledges the support from the Ministry of Higher Education of Malaysia (KPT(BS)860520465478).
