Professional Documents
Culture Documents
Unsupervised Learning of Image Correspondences in Medical Imaging Analysis
Unsupervised Learning of Image Correspondences in Medical Imaging Analysis
edu
+ in future FreeSurfer release
Stroke
field 𝜙𝜙
scan 𝒎𝒎 scan 𝒇𝒇
field 𝜙𝜙
warped scan 𝑚𝑚
• significant development
• slow for two images
Learning-based methods
CNN
CNN
…
Fixed image
(𝒇𝒇) network 𝑔𝑔𝜃𝜃
parameters 𝜃𝜃
Supervised:
2
ℒ = 𝜙𝜙 − 𝜙𝜙𝑔𝑔𝑔𝑔
VoxelMorph
deformation
Moving image
(𝒎𝒎)
…
Fixed image
(𝒇𝒇) network 𝑔𝑔𝜃𝜃
parameters 𝜃𝜃
Unsupervised:
ℒ = ‖𝑚𝑚 ∘ 𝜙𝜙 − 𝑓𝑓‖ + 𝜆𝜆 Reg(𝜙𝜙)
images match smooth field
VoxelMorph Loss
deformation
Moving image
(𝒎𝒎)
…
Fixed image
(𝒇𝒇) network 𝑔𝑔𝜃𝜃
parameters 𝜃𝜃
…
Fixed image
(𝒇𝒇) network 𝑔𝑔𝜃𝜃
parameters 𝜃𝜃
ℒ(𝜃𝜃; data) = � ‖𝑚𝑚𝑖𝑖 ∘ 𝑔𝑔𝜃𝜃 (𝑚𝑚𝑖𝑖. 𝑓𝑓𝑖𝑖 ) − 𝑓𝑓𝑖𝑖𝑖𝑖 ‖ + 𝜆𝜆 Reg(𝑔𝑔𝜃𝜃 (𝑚𝑚𝑖𝑖. 𝑓𝑓𝑖𝑖 ))
𝑖𝑖,𝑗𝑗
𝜙𝜙𝑖𝑖𝑖𝑖 𝜙𝜙𝑖𝑖𝑖𝑖
VoxelMorph Loss
deformation
Moving image
Moved (𝒎𝒎 ∘ 𝝓𝝓𝑧𝑧 )
(𝒎𝒎)
…
spatial
Fixed image transform
(𝒇𝒇) network 𝑔𝑔𝜃𝜃
parameters 𝜃𝜃 moved image 𝑚𝑚 ∘ 𝜙𝜙
should be close to 𝑓𝑓
ℒ(𝜃𝜃; data) = � ‖𝑚𝑚𝑖𝑖 ∘ 𝑔𝑔𝜃𝜃 (𝑚𝑚𝑖𝑖. 𝑓𝑓𝑖𝑖 ) − 𝑓𝑓𝑖𝑖𝑖𝑖 ‖ + 𝜆𝜆 Reg(𝑔𝑔𝜃𝜃 (𝑚𝑚𝑖𝑖. 𝑓𝑓𝑖𝑖 ))
𝑖𝑖,𝑗𝑗
𝜙𝜙𝑖𝑖𝑖𝑖 𝜙𝜙𝑖𝑖𝑖𝑖
Training
deformation
Moving image
Moved (𝒎𝒎 ∘ 𝝓𝝓𝑧𝑧 )
(𝒎𝒎)
…
spatial
Fixed image transform
(𝒇𝒇) network 𝑔𝑔𝜃𝜃
parameters 𝜃𝜃 moved image 𝑚𝑚 ∘ 𝜙𝜙
should be close to 𝑓𝑓
• SGD based techniques
• Each image pair contributes slightly to 𝜃𝜃
Classical optimization: slightly update 𝜙𝜙 for an image pair
Registration
deformation
Moving image
(𝒎𝒎)
…
Fixed image
(𝒇𝒇) network 𝑔𝑔𝜃𝜃
parameters 𝜃𝜃
Probabilistic model
𝑚𝑚 ∘ 𝜙𝜙𝑧𝑧 + 𝜖𝜖 = 𝑓𝑓
𝑧𝑧 ~ 𝒩𝒩 𝑧𝑧; 0, Λ−1
stationary velocity field smoothness via Laplacian
0.9
ANTs (Syn)
0.85 Test Set
Test Set (Instance-Specific Opt.)
0.8
Dice score
0.75
0.7
0.65
0.6
10 25 50 100
Training set size
Segmentation Maps available at training
Moving 3D Image (𝒎𝒎)
deformation
field (𝝓𝝓𝑧𝑧 )
Moving Image
Segmentations
(𝒔𝒔𝒎𝒎 ) Move Image
spatial
transform Segmentations
(𝒔𝒔𝒎𝒎 ∘ 𝝓𝝓𝑧𝑧 )
Fixed Image
Segmentations
(𝒔𝒔𝒇𝒇 )
Dice Loss (ℒ)
Test time performance
0.9
0.85
0.8
Dice score
0.75
0.7
0.65
0.6
VoxelMorph VoxelMorph w/ seg
SynthMorph
(do we need real data?)
Hoffmann et al in submission
Inputlabels Deformed labels GMM Sampling Blurred image Finalimage
Modality
agnostic
CNN ℎ𝜃𝜃 (𝑚𝑚, 𝑓𝑓)
Fixed
image 𝑓𝑓
Moving Moving
labels 𝑠𝑠𝑚𝑚 image 𝑚𝑚
Label
Image sampling Modality
sampling
model agnostic
model
𝑔𝑔𝐼𝐼 (𝑠𝑠𝑚𝑚 , 𝑠𝑠𝑓𝑓 , 𝑧𝑧) CNN ℎ𝜃𝜃 (𝑚𝑚, 𝑓𝑓)
𝑔𝑔𝑠𝑠 (𝑧𝑧)
Fixed Fixed
labels 𝑠𝑠𝑓𝑓 image 𝑓𝑓
Warped
Moving Moving Deformation 𝜙𝜙𝑣𝑣 segmentation
labels 𝑠𝑠𝑚𝑚 image 𝑚𝑚 via 𝑣𝑣 integration 𝑠𝑠𝑚𝑚 ∘ 𝜙𝜙𝑣𝑣
Label
Image sampling Modality
sampling
model agnostic
model
𝑔𝑔𝐼𝐼 (𝑠𝑠𝑚𝑚 , 𝑠𝑠𝑓𝑓 , 𝑧𝑧) CNN ℎ𝜃𝜃 (𝑚𝑚, 𝑓𝑓)
𝑔𝑔𝑠𝑠 (𝑧𝑧)
Fixed Fixed
labels 𝑠𝑠𝑓𝑓 image 𝑓𝑓 Regularization term ℒ 𝑟𝑟𝑟𝑟𝑟𝑟
Dice term ℒ𝑑𝑑𝑑𝑑𝑑𝑑
Model Using λ1
Model Using λ2
Model Using λ3 λ
Train Multiple Models Validate Each Choose Optimal Train Single Efficiently Optimize
Model Hyperparameter λ Model Hyperparameter λ Using
Validation Data
Model Using λ1
deformation
velocity field (𝝓𝝓𝑖𝑖 ) Moved (𝒕𝒕 ∘ 𝝓𝝓𝑖𝑖 )
field
Spatial
𝑔𝑔𝜽𝜽𝑣𝑣 𝒗𝒗𝑖𝑖 Transform
Image (𝒙𝒙𝑖𝑖 )
integration
layer
Loss (ℒ)
Hyperparameter
Input 𝜆𝜆
HyperMorph
Learned template (𝒕𝒕)
deformation
velocity field (𝝓𝝓𝑖𝑖 ) Moved (𝒕𝒕 ∘ 𝝓𝝓𝑖𝑖 )
field
Spatial
𝑔𝑔𝜽𝜽𝑣𝑣 𝒗𝒗𝑖𝑖 Transform
Image (𝒙𝒙𝑖𝑖 )
integration
θv layer
Loss (ℒ)
Hypernetwork (h)
Hyperparameter
Input 𝜆𝜆
Baseline Comparison
Dice
Runtime (GPU-hours)
VoxelMorph (~10 models): 765
HyperMorph: 147
Optimal Hyperparameters vary by dataset
Optimal Hyperparameters vary by task
… even by anatomical region!
Template construction
Dalca, Rakic, Guttag, Sabuncu, NeurIPS 2019
Joshi et al, 2004
Template Construction
Learned template (𝒕𝒕)
param:
deformation
Ψatl
velocity field (𝝓𝝓𝑖𝑖 ) Moved (𝒕𝒕 ∘ 𝝓𝝓𝑖𝑖 )
field
Spatial
𝑔𝑔𝑣𝑣,𝜽𝜽𝑣𝑣 𝒙𝒙𝑖𝑖 , 𝒕𝒕 𝒗𝒗𝑖𝑖 Transform
Image (𝒙𝒙𝑖𝑖 )
integration
layer
Loss (ℒ)
Conditional template construction
Learned template (𝒕𝒕)
Attribute (𝒂𝒂𝑖𝑖 )
deformation
velocity field (𝝓𝝓𝑖𝑖 ) Moved (𝒕𝒕 ∘ 𝝓𝝓𝑖𝑖 )
field
Spatial
𝑔𝑔𝑣𝑣,𝜽𝜽𝑣𝑣 𝒙𝒙𝑖𝑖 , 𝒕𝒕 𝒗𝒗𝑖𝑖 Transform
Image (𝒙𝒙𝑖𝑖 )
integration
layer
Loss (ℒ)
15 90
29 weeks 32 35 38 41 44
voxelmorph.mit.edu
vo elmorph
• Probabilistic generative model • Very fast for new image pair
for diffeomorphisms • State-of-the-art accuracy
• Variational Inference • Diffeomorphic deformations
• Unsupervised Neural Network • Uncertainty estimation
voxelmorph.mit.edu
voxelmorph.mit.edu