CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target...

37
CROSS MODALITY SYNTHESIS (MRI TO CT) Yu Zhao, [email protected]

Transcript of CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target...

Page 1: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are

CROSS MODALITY SYNTHESIS (MRI TO CT)

Yu Zhao, [email protected]

Page 2: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are

Background

¨ Radiotherapy treatment planning:¤ Target delineation, treatment response assessment,

adaptive therapies, etc.

¨ MR images are necessary: superior soft-tissue contrast, etc. Missing: electron density info, provided by CT images è CT-MRI coregistration: 1. misalignment; 2. workflow, patients and costs pressure.

¨ MRI-only simulation workflow: providing both info on tumor volume and location and electron density information.

Page 3: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are

Background

¨ CT-MR registration VS MR-only simulation

Page 4: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are

Background

¨ MRI images: in-phase (IP), out-of-phase (OOP): sequences correspond to paired MRI gradient echo (GRE) sequences obtained with the same repetition time (TR) but with two different echo time (TE)values.

¨ The main application of the IP-OOP sequences is to identify pathological (microsopic) fat content of tissues in the abdomen by showing signal intensities drop on the OOP images compared to the IP images.

Page 5: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are

Background

fat only = in-phase - opposed phase water only = in-phase + opposed phase

Page 6: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are

Methods

¨ 1. Style Transfer ¨ 2. Image Analogy ¨ 3. Deep Learning Nets

Page 7: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are

Style

Content

Synthetic

1. Style Transfer

Page 8: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are

1. Style Transfer

¨ A Neural Algorithm of Artistic StyleLeon A. Gatys, Alexander S. Ecker, Matthias Bethge

v 1). A new way to visualize feature. Use gradient decent to update the image till it generate similar feature response.

v 2). content reconstructionv 3). style reconstruction

Page 9: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are

A schematic of the style loss.

A schematic of the content loss.

1. Style Transfer

Page 10: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are

1. Style Transfer

¨ Feature extraction nets for losses

Page 11: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are

Content features at different level

1. Style Transfer

Feature visualization

Page 12: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are

¨ Content loss

¨ The content loss is the (scaled, squared) Euclidean distance between feature representations of the content and combination images.

¨ draw the content feature from block2_conv2 è structural detail

Page 13: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are

¨ Style loss

¨ Gram matrix¨ The terms of this matrix are proportional to the covariances

of corresponding sets of features, and thus capturesinformation about which features tend to activate together.By only capturing these aggregate statistics across theimage, they are blind to the specific arrangement of objectsinside the image. This is what allows them to captureinformation about style independent of content.

¨ The style loss is then the (scaled, squared) Frobenius norm ofthe difference between the Gram matrices of the style andcombination images.

Page 14: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are

¨ Total loss

¨ Total loss= content loss + style loss + variation loss (a regularization term that encourages spatial smoothness)

Page 15: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are

¨ Regression

¨ Gradient Decent

Page 16: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are

¨ Reproduction results Style Image

Content Images

Page 17: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are

2. Using Image Analogies:

¨ http://www.mrl.nyu.edu/publications/image-analogies/analogies-fullres.pdf

Page 18: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are

3. Deep Learning Nets

U-Net: Convolutional Networks for Biomedical Image Segmentation

Page 19: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are

3. Deep Learning Nets

U-Net: Convolutional Networks for Biomedical Image Segmentation

Page 20: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are

3. Deep Learning Nets

pk(x) ≈ 1 for the k that has the maximum activation ak(x) and pk(x) ≈ 0 for all other k.

U-Net: Convolutional Networks for Biomedical Image Segmentation

Page 21: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are

3. Deep Learning Nets

U-Net: Convolutional Networks for Biomedical Image Segmentation

Page 22: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are

3. Deep Learning Nets

U-Net: Convolutional Networks for Biomedical Image Segmentation

Data augmentation is essential to teach the network the desired invariance and robustness properties, when only few training samples are available. Especially random elastic deformations of the training samples seem to be the key concept to train a segmentation network with very few annotated images. We generate smooth deformations using random displacement vectors on a coarse 3 by 3 grid. The displacements are sampled from a Gaussian distribution with 10 pixels standard deviation.

Page 23: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are

3. Deep Learning Nets

U-Net: Convolutional Networks for Biomedical Image Segmentation

Page 24: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are

3. Deep Learning Nets

V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentation

Page 25: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are

3. Deep Learning Nets

¨ Deep learning for MRI-CT synthesis for radiotherapy¤ perfectly registered CT-MR pairs ¤ Memory issue for training of a standard 3D deep

network

¨ Multi-view Deep Convolutional Neural Networks ¤ Effective cost function for misalignments¤ Multi-view combination for memory efficiency without

losing 3D context

Page 26: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are

3. Deep Learning Nets

1). Unet based structure:

U-Net: Convolutional Networks for Biomedical Image Segmentation U-Net: Our modified version

Page 27: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are

3. Deep Learning Nets

Page 28: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are

3. Deep Learning Nets

Page 29: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are

3. Deep Learning Nets

¨ Misalignment between Ground Truth and Inputs --Max Pooled loss:

¨ 1). Maxpooling before calculating actual loss

Page 30: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are

3. Deep Learning Nets

¨ Misalignment between Ground Truth and Inputs --Max Pooled loss:

A schematic explanation of the impact of mis-registration to intensity transformation. (a) Image1 (Modality 1), (b) A perfectly registered Image2

(Modality 2), (c) Image 2 with rigid mis-alignment, (d) Image 2 with non-rigid mis-alignment. Triangles in (a)-(d) represent the same object. Dashed lines in (c)

and (d) denote the locations of the perfectly registered Image2.

Page 31: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are

3. Deep Learning Nets

¨ Misalignment between Ground Truth and Inputs --Max Pooled loss:

¨ 2). using L1 distance rather than L2 as L1 encourages less blurring (Image-to-Image Translation with Conditional Adversarial Networks, Phillip Isola, et al, 2016)

Page 32: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are

3. Deep Learning Nets

¨ Final loss: Maxpooling Hinge-like Huber loss

! " =

0, " < 0.0112"

*, 0.01 < " < +

+ " − 12+ , -.ℎ012340

Page 33: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are

3. Deep Learning Nets

¨ 3D models (whole volume in)

Page 34: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are

3. Deep Learning Nets 2.5D: 2D at different views è combine

¨ Axial view: good but with serious stitching blurs¨ Sagittal view: overall good¨ Coronal view: very bad

Page 35: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are

3. Deep Learning Nets 2.5D: 2D at different views è combine

¨ Combine:

Page 36: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are

Summary

¨ Multi-channel Multi-view net with hinge-like smL1 loss

Page 37: CROSS MODALITY SYNTHESIS (MRI TO CT) · Background ¨ Radiotherapy treatment planning: ¤Target delineation, treatment response assessment, adaptive therapies, etc. ¨MR images are