Joint Learning of Motion Estimation and Segmentation for Cardiac MR Image Sequences

06/11/2018 ∙ by Chen Qin, et al. ∙ 0

Cardiac motion estimation and segmentation play important roles in quantitatively assessing cardiac function and diagnosing cardiovascular diseases. In this paper, we propose a novel deep learning method for joint estimation of motion and segmentation from cardiac MR image sequences. The proposed network consists of two branches: a cardiac motion estimation branch which is built on a novel unsupervised Siamese style recurrent spatial transformer network, and a cardiac segmentation branch that is based on a fully convolutional network. In particular, a joint multi-scale feature encoder is learned by optimizing the segmentation branch and the motion estimation branch simultaneously. This enables the weakly-supervised segmentation by taking advantage of features that are unsupervisedly learned in the motion estimation branch from a large amount of unannotated data. Experimental results using cardiac MRI images from 220 subjects show that the joint learning of both tasks is complementary and the proposed models outperform the competing methods significantly in terms of accuracy and speed.

READ FULL TEXT VIEW PDF
POST COMMENT

Comments

There are no comments yet.

Authors

page 7

This week in AI

Get the week's most popular data science and artificial intelligence research sent straight to your inbox every Saturday.

1 Introduction

Cardiac magnetic resonance imaging (MRI) is one of the reference methods to provide qualitative and quantitative information of the morphology and function of the heart, which can be utilized to assess cardiovascular diseases. Both cardiac MR image segmentation and motion estimation are crucial steps for the dynamic exploration of the cardiac function, which enable the accurate quantification of regional function measures such as changes in ventricular volumes and the elasticity and contractility properties of the myocardium [13]. Traditionally, most approaches consider segmentation and motion estimation as two separate problems. However, these two tasks are known to be closely related [6, 17], and learning meaningful representations for one problem should be helpful to learn representations for the other one.

In this paper, we propose a joint deep learning network for predicting the segmentation and motion estimation simultaneously for cardiac MR sequences. In particular, the proposed architecture consists of two branches: one is an unsupervised Siamese style spatial transformer network for cardiac motion estimation, which exploits multi-scale features and recurrent units to accurately predict sequences of motion fields while ensuring spatio-temporal smoothness; and the other one is a segmentation branch which takes advantage of the joint feature learning to enable weakly-supervised segmentation for temporally sparse annotated data. We formulate the problem as a composite loss function optimized by training both tasks simultaneously. Using experiments with cardiac MRI from 220 subjects, we show that the proposed models can significantly improve the performance.

1.1 Related Work

In recent years, many works in deep learning domain have been proposed for cardiac MR image segmentation. Most of these approaches employ a fully convolutional network which learns useful features by training on manually annotated images and predicts a pixel-wise label map [2, 3, 4, 10]. However, in real world applications, normally only end-distolic (ED) and end-systolic (ES) frames are manually annotated in a sequence of cardiac MR images, while information contained in other frames is not exploited in previous works. On the other hand, traditional methods commonly extended classical optical flow or image registration methods for cardiac motion estimation [7, 13, 14, 16]. For instance, De Craene et al. [7] optimized a 4D velocity field parameterized by B-Spline spatio-temporal kernels to introduce temporal consistency, and Shi et al. [14] combined different MR sequences to estimate myocardial motion using a series of free-form deformations (FFD) [12]. In recent years, some deep learning works [15, 18] have also been proposed for medical image registration. They either trained networks to learn similarity metrics or simulated transformations as ground truth to learn the regression. In contrast, our proposed method is a unified model for learning both cardiac motion estimation and segmentation, where no motion ground truth is required and only temporally sparse annotated frames in a cardiac cycle are needed. Of particular relevance to our approach are works [6, 11]

proposed in computer vision domain. Segflow

[6] used a joint learning framework for natural video object segmentation and optical flow, and the work [11] propagated labels using the estimated flow to enable weakly-supervised segmentation. In comparison, our method proposes a different way to couple both tasks by learning a joint feature encoder, which exploits the massive information contained in unlabeled data and explores the redundancy of the feature representation for both tasks.

2 Methods

Our goal is to realize the simultaneous motion estimation and segmentation for cardiac MR image sequences. Here we construct a unified model consisting of two branches: an unsupervised motion estimation branch based on a Siamese style recurrent multi-scale spatial transformer network, and a segmentation branch based on a fully convolutional neural network, where the two branches share a joint feature encoder. The overall architecture of the model is shown in Fig.

1.

Figure 1: The overall schematic architecture of proposed network for joint estimation of cardiac motion and segmentation. (a) The proposed Siamese style multi-scale recurrent motion estimation branch. (b) The segmentation branch which shares the joint feature encoder with motion estimation branch. The architecture for feature encoder is adopted from VGG-16 net before FC layer. Both branches have the same head architecture as the one proposed in [4], and the concatenation layers of motion estimation branch are from last layers at different scales of the feature encoder.

2.1 Unsupervised Cardiac Motion Estimation

Deep learning methods normally rely heavily on the ground truth labeled data. However, in problems of cardiac motion estimation, dense transformation maps between frames are rarely available. Inspired by the success of spatial transformer network [5, 9, 11] which effectively encodes optical flow to describe motion, here we propose a novel Siamese style multi-scale recurrent network for estimating the cardiac motion of MR image sequences without supervision effort. A schematic illustration of the model is shown in Fig. 1(a).

The task is to find a sequence of consecutive optical flow representations between the target frame and the source frames , where the output is pixel-wise 2D motion fields representing the displacement in and

directions. In order to realize this, the proposed network mainly consists of four components: a Siamese network for the feature extraction of both target frame and source frame; a multi-scale concatenation of features from pairs of frames; a convolutional recurrent unit (RNN) which propagates information along temporal dimension; and a sampler that warps the source frame to the target one by using the estimated displacement field. In details, inspired by the success of cardiac segmentation network proposed in

[4], we determine the Siamese feature encoder as the one in [4] which is adapted from VGG-16 net. For the combination of information from frame pairs, motivated by the traditional multi-level registration method [12], here we propose to concatenate multi-scale features from both streams of Siamese network to exploit information at different scales. This is followed by a convolution and upsampling operation back to the original resolution, and combined using a concatenation layer. In addition, in order to exploit information from consecutive frames and also to ensure the spatio-temporal smoothness of the estimated motion fields, we additionally incorporate a convolutional simple RNN with tanh function at the last layer to propagate motion information along the temporal dimension and to estimate flow with two feature maps corresponding to displacements for the and dimensions, where the network is parameterized by . Finally, the source frames

are transformed using bilinear interpolation to the target frame, which can be expressed as

.

To train the spatial transformer, we optimize the network by minimizing the pixel-wise mean squared error between the transformed frames and the target frame. To ensure local smoothness, we penalize the gradients of flow map by using an approximation of Huber loss proposed in [5], namely and similarly, we use a regularization term to constrain the flow to behave smoothly in temporal dimension, where . Therefore, the loss function can be described as follows:

(1)

where is the number of sequence, and are regularization parameters to trade off between image dissimilarity, local and temporal smoothness.

2.2 Joint Model for Cardiac Motion Estimation and Segmentation

As we know, motion estimation and segmentation tasks are closely related, and previous works in computer vision domain have shown that the learning of one task is able to benefit the other [6, 17]

. Motivated by the success of self-supervised learning which learns features from intrinsic freely available signals

[1, 8], here we propose a joint learning model for cardiac motion estimation and segmentation, where features learned from unsupervised (or self-supervised) motion estimation are exploited for segmentation. By coupling the motion estimation and segmentation network, the proposed approach can be viewed as a weakly-supervised method with temporally sparse annotated data while motion estimation facilitates the feature learning by exploring those unlabeled data. The schematic architecture of the unified model is shown in Fig. 1.

In details, the proposed joint model consists of two branches: the motion estimation branch proposed in Section 2.1, and the segmentation branch based on the effective network proposed in [4]. Here both branches share the joint feature encoder (Siamese style network) as shown in Fig. 1, so that the features learned can better capture the useful related representations for both tasks. Here a categorical cross-entropy loss on labeled data set is used for segmentation branch, in which we define as the input data, as the ground truth, and is the segmentation function parameterized by . In addition, to further exploit the input unlabeled data, we add an additional spatial transformer in segmentation branch, which warps the predicted segmentation to the target frame using the motion fields estimated from motion estimation branch. Similarly, a categorical cross-entropy loss is used between the warped segmentations and the target, where stands for unlabeled data set, and is plus the warp operation. This component mainly works as a regularization for the motion estimation branch, which is supposed to improve the estimation around boundaries.

As a result, a composite loss function consisting of image similarity error, smoothness penalty of motion fields, and pixel-wise cross entropy segmentation losses with the softmax function can be defined as follows:

(2)

where and are trade-off parameters for different tasks. To initialize the joint model, we first train the motion estimation branch using all the available data we have. Then we fix the weights of the shared feature encoder, and train the segmentation branch with the available annotated data. Lastly, we jointly train both branches by minimizing the composite loss function on training set.

3 Experiments and Results

Experiments were performed on 220 short-axis cardiac MR sequences from UK Biobank study. Each scan contains a sequence of 50 frames, where manual segmentations of left-ventricular (LV) cavity, the myocardium (Myo) and the right-ventricular (RV) cavity are available on ED and ES frames. A short-axis image stack typically consists of 10 image slices. For pre-processing, all training images were cropped to the same size of , and intensity was normalized to the range of [0,1]. In our experiments, we split the data into 100/100/20 for training/testing/validation. Parameters used in the loss function were set to be , , and , which were chosen via validation set. The number of image sequence for RNN during training was , and a learning rate of 0.0001 was used. Data augmentation was performed on-the-fly, with random rotation, translation, and scaling.

Evaluation was performed with respect to both segmentation and motion estimation. We first evaluated the segmentation performance of the joint model by comparing it with the baseline method, i.e., training the segmentation branch only (Seg only). Results reported in Table 1 are Dice scores computed with manual annotations on LV, Myo, and RV. It shows that the proposed joint model significantly outperforms the results of Seg only on all three structures with using Wilcoxon signed rank test, especially on Myo where motion normally affects the segmentation accuracy greatly. This indicates the merits of joint feature learning, where features explored by motion estimation are beneficial for segmentation task.

Method LV Myo RV
Seg only 0.9217 (0.0450) 0.8373 (0.0309) 0.8676 (0.0513)
Joint model 0.9348 (0.0408) 0.8640 (0.0295) 0.8861 (0.0453)
Table 1:

Evaluation of segmentation accuracy for the proposed joint model and the baseline (Seg only) method in terms of the Dice Metric (mean and standard deviation).

We also evaluated the performance of motion estimation by comparing the results obtained using a B-spline free-form deformation (FFD) algorithm111https://github.com/BioMedIA/MIRTK [12], network proposed in Section 2.1 (Motion only), and the joint model proposed in Section 2.2. We warped the segmentations of ES frame to ED frame by using the estimated motion fields, and mean contour distance (MCD) and Hausdorff distance (HD) were computed between the transformed segmentations and the segmentations of ED frame. Table 2 shows the comparison results of these methods. It can be observed that both of the proposed methods outperform FFD registration method in terms of MCD and HD on all the three structures () and similarly, the joint model shows better performance than the model trained for motion estimation only ( on LV and RV, and on Myo). Additionally, we compared the test time needed for motion estimation on 50 frames of a single slice in a cardiac cycle, and results indicated a faster speed of proposed methods compared to FFD.

Method MCD HD Time
LV Myo RV LV Myo RV
FFD 1.83(0.53) 2.47(0.74) 3.53(1.25) 5.10(1.28) 6.47(1.69) 12.04(4.85) 13.22s
Motion only 1.55(0.49) 1.23(0.30) 3.14(1.12) 4.20(1.04) 3.51(0.88) 11.72(4.28) 1.23s
Joint model 1.30(0.34) 1.19(0.26) 3.03(1.08) 3.52(0.82) 3.43(0.87) 11.38(4.34) 2.80s
Table 2: Evaluation of motion estimation accuracy for FFD, proposed model in Section 2.1 (Motion only) and the proposed joint model in terms of the mean contour distance (MCD) and Hausdorff distance (HD) in mm (mean and standard deviation). Time reported is testing time on 50 frames in a cardiac cycle per slice.

Furthermore, the proposed joint method is capable of predicting a sequence of estimated motion fields and segmentations simultaneously. Here we show a visualization result of the network predictions with segmentations and motions combined on frames in a cardiac cycle in Fig. 2. Myocardial motion indicated by the yellow arrows were established between ED and other time frames. Note that the network predicts dense motion fields, while for better visualization, we only show a sparse representation around myocardium. To further validate the proposed unified model in terms of the motion estimation, Fig. 3(a)(b) shows a labeling results of the LV and RV boundaries along temporal dimension, which is obtained by warping the labeled segmentations available in ED frame to other time points, and Fig. 3(c) calculated the transformed LV volume over the cardiac cycle. These show that the proposed model is able to produce an accurate estimation, which is also smooth and consistent over time.

Figure 2: Visualization results for simultaneous prediction of motion estimation and segmentation. Myocardial motions are from ED to other time points. Please refer to supplementary material for a dynamic video of a cardiac cycle.
Figure 3: (a) (b)Labeling results obtained by warping the ED frame segmentation to other time points using FFD and the proposed joint model. Results are shown in temporal views of the red short-axis line. (c) Left ventricular volume (ml) of the subject by warping the ED frame segmentation to other time points in a cardiac cycle.

4 Conclusion

In this paper, we have presented a novel deep learning model for joint motion estimation and segmentation of cardiac MR image sequence. The proposed architecture is composed of two branches: a proposed unsupervised Siamese style recurrent spatial transformer network for motion estimation and a segmentation branch based on a fully convolutional network. A joint feature encoder is shared between the two branches, which enables the effective feature learning via multi-task training and also the weakly-supervised segmentation in terms of the temporally sparse annotated data. Experimental results showed significant improvements of proposed models against baseline approaches in terms of accuracy and speed. For the future work, we will validate our method on a larger scale dataset, and will also investigate its usefulness on 3D applications.

References

  • [1] Agrawal, P., Carreira, J., Malik, J.: Learning to see by moving. In: ICCV. pp. 37–45 (2015)
  • [2] Avendi, M., Kheradvar, A., Jafarkhani, H.: A combined deep-learning and deformable-model approach to fully automatic segmentation of the left ventricle in cardiac MRI. Medical image analysis 30, 108–119 (2016)
  • [3]

    Bai, W., Oktay, O., Sinclair, M., et al.: Semi-supervised learning for network-based cardiac MR image segmentation. In: MICCAI. pp. 253–260 (2017)

  • [4] Bai, W., Sinclair, M., Tarroni, G., et al.: Automated cardiovascular magnetic resonance image analysis with fully convolutional networks. Journal of Cardiovascular Magnetic Resonance (2018)
  • [5]

    Caballero, J., Ledig, C., Aitken, A., et al.: Real-time video super-resolution with spatio-temporal networks and motion compensation. In: CVPR (2017)

  • [6] Cheng, J., Tsai, Y.H., Wang, S., Yang, M.H.: Segflow: Joint learning for video object segmentation and optical flow. In: ICCV. pp. 686–695 (2017)
  • [7] De Craene, M., Piella, G., Camara, O., et al.: Temporal diffeomorphic free-form deformation: Application to motion and strain estimation from 3D echocardiography. Medical image analysis 16(2), 427–450 (2012)
  • [8] Doersch, C., Zisserman, A.: Multi-task self-supervised visual learning. In: ICCV (2017)
  • [9] Jaderberg, M., Simonyan, K., Zisserman, A., et al.: Spatial transformer networks. In: NIPS. pp. 2017–2025 (2015)
  • [10] Ngo, T.A., Lu, Z., Carneiro, G.: Combining deep learning and level set for the automated segmentation of the left ventricle of the heart from cardiac cine magnetic resonance. Medical image analysis 35, 159–171 (2017)
  • [11]

    Patraucean, V., Handa, A., Cipolla, R.: Spatio-temporal video autoencoder with differentiable memory. ICLR workshop (2016)

  • [12] Rueckert, D., Sonoda, L.I., Hayes, C., et al.: Nonrigid registration using free-form deformations: application to breast MR images. IEEE transactions on medical imaging 18(8), 712–721 (1999)
  • [13] Shen, D., Sundar, H., Xue, Z., Fan, Y., Litt, H.: Consistent estimation of cardiac motions by 4D image registration. In: MICCAI. pp. 902–910 (2005)
  • [14] Shi, W., Zhuang, X., Wang, H., et al.: A comprehensive cardiac motion estimation framework using both untagged and 3-D tagged MR images based on nonrigid registration. IEEE transactions on medical imaging 31(6), 1263–1275 (2012)
  • [15] Simonovsky, M., Gutiérrez-Becker, B., Mateus, D., et al.: A deep metric for multimodal registration. In: MICCAI. pp. 10–18 (2016)
  • [16] Tobon-Gomez, C., De Craene, M., Mcleod, K., et al.: Benchmarking framework for myocardial tracking and deformation algorithms: An open access database. Medical image analysis 17(6), 632–648 (2013)
  • [17] Tsai, Y.H., Yang, M.H., Black, M.J.: Video segmentation via object flow. In: CVPR. pp. 3899–3908 (2016)
  • [18] Uzunova, H., Wilms, M., Handels, H., et al.: Training CNNs for image registration from few samples with model-based data augmentation. In: MICCAI (2017)