sparsity based spectral embedding: application to multi-atlas echocardiography segmentation

20
Sparsity Based Spectral Embedding: Application to Multi-Atlas Echocardiography Segmentation Ozan Oktay, Wenzhe Shi, Jose Caballero, Kevin Keraudren, and Daniel Rueckert Second International Workshop on Sparsity Techniques in Medical Imaging 14 th September 2014 Department of Compu.ng Imperial College London

Upload: kevin-keraudren

Post on 20-Jun-2015

247 views

Category:

Science


1 download

DESCRIPTION

Slides from Ozan Oktay at the MICCAI workshop on Sparsity Techniques in Medical Imaging (STMI2014), presenting one of the methods we used in the CETUS challenge (http://www.creatis.insa-lyon.fr/Challenge/CETUS/index.html).

TRANSCRIPT

Page 1: Sparsity Based Spectral Embedding: Application to Multi-Atlas Echocardiography Segmentation

Sparsity Based Spectral Embedding: Application to Multi-Atlas

Echocardiography Segmentation!

Ozan Oktay, Wenzhe Shi, Jose Caballero, !Kevin Keraudren, and Daniel Rueckert!

!!!!!

Second International Workshop on! Sparsity Techniques in Medical Imaging!

14th September 2014!

Department  of  Compu.ng  Imperial  College  London  

Page 2: Sparsity Based Spectral Embedding: Application to Multi-Atlas Echocardiography Segmentation

2!

Problem Definition and Literature Review!

STMI’14 – September 2014!

Problem:!Left ventricle (LV) endocardium segmentation in 3D Echo

Images !!Motivation:!

Estimation of clinical indices: (1) ejection fraction, !(2) stroke volume, and (3) cardiac motion!!

The existing work in the literature:!1.  B-spline based active surfaces [Barbosa et al., 2013]!

2.  Statistical-shape models [Butakoff et al., 2011]!3.  Edge based level-set segmentation [Rajpoot et al., 2011]!!

RV!

RA!

LV!

LA!

Page 3: Sparsity Based Spectral Embedding: Application to Multi-Atlas Echocardiography Segmentation

3!

Multi-Atlas Image Segmentation!

STMI’14 – September 2014!

§  It uses the manually labeled atlases to segment the target organ.!

§  It does not require any training or prior estimation.!

§  It is more flexible in segmenting images with different left ventricle anatomy.!

§  Successfully applied in !

i.  Brain MRI Segmentation!![Aljabar et al. 2009 NeuroImage]!

!ii.  Cardiac MRI Segmentation !! ![Isgum et al. 2009 TMI]!

Linear and!deformable registration!

Propagate the labels & majority voting!

Atlas-1!

Atlas-2!

Atlas-3,4,5!

Target!image!

Page 4: Sparsity Based Spectral Embedding: Application to Multi-Atlas Echocardiography Segmentation

4!

Proposed Segmentation Framework!

Image Registration!

Segmentation Output!

STMI’14 – September 2014!

Target!Image!

Speckle Reduction!

Structural Representation!

Atlas!Image!

Speckle Reduction!

Structural Representation!

Page 5: Sparsity Based Spectral Embedding: Application to Multi-Atlas Echocardiography Segmentation

5!STMI’14 – September 2014!

1.  Wachinger C. and Navab N.: “Entropy and Laplacian images: Structural representations for multi-modal registration” !

Medical Image Analysis 2012. !

20 40 60 80 100 120

20

40

60

80

100

120

140

160

180

20 40 60 80 100 120

20

40

60

80

100

120

140

160

18020 40 60 80 100 120

20

40

60

80

100

120

140

160

18020 40 60 80 100 120

20

40

60

80

100

120

140

160

180

Patch Based Spectral Representation!

§  Structural image representation!

§  Unsupervised learning of shape and contextual information. !

§  Laplacian Eigenmaps.!

§  Useful for echo images since intensity data does not explicitly reveal the structural information.!

Input image! Patch matrix!

Lower dimensional! embedding!

Manifold!

Structural representation!

Page 6: Sparsity Based Spectral Embedding: Application to Multi-Atlas Echocardiography Segmentation

6!

Dictionary Based Spectral Representation!

STMI’14 – September 2014!

Training!Images!

Dictionary!Learning!

Spectral Embedding of Dictionary Atoms!

What is the main motivation for the sparsity and dictionary learning ?!!§  Computationally efficient due to elimination of redundancy!

§  Large number of images can be mapped to the same embedding space!

Page 7: Sparsity Based Spectral Embedding: Application to Multi-Atlas Echocardiography Segmentation

7!

Dictionary Based Spectral Representation!

STMI’14 – September 2014!

Training!Images!

Dictionary!Learning!

Spectral Embedding of Dictionary Atoms!

Sparse and Local Coding !

Spectral Representation!

Query Image!Patches!

Mapping to the manifold

space!

Page 8: Sparsity Based Spectral Embedding: Application to Multi-Atlas Echocardiography Segmentation

8!

20 40 60 80 100 120 140 160

20

40

60

80

100

120

140

160

Dictionary KSVD

Input Image! Dictionary Learning!

2 4 6 8 10 12

2

4

6

8

10

12

2 4 6 8 10 12

2

4

6

8

10

12

2 4 6 8 10 12

2

4

6

8

10

12

2 4 6 8 10 12

2

4

6

8

10

12

minC,X

kY �CXk2 s.t. 8i , kxik0 T0.

minC,X

kY �CXk2 s.t. 8i , kxik0 T0.

Dictionary KSVD

Dictionary Atoms!

Spectral Embedding!

L = V>⇤V

L = D�W

Laplacian Graph!

Eigen decomposition!

STMI’14 – September 2014!

Dictionary Based Spectral Representation!

Page 9: Sparsity Based Spectral Embedding: Application to Multi-Atlas Echocardiography Segmentation

9!

0.044042 0.97331

Sparse and Local Coding for Spectral Representation!

1.  K. Yu et al. : “Nonlinear learning using local coordinate coding” NIPS 2009!2.  J. Wang et al. : “Locality-constrained linear coding for image classification” CVPR 2010!

Locality constrained linear coding!

minX

NPn=1

kyn �Cx̃nk2 + � kbn � x̃nk2 s.t. 8n ,1>x̃n = 1

b(n,m) = exp ( k(yn � cmk2 /� )

bn = [b(n,1), . . . , b(n,M)]

STMI’14 – September 2014!

20 40 60 80 100 120 140 160

20

40

60

80

100

120

140

160

Input Image!

Query Patch!2 4 6 8 10 12

2

4

6

8

10

12

2 4 6 8 10 12

2

4

6

8

10

12

2 4 6 8 10 12

2

4

6

8

10

12

2 4 6 8 10 12

2

4

6

8

10

12

Atom 1! Atom 2! Atom 3!

=   b1×   +  b2×   +  b3×  

Page 10: Sparsity Based Spectral Embedding: Application to Multi-Atlas Echocardiography Segmentation

10!

20 40 60 80 100 120 140 160

20

40

60

80

100

120

140

160

20 40 60 80 100 120 140 160

20

40

60

80

100

120

140

160

Dictionary Based Spectral Representation!

20 40 60 80 100 120 140 160

20

40

60

80

100

120

140

160

First Spectral Mode!

Second Spectral Mode!

Input Image!

STMI’14 – September 2014!

Page 11: Sparsity Based Spectral Embedding: Application to Multi-Atlas Echocardiography Segmentation

11!

Registration Strategy!

.!

.!

.!

Mode K!

Mode 1!.!.!.!

Mode K!

Mode 1!

Atlas Image! Target Image!

STMI’14 – September 2014!

KX

k=1

kSAk( T (p) )� STk(p)k2 + �R(p)

p 2 R3 , T : R3 7! R3

§  B-spline free-form deformations.[Rueckert et al. TMI 99]!

§  Sum of squared differences similarity

measure.!

§  Number of modes K = 4.!Single deformation field (T )

for all 3D-3D spectral image pairs

T

T

T

(SAK ) (STK )

Page 12: Sparsity Based Spectral Embedding: Application to Multi-Atlas Echocardiography Segmentation

12!

The Proposed Segmentation Framework!

STMI’14 – September 2014!

Training !dataset (atlases)!

Dictionary!learning!

Spectral embedding!

Speckle reduction!

Target!image!

Locality constrained

sparse coding!

Target !image shape

representation!Multi-atlas

segmentation!

Atlas shape representations!

Page 13: Sparsity Based Spectral Embedding: Application to Multi-Atlas Echocardiography Segmentation

13!

Validation Dataset!1.  Training Dataset (Atlases) (15 Patients)!

•  3D+T echo scans.!•  Cross-validation is performed on the training dataset.!•  Ground-truth segmentations are available for ED and ES frames.!

2.  Testing Dataset (15 Patients)!

•  3D+T echo scans, obtained from different view angles.!•  Only the ED and ES frames are segmented!

STMI’14 – September 2014!

Page 14: Sparsity Based Spectral Embedding: Application to Multi-Atlas Echocardiography Segmentation

14!

Other Image Representations!

Original echo image! Local phase image [2]!

Spectral Representation!Boundary Image [1]!

1.  Rajpoot, K. et al.: ISBI 2009 !

2.  Zhuang, X. et al.: ISBI 2010 !

Intensity and phase features!!•  Encodes only the tissue

boundary information.!

•  It is not sufficient for image analysis applications!

Spectral representation!!•  Encodes the contextual

information!!

STMI’14 – September 2014!

Page 15: Sparsity Based Spectral Embedding: Application to Multi-Atlas Echocardiography Segmentation

15!

Surface to Surface Distance Errors!

STMI’14 – September 2014!

0.0 0.2 0.4 0.6 0.8 1.00.0

0.2

0.4

0.6

0.8

1.0

Mes

hS

urfa

ceD

ista

nce

Err

ors

(mm

)

Testing Dataset Training Dataset (Cross-validation)1.5

2.0

2.5

3.0

3.5

4.0

4.5Mean Surface Distance

Unprocessed ImagesSpeckle Reduced Images

Phase Symmetry ImagesLocal Phase Images

Spectral Representation

Testing Dataset Training Dataset (Cross-validation)6

7

8

9

10

11

12

13Maximum Surface Distance

Page 16: Sparsity Based Spectral Embedding: Application to Multi-Atlas Echocardiography Segmentation

16!

Estimation of Clinical Indices!

STMI’14 – September 2014!

0.0 0.2 0.4 0.6 0.8 1.00.0

0.2

0.4

0.6

0.8

1.0

Perc

enta

geA

gree

men

twith

the

Gro

und

Trut

hVa

lues

Testing Dataset Training Dataset (Cross-validation)0.75

0.80

0.85

0.90

0.95

1.00Dice Coefficient

Testing EF Testing SV Training EF Training SV0.5

0.6

0.7

0.8

0.9

1.0Ejection Fraction and Stroke Volume Correlation with Reference Values

Unprocessed ImagesSpeckle Reduced Images

Phase Symmetry ImagesLocal Phase Images

Spectral Representation

Page 17: Sparsity Based Spectral Embedding: Application to Multi-Atlas Echocardiography Segmentation

17!

Qualitative Results!

Segmentation using the proposed spectral representation

Segmentation using local phase image

Ground-truth segmentation

Testing Dataset

Training Dataset

STMI’14 – September 2014!

Page 18: Sparsity Based Spectral Embedding: Application to Multi-Atlas Echocardiography Segmentation

18!

Comparison against the state-of-the-art echocardiographic image segmentation methods !

Table 1: Comparison of the proposed multi-atlas approach (A) against the

state-of-the-art echocardiogaphy segmentation: active surfaces [1] and active

shape model [2]. Estimated ejection fraction (EF ) and end-diastolic volume

(EDV ) are compared against their reference values. The correlation accuracy

is reported in terms of Pearson’s coe�cient (R) and Bland-Altman’s limit of

agreement (BA).

Mean (mm) REF BAEF (µ± 2�) REDV BAEDV (µ± 2�) # of Patients

(A) 2.32±0.78 0.923 -0.74±6.26 0.926 12.88±35.71 15[1] - 0.907 -2.4±23 0.971 -24.60±21.80 24

[2] 1.84±1.86 - 0±19 - 3.06±46.86 10

1.  Barbosa, D., et al.: Fast and fully automatic 3-D echocardiographic segmentation using B-spline explicit active surfaces. Ultrasound in medicine and biology (2013) !

2.  Butakoff, C., et al: Order statistic based cardiac boundary detection in 3D+T echocardiograms. FIMH. Springer (2011) !

Page 19: Sparsity Based Spectral Embedding: Application to Multi-Atlas Echocardiography Segmentation

19!

Accuracy of the Derived Clinical Indices!

Table 2: This table shows the accuracy of the derived clinical indices for thetraining (Patient 1 to 15) and testing datasets (Patient 16 to 30). Pearson’scorrelation coe�cient (PCC) and Bland-Altman’s limit of agreement (µ±1.96�)values are given for the following indices: ejection fraction, stroke volume, end-systolic volume, and end-diastolic volume.

Testing dataset PCC LOA (µ± 1.96�)

ED volume (ml) 0.926 12.81±33.77ES volume (ml) 0.936 -7.77±28.27Ejection fraction (%) 0.923 0.74±7.58Stroke volume (ml) 0.832 -5.05±12.49

Training dataset PCC LOA (µ± 1.96�)

ED volume (ml) 0.983 9.80±45.66ES volume (ml) 0.961 11.21±56.91Ejection fraction (%) 0.787 -1.07±18.52Stroke volume (ml) 0.856 -1.38±27.08

Page 20: Sparsity Based Spectral Embedding: Application to Multi-Atlas Echocardiography Segmentation

20!

Conclusion!§  Summary!•  Multi-atlas approaches can achieve state-of-the-art segmentation

accuracy for echocardiographic images.!•  Spectral representation is an effective ultrasound image feature.!•  Spectral representation can be well approximated using sparse

coding and dictionary learning.!

§  Future work!•  Other application areas!-  Multi-modal image registration using spectral features.!-  Echocardiography strain analysis.!

•  Method improvements!-  Use of gradient information in spectral coordinate mapping.!