image-domain material decomposition using data-driven...

26
Image-Domain Material Decomposition Using Data-Driven Sparsity Models for Dual-Energy CT Zhipeng Li 1 , Saiprasad Ravishankar 2 , Yong Long 1 , Jeffrey A. Fessler 2 1 University of Michigan - Shanghai Jiao Tong University Joint Institute, Shanghai Jiao Tong University, Shanghai, China 2 Department of Electrical Engineering and Computer Science, University of Michigan, MI, USA April 5, 2018

Upload: others

Post on 23-May-2020

2 views

Category:

Documents


0 download

TRANSCRIPT

Page 1: Image-Domain Material Decomposition Using Data-Driven ...web.eecs.umich.edu/~fessler/papers/files/talk/18/li-18-idm.pdf · Image-Domain Material Decomposition Using Data-Driven Sparsity

Image-Domain Material Decomposition UsingData-Driven Sparsity Models for Dual-Energy CT

Zhipeng Li1, Saiprasad Ravishankar2, Yong Long1, Jeffrey A. Fessler2

1University of Michigan - Shanghai Jiao Tong University Joint Institute,Shanghai Jiao Tong University, Shanghai, China

2Department of Electrical Engineering and Computer Science,University of Michigan, MI, USA

April 5, 2018

Page 2: Image-Domain Material Decomposition Using Data-Driven ...web.eecs.umich.edu/~fessler/papers/files/talk/18/li-18-idm.pdf · Image-Domain Material Decomposition Using Data-Driven Sparsity

Outline

1 Introduction

2 Problem Formulation

3 Optimization Algorithm

4 Experiments and Results

5 Conclusions and Future Work

Zhipeng Li (UM-SJTU JI) DECT-ST 2 / 25

Page 3: Image-Domain Material Decomposition Using Data-Driven ...web.eecs.umich.edu/~fessler/papers/files/talk/18/li-18-idm.pdf · Image-Domain Material Decomposition Using Data-Driven Sparsity

Motivation

Dual-Energy CT (DECT)

Enables characterizing concentration of constituent materials inscanned objects, known as material decomposition1

1[Mendonca et al., IEEE T-MI, 2014]Zhipeng Li (UM-SJTU JI) DECT-ST 3 / 25

Page 4: Image-Domain Material Decomposition Using Data-Driven ...web.eecs.umich.edu/~fessler/papers/files/talk/18/li-18-idm.pdf · Image-Domain Material Decomposition Using Data-Driven Sparsity

Image-Domain Decomposition

Image measurements (attenuation maps at high and low energy)are directly available on commercial DECT scanners

Conventional image-domain decomposition

Direct matrix inversion decomposition2

Susceptible to artifacts and noise.

Regularized (model-based) decomposition

Statistical measurement model + Object prior modelImproves image quality and decomposition accuracy

2[Niu et al., MP, 2014]Zhipeng Li (UM-SJTU JI) DECT-ST 4 / 25

Page 5: Image-Domain Material Decomposition Using Data-Driven ...web.eecs.umich.edu/~fessler/papers/files/talk/18/li-18-idm.pdf · Image-Domain Material Decomposition Using Data-Driven Sparsity

Regularization Approches for DECT

Non-adaptive regularizationMaterial-wise Edge-Preserving (EP) 3

Suppress noise while retaining boundary sharpnessUse simple prior models

Learning-based regularizationDictionary Learning

have shown promising results for DECT4

highly non-convex and NP-Hard sparse codingComputation: O(m3N)m is patch size, N is the number of patches

Deep learning for spectral CT5 O(?)Sparsifying Transform (ST) learning

DECT-ST: proposed approachComputation: O(m2N)

3[Xue et al., MP, 2017]4[Li et al., ISBI, 2012]5[Chen & Li, F3D 2017]Zhipeng Li (UM-SJTU JI) DECT-ST 5 / 25

Page 6: Image-Domain Material Decomposition Using Data-Driven ...web.eecs.umich.edu/~fessler/papers/files/talk/18/li-18-idm.pdf · Image-Domain Material Decomposition Using Data-Driven Sparsity

Regularization Approches for DECT

Non-adaptive regularizationMaterial-wise Edge-Preserving (EP) 3

Suppress noise while retaining boundary sharpnessUse simple prior models

Learning-based regularizationDictionary Learning

have shown promising results for DECT4

highly non-convex and NP-Hard sparse codingComputation: O(m3N)m is patch size, N is the number of patches

Deep learning for spectral CT5 O(?)Sparsifying Transform (ST) learning

DECT-ST: proposed approachComputation: O(m2N)

3[Xue et al., MP, 2017]4[Li et al., ISBI, 2012]5[Chen & Li, F3D 2017]Zhipeng Li (UM-SJTU JI) DECT-ST 5 / 25

Page 7: Image-Domain Material Decomposition Using Data-Driven ...web.eecs.umich.edu/~fessler/papers/files/talk/18/li-18-idm.pdf · Image-Domain Material Decomposition Using Data-Driven Sparsity

Material-Wise Sparsifying Transform (ST)

Sparsifying Transform Learning6:A generalized analysis operator learning approachClosed-form solutions for simple thresholding-based sparse coding

6[Ravishankar & Bresler, IEEE T-SP, 2015]Zhipeng Li (UM-SJTU JI) DECT-ST 6 / 25

Page 8: Image-Domain Material Decomposition Using Data-Driven ...web.eecs.umich.edu/~fessler/papers/files/talk/18/li-18-idm.pdf · Image-Domain Material Decomposition Using Data-Driven Sparsity

Outline

1 Introduction

2 Problem Formulation

3 Optimization Algorithm

4 Experiments and Results

5 Conclusions and Future Work

Zhipeng Li (UM-SJTU JI) DECT-ST 7 / 25

Page 9: Image-Domain Material Decomposition Using Data-Driven ...web.eecs.umich.edu/~fessler/papers/files/talk/18/li-18-idm.pdf · Image-Domain Material Decomposition Using Data-Driven Sparsity

Training Sparsifying Transforms

Learn two sparsifying transforms independently, for material l = 1, 2:

argminΩl

minZl

Sparsification error︷ ︸︸ ︷‖ΩlYl − Zl‖2F +

Transform regularizer︷ ︸︸ ︷λ(‖Ωl‖2F − log |det Ωl|

)+

Non-sparse penalty︷ ︸︸ ︷N ′∑i=1

η2‖Zli‖0

Ωl: m×m square transform to be learned for lth material type

Yl: m×N ′ matrix of training patches from lth material images

Zl: m×N ′ matrix of sparse codes of Yl (discard after training)

‖Ωl‖2F − log |det Ωl|: prevents trivial solutions, controls transformcondition number7

Training ST uses an efficient alternating algorithm

7[Ravishankar & Bresler, IEEE T-SP, 2015]Zhipeng Li (UM-SJTU JI) DECT-ST 8 / 25

Page 10: Image-Domain Material Decomposition Using Data-Driven ...web.eecs.umich.edu/~fessler/papers/files/talk/18/li-18-idm.pdf · Image-Domain Material Decomposition Using Data-Driven Sparsity

Proposed DECT-ST Method

Optimization problem:

argminx∈R2Np

minzlj

1

2‖y −Ax‖2W +

2∑l=1

N∑j=1

βl

‖ΩlPljx− zlj‖22 + γ2l ‖zlj‖0

y = (yTH ,y

TL)T ∈ R2Np : attenuation maps at high and low energy

x = (xT1 ,xT2 )T ∈ R2Np : unknown material density images

A = A0 ⊗ INp : matrix of (calibrated) mass attenuation coefficients:

A0 =

(ϕ1H ϕ2H

ϕ1L ϕ2L

).

W = Wj ⊗ INp : weight matrix with Wj = diag(σ2H , σ2L)−1

Pj ∈ Rm×Np : extracts the jth patch of xl as a vector Pjx

zlj ∈ Rm: sparse codes of Pljx

N : number of image patches

Zhipeng Li (UM-SJTU JI) DECT-ST 9 / 25

Page 11: Image-Domain Material Decomposition Using Data-Driven ...web.eecs.umich.edu/~fessler/papers/files/talk/18/li-18-idm.pdf · Image-Domain Material Decomposition Using Data-Driven Sparsity

Outline

1 Introduction

2 Problem Formulation

3 Optimization Algorithm

4 Experiments and Results

5 Conclusions and Future Work

Zhipeng Li (UM-SJTU JI) DECT-ST 10 / 25

Page 12: Image-Domain Material Decomposition Using Data-Driven ...web.eecs.umich.edu/~fessler/papers/files/talk/18/li-18-idm.pdf · Image-Domain Material Decomposition Using Data-Driven Sparsity

Sparse Coding Step

Overall optimization problem:

argminx∈R2Np

minzlj

1

2‖y −Ax‖2W +

2∑l=1

N∑j=1

βl

‖ΩlPljx− zlj‖22 + γ2l ‖zlj‖0

Sparse code update:

zlj = argminzlj

2∑l=1

N∑j=1

βl

‖ΩlPljx− zlj‖22 + γ2l ‖zlj‖0

zlj = Hγl(ΩlPljx)

(1)

Hard-thresholding operator Hγ(b): returns 0 if |b| < γ

Zhipeng Li (UM-SJTU JI) DECT-ST 11 / 25

Page 13: Image-Domain Material Decomposition Using Data-Driven ...web.eecs.umich.edu/~fessler/papers/files/talk/18/li-18-idm.pdf · Image-Domain Material Decomposition Using Data-Driven Sparsity

Image Update Step - Quadratic Majorizer

minx∈R2Np

1

2‖y −Ax‖2W +

R2(x)︷ ︸︸ ︷2∑l=1

N∑j=1

βl ‖ΩlPljx− zlj‖22 (2)

Quadratic majorizer ψM(x,u(i)) at the ith iteration:

ψM(x;u(i)) =1

2‖x− ξ(i)‖2M (3)

where ξ(i) = u(i) −M−1∇R2(u(i)).

Image update:

x(i+1) = argminx

1

2‖y −Ax‖2W + ψM(x;u(i)), (4)

Zhipeng Li (UM-SJTU JI) DECT-ST 12 / 25

Page 14: Image-Domain Material Decomposition Using Data-Driven ...web.eecs.umich.edu/~fessler/papers/files/talk/18/li-18-idm.pdf · Image-Domain Material Decomposition Using Data-Driven Sparsity

Image Update Step - Majorizing Matrix

Design of the diagonal majorizing matrix M:

M ∇2R2(x) = 2

2∑l=1

βl

N∑j=1

P′ljΩl′ΩlPlj . (5)

With patch stride of 1 pixel, the entries of the diagonal matrix∑Nj=1 P′ljPlj corresponding to the lth material are equal to mINp

Diagonal majorizer M :

M =

(2βmλmax(Ω

′1Ω1)INp 0

0 2βmλmax(Ω′2Ω2)INp

).

Zhipeng Li (UM-SJTU JI) DECT-ST 13 / 25

Page 15: Image-Domain Material Decomposition Using Data-Driven ...web.eecs.umich.edu/~fessler/papers/files/talk/18/li-18-idm.pdf · Image-Domain Material Decomposition Using Data-Driven Sparsity

Image Update Step

Pixel-wise update involves one 2× 2 matrix per voxel:

x(i+1)j = argmin

xj

1

2‖yj −A0xj‖2Wj

+1

2‖xj − ξ

(i)j ‖

2Mj, (6)

where Mj ∈ R2×2 is a diagonal weighting matrix for (x1j , x2j)T .

Quadratic majorizer used within FGM (Fast Gradient Method)8

(Instead of usual generic Lipschitz constant)

blockdiagA′0WjA0 + Mj−1 vs1

L

8[Nesterov, Doklady AN USSR, 1983]Zhipeng Li (UM-SJTU JI) DECT-ST 14 / 25

Page 16: Image-Domain Material Decomposition Using Data-Driven ...web.eecs.umich.edu/~fessler/papers/files/talk/18/li-18-idm.pdf · Image-Domain Material Decomposition Using Data-Driven Sparsity

Outline

1 Introduction

2 Problem Formulation

3 Optimization Algorithm

4 Experiments and Results

5 Conclusions and Future Work

Zhipeng Li (UM-SJTU JI) DECT-ST 15 / 25

Page 17: Image-Domain Material Decomposition Using Data-Driven ...web.eecs.umich.edu/~fessler/papers/files/talk/18/li-18-idm.pdf · Image-Domain Material Decomposition Using Data-Driven Sparsity

Fan-beam DECT with XCAT phantom9

TrainingTraining set: patches extracted from five slices of water and boneimages of XCAT phantom, respectively.Patch size 8× 8 and patch stride 1× 1.

Example training image slices for water (left) and bone (right).

9[Segars et al., MP, 2008]Zhipeng Li (UM-SJTU JI) DECT-ST 16 / 25

Page 18: Image-Domain Material Decomposition Using Data-Driven ...web.eecs.umich.edu/~fessler/papers/files/talk/18/li-18-idm.pdf · Image-Domain Material Decomposition Using Data-Driven Sparsity

Learned Transforms

Learned transforms Ωl for water (left) and bone (right).

Transforms (Ω1,Ω2) are initialized with 2D DCT.

Rows of learned transforms shown as 8× 8 patches.

Zhipeng Li (UM-SJTU JI) DECT-ST 17 / 25

Page 19: Image-Domain Material Decomposition Using Data-Driven ...web.eecs.umich.edu/~fessler/papers/files/talk/18/li-18-idm.pdf · Image-Domain Material Decomposition Using Data-Driven Sparsity

DECT Simulation Setup

NCAT phantom sinogram simulation:

Image size: 1024× 1024Poly-energetic source: 80kVp and 140kVpwith 1.86× 105 and 1× 106 incidentphotons per raySinogram size: 888× 984Reconstruct attenuation images via FBP

Reconstruction and decomposition:

Image size: 512× 512Pixel size: 0.98× 0.98 mm2

Optimal parameter combinations to achievethe best image quality and decompositonaccuracy

0

0.1

0.2

0.3

0.4

0.5

0.6

0.7

0.8

High energy atten. image

0

0.1

0.2

0.3

0.4

0.5

0.6

0.7

0.8

Low energy atten. image

Zhipeng Li (UM-SJTU JI) DECT-ST 18 / 25

Page 20: Image-Domain Material Decomposition Using Data-Driven ...web.eecs.umich.edu/~fessler/papers/files/talk/18/li-18-idm.pdf · Image-Domain Material Decomposition Using Data-Driven Sparsity

RMSE Comparison

Table: RMSE of estimated material densities in mg/cm3.

Method Direct Inversion DECT-EP DECT-ST

Water 77.7 39.5 35.1Bone 78.7 53.8 46.2

Direct Inversion: obtain material images directly by matrix inversion

DECT-EP: Hyperbola Edge-Preserving regularizer withδ1 = 0.01 g/cm3 and δ2 = 0.02 g/cm3

DECT-ST further decreases RMSE achieved by DECT-EP

Zhipeng Li (UM-SJTU JI) DECT-ST 19 / 25

Page 21: Image-Domain Material Decomposition Using Data-Driven ...web.eecs.umich.edu/~fessler/papers/files/talk/18/li-18-idm.pdf · Image-Domain Material Decomposition Using Data-Driven Sparsity

Results

0.3

0.7

1.1

1.5

True Direct Inversion DECT-EP DECT-ST

water

0.3

0.7

1.1

1.5

True Direct Inversion DECT-EP DECT-ST

water

0.3

0.7

1.1

1.5

True Direct Inversion DECT-EP DECT-ST

water

0.3

0.7

1.1

1.5

True Direct Inversion DECT-EP DECT-ST

water

0.3

0.7

1.1

1.5

True Direct Inversion DECT-EP DECT-ST

water

0.14

0.19

0.24

bone

0.14

0.19

0.24

bone

0.14

0.19

0.24

bone

0.14

0.19

0.24

bone

0.14

0.19

0.24

bone

Estimated material images of water and bone,display window [0.3 1.7] g/cm3 and [0.14 0.25] g/cm3, respectively.

Zhipeng Li (UM-SJTU JI) DECT-ST 20 / 25

Page 22: Image-Domain Material Decomposition Using Data-Driven ...web.eecs.umich.edu/~fessler/papers/files/talk/18/li-18-idm.pdf · Image-Domain Material Decomposition Using Data-Driven Sparsity

Patient Study

Obtained by Siemens SOMATOM Force CT scanner using DECTimaging protocols

Dual-source at 150 kVp and 80kVp

0.15

0.2

0.25

0.3

150kVp 80kVp

Thigh CT images of a patient. Display window is [0.12 0.32] cm−1.

Zhipeng Li (UM-SJTU JI) DECT-ST 21 / 25

Page 23: Image-Domain Material Decomposition Using Data-Driven ...web.eecs.umich.edu/~fessler/papers/files/talk/18/li-18-idm.pdf · Image-Domain Material Decomposition Using Data-Driven Sparsity

Results

0.5

1

1.5Direct inversion DECT-EP DECT-ST

water

0.5

1

1.5Direct inversion DECT-EP DECT-ST

water

0.5

1

1.5Direct inversion DECT-EP DECT-ST

water

0.5

1

1.5Direct inversion DECT-EP DECT-ST

water

0

0.1

0.2

0.3

bone

0

0.1

0.2

0.3

bone

0

0.1

0.2

0.3

bone

0

0.1

0.2

0.3

bone

Estimated material images of water and bone;display window [0.25 1.5] g/cm3 and [0 0.3] g/cm3, respectively.

Zhipeng Li (UM-SJTU JI) DECT-ST 22 / 25

Page 24: Image-Domain Material Decomposition Using Data-Driven ...web.eecs.umich.edu/~fessler/papers/files/talk/18/li-18-idm.pdf · Image-Domain Material Decomposition Using Data-Driven Sparsity

Outline

1 Introduction

2 Problem Formulation

3 Optimization Algorithm

4 Experiments and Results

5 Conclusions and Future Work

Zhipeng Li (UM-SJTU JI) DECT-ST 23 / 25

Page 25: Image-Domain Material Decomposition Using Data-Driven ...web.eecs.umich.edu/~fessler/papers/files/talk/18/li-18-idm.pdf · Image-Domain Material Decomposition Using Data-Driven Sparsity

ConclusionsWe proposed DECT-ST that combines an image-domain WLS termwith regularizer involving learned sparsifying transforms.DECT-ST outperformed the DECT-EP method (which uses a fixedfinite differencing type sparsifying model) in terms of image qualityand material decomposition accuracy.

Future WorkInvestigate cross-material ST that accounts for correlation betweenmaterial images.Investigate decomposition methods using a more accurate DECTmeasurement model10 with ST-based regularization.

10[Long & Fessler, IEEE T-MI, 2014]Zhipeng Li (UM-SJTU JI) DECT-ST 24 / 25

Page 26: Image-Domain Material Decomposition Using Data-Driven ...web.eecs.umich.edu/~fessler/papers/files/talk/18/li-18-idm.pdf · Image-Domain Material Decomposition Using Data-Driven Sparsity

Thanks for your attention!

Zhipeng Li (UM-SJTU JI) DECT-ST 25 / 25