inter-subject image registration

37
mert [email protected] talk: bwh 05.16.07 1 Inter-subject Image Registration Mert Rory Sabuncu Computer Science and Artificial Intelligence Lab MIT

Upload: ngodat

Post on 13-Feb-2017

223 views

Category:

Documents


0 download

TRANSCRIPT

Page 1: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 1

Inter-subject Image Registration

Mert Rory SabuncuComputer Science and Artificial Intelligence LabMIT

Page 2: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 2

Image Registration

n Image: 2,3,4 D function of spatial variablesn Registration: Spatial alignmentn Goal: Point-wise correspondencen Modality: MRI, fMRI, DTI, US, CT, X-ray

Images from Ben Singer (CSBMB, Princeton) and Lauren O’Donnell (BWH, Harvard)

Page 3: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 3

Inter-subject Image Registration

n Goal: Compare and/or fuse information from multiple subjects

n Scenes are slightly differentn Sophisticated warp models (Nonlinear)

Rigid Affine B-splines Dense

n Applications: surgery planning, disease monitoring, brain mapping, morphometry

Page 4: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 4

Typical Registration Algorithm

n Warp space: e.g. rotate around image centern Alignment measure: e.g. mean absolute differencen Optimization: e.g. exhaustive search

Fixed Image Floating Image Absolute Difference

Page 5: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 5

Toy example: apple and orange

Original Picture Flipped Picture

Page 6: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 6

Apple and orange registered

Original Picture Flipped Picture

“Level-set Entropy as a fast and simple alignment measure,” Sabuncu and Ramadge, submitted to ICIP‘07

Page 7: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 7

You can’t compare apples and oranges

n Correspondence depends on contextn Typical alignment measures use local image

features, e.g. pixel intensitiesn Warp models should be motivated by applicationn Warps are typically very (!) high dimensionaln Optimizer usually guarantees local convergencen Registration is a tool n Keep in mind the real question !

Page 8: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 8

Three Applications

n Functional registration of the cerebral cortexn DTI-tract based inter-subject registrationn Atlas-based parcellation of the cerebral

cortex

n Current and future work

Page 9: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 9

Functional Registration of the Human Cerebral Cortexn Cerebral Cortex: 2-4 mm

thick sheet of tissue - outer portion of the brain.

n Convoluted: gyri and sulcin Responsible for many brain

functions inc. memory, attention, perceptual awareness, etc. http://www.toosmarttostart.samhsa.gov/Interactive

Body/html/cerebral.htm

“Function-based inter-subject alignment of cortical anatomy,” M Sabuncu, Singer B, Bryan R, Ramadge P and Haxby J

Page 10: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 10

MRI

n Structural MRI: Displays tissues, e.g. gray, white matter and CSF.

n Functional MRI: Displays neural activity. Typically of 3mm spatial resolution and 2-4 sec. temporal resolution.

Images from http://en.wikipedia.org/wiki/MRI

Page 11: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 11

Motivation

n A method to determine inter-subject functional correspondence based on MRI.n The “Movie” experiment (Hasson et al. Science 2004)

Inter-subject correlations

Anatomical correspondence Functional correspondence

Page 12: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 12

Method: Pre-processing

Tools from AFNI and FreeSurfer to:n Extract and triangulate inner and outer boundaries

of the cortex based on sMRI.n Inflate to smooth these surfaces.n Universal representation on a sphere. n Non-linear anatomical registration on sphere.n Uniform re-sampling of mesh.

Page 13: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 13

Method: Correlation as an Alignment Measuren With fMRI, at each mesh node we have long

time-series, Y(x,t).n The inter-subject correlation between these

time-series measures linear dependency.n GLM: Y = X β + εn So, we’re assuming: functionally equivalent

point: β1 = a β2

Page 14: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 14

Method: Regularizing the warp

n W/o regularization, risk of overfitting.

n Mildest form of regularization: invertibility.

Page 15: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 15

Implementation

n Iterative gradient-ascent to maximize a functional:Total Correlation – λ Warp Penalty

n To initialize: q Raw warp field from exhaustive search within a 3

cm radius from anatomical correspondence. Use nodes with a correlation higher than a threshold (0.3).

q Smooth raw field w/ a Gaussian.

Page 16: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 16

Empirical Results

n Functionally register based on first half of movie experiment. Test on second half of movie.

Movies Second Half Left Hemisphere

0

500

1000

1500

2000

2500

3000

3500

4000

4500

5000

5500

6000

6500

-0.10

-0.07

-0.04

-0.01

0.02

0.05

0.08

0.11

0.14

0.17

0.20

0.23

0.26

0.29

0.32

0.35

0.38

Correlation

Movies Second Half Right Hemisphere

Correlation

AnatFunc

Page 17: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 17

Empirical Results (cont’d)

Correlations on the brain

r.5

.3

.1

Page 18: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 18

Generalization to a Visual Category Experiment

GLM Results on 7 subjects

t+10

0

-10

Faces

Objects

Anatomical Correspondence Functional Correspondence

Page 19: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 19

Discussion

n Promising results that support the plausibility to perform functional registration based on fMRI

n Further experimentation required for validation

n Other approaches should be testedn The effect of functional normalization should

be tested in different analyses, e.g. MVPA

Page 20: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 20

DTI-tract based Registration

n DTI measures water diffusionn Tractography aims at reconstructing white matter tracts

q Our resolution is too low to identify individual axonsq Tractography bundles similar to white matter tracts

n Using atlas-based “clustering” we can identify these bundles (O’Donnell, MIT PhD Thesis, 2006)

Page 21: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 21

DTI-tract based Registration

n Goal: Perform group studies on DTI-tracts.n Requires correspondence between tracts.n Tract-based registration is not well-studied.q Leemans et al. 2006

n Problem: Too many tracts. Too noisy data.

Page 22: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 22

Our Approach

n We define the spatial distribution of bundles:

n Discretize to a voxel-based image – can use existing methods to register medical data

n Find affine registration for each bundle, e.g. by maximizing correlation

“Fiber bundle-based nonlinear registration of diffusion MR images,” U Ziyan, Sabuncu M, O’Donnell L, Westin C-F, submitted.

Red: Corona Radiata

Green: Corpus Collosum

Blue: Arcuate

Page 23: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 23

Polyaffine Framework(Arsigny et al. 2006)

n We have a set of affine parameters A(x) along with a spatial distribution of each structure. How do we fuse them into one well behaved warp field?

n Obvious Solution: A weighted summation of the affine transformations:

n A weighted summation of the speed vectors lead into a well behaved transformation:

Page 24: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 24

Tensor Rotationn The Diffusion Tensor (D) Field needs to be reoriented after a warp:

q D’ = RTDR, where R is the rotational component of the warp.q The rotational component is usually not readily available for a non-linear transformationq R can be estimated from the Jacobian

n We showed: the Jacobian of the polyaffine can be calculated analytically

Page 25: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 25

Quantification (FiT measure)n We want to quantify how well the transformed tracts fit to the DTI data of the

other subjectq Or, equivalently the fit of transformed DTI data to the tracts of the other subject

n ODF is defined as:

n We define FiT:

Page 26: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 26

Results

Aff: Global affine registration on FA volumes

Dem: The “Demons Algorithm” on FA volumes (Park et al. NeuroImage 2003)

PA: Proposed Algorithm

Page 27: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 27

Results: FiT values

Page 28: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 28

Parcellation of the Cerebral Cortex

“Effects of Registration Regularization and Atlas Sharpness on Segmentation Accuracy,” T Yeo, Sabuncu M, Fischl B, and Golland P. submitted

Page 29: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 29

Approach

n Build an atlas from manually labeled data sets.q Jointly register the images. How? Using labels?

Image features? Or both?q How much warp should be allowed? A sharp atlas

or a blurry atlas?

n Register a new image to the atlas AND infer the labels.

Page 30: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 30

Joint registration and segmentation

n Atlas-based segmentation quality is determined by registration quality.

n Registration quality can be improved if we had the labels.

n So, do registration and segmentation together. (iteratively, using EM)(Extended the ideas of Killian et al. 2006)

Page 31: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 31

An EM Algorithm

New (unlabeled) image

Fixed Atlas: Conditional densities + label frequencies + label n’borhood frequencies

Estimate Labels•A HMRF model

•Mean Field Solution to get “fuzzy labels”

Register•Employ fuzzy labels

•Maximize the image likelihood

Page 32: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 32

Building an atlas

Manually labeled images

“Rigidly” co-register (FreeSurfer)

Compute “rigid” atlas: •Count labels at each location

•Count label neighborhood instances

•Compute conditional means and variances

Manually labeled images

RegisterAtlas

•Employ manual labels

•Maximize the image likelihood

•Initialize Atlas from previous iteration•Decrease Warp constraint

Update Atlas

N Iterations

Page 33: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 33

Stack of Atlases (Multi-scale Atlas)

Fuzzy Atlas:

•Manual labels are not lined up perfectly

•Conditional distributions are fat

Sharp Atlas:•Manual labels are lined up.•Conditional distributions are tight

Built with a highly constrained warp Built with a highly non-linear warp

Page 34: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 34

Three Implementations

n SASS: Single Atlas scale, Single warp Scalen SAMS: Single Atlas scale, multiple warp

Scalesn MAMS: Multiple Atlas scales, Multiple warp

Scales

Page 35: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 35

ResultsQuality of Segmentation vs. Function of Warp Scale

Page 36: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 36

Conclusion

n Joint registration segmentation improves segmentation accuracy.

n In this context, optimal MAMS, SAMS and SASS have similar performance

n Picking the optimal warp scale is important

Page 37: Inter-subject Image Registration

mert [email protected] talk: bwh 05.16.07 37

Other Work

n Joint registration and segmentation in the volumeq Warp definition using poly-affine on structure-

specific affinesq PCA on the warp spaceq Morphometry on the warp space

n Extend the idea of multiple atlasesn Functional Registration