Image Registration and Predictive Modeling: Learning the Metric on the Space of Diffeomorphisms

Ayagoz Mussabayeva*, Alexey Kroshnin, Anvar Kurmukov, Yulia Denisova, Li Shen, Shan Cong, Lei Wang, Boris A. Gutman

*Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingConference contribution

1 Scopus citations

Abstract

We present a method for metric optimization in the Large Deformation Diffeomorphic Metric Mapping (LDDMM) framework, by treating the induced Riemannian metric on the space of diffeomorphisms as a kernel in a machine learning context. For simplicity, we choose the kernel Fischer Linear Discriminant Analysis (KLDA) as the framework. Optimizing the kernel parameters in an Expectation-Maximization framework, we define model fidelity via the hinge loss of the decision function. The resulting algorithm optimizes the parameters of the LDDMM norm-inducing differential operator as a solution to a group-wise registration and classification problem. In practice, this may lead to a biology-aware registration, focusing its attention on the predictive task at hand such as identifying the effects of disease. We first tested our algorithm on a synthetic dataset, showing that our parameter selection improves registration quality and classification accuracy. We then tested the algorithm on 3D subcortical shapes from the Schizophrenia cohort Schizconnect. Our Schizophrenia-Control predictive model showed significant improvement in ROC AUC compared to baseline parameters.

Original languageEnglish (US)
Title of host publicationShape in Medical Imaging - International Workshop, ShapeMI 2018, Held in Conjunction with MICCAI 2018, Proceedings
EditorsHervé Lombaert, Beatriz Paniagua, Bernhard Egger, Marcel Lüthi, Martin Reuter, Christian Wachinger
PublisherSpringer Verlag
Pages160-168
Number of pages9
ISBN (Print)9783030047467
DOIs
StatePublished - 2018
EventInternational Workshop on Shape in Medical Imaging, ShapeMI 2018 held in conjunction with 21st International Conference on Medical Image Computing and Computer Assisted Intervention, MICCAI 2018 - Granada, Spain
Duration: Sep 20 2018Sep 20 2018

Publication series

NameLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
Volume11167 LNCS
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Other

OtherInternational Workshop on Shape in Medical Imaging, ShapeMI 2018 held in conjunction with 21st International Conference on Medical Image Computing and Computer Assisted Intervention, MICCAI 2018
Country/TerritorySpain
CityGranada
Period9/20/189/20/18

Keywords

  • Expectation Maximization
  • Image registration
  • LDDMM
  • Machine learning
  • Metric learning
  • Subcortical shape

ASJC Scopus subject areas

  • Theoretical Computer Science
  • Computer Science(all)

Fingerprint

Dive into the research topics of 'Image Registration and Predictive Modeling: Learning the Metric on the Space of Diffeomorphisms'. Together they form a unique fingerprint.

Cite this