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 - Jan 1 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
CountrySpain
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

    Mussabayeva, A., Kroshnin, A., Kurmukov, A., Denisova, Y., Shen, L., Cong, S., Wang, L., & Gutman, B. A. (2018). Image Registration and Predictive Modeling: Learning the Metric on the Space of Diffeomorphisms. In H. Lombaert, B. Paniagua, B. Egger, M. Lüthi, M. Reuter, & C. Wachinger (Eds.), Shape in Medical Imaging - International Workshop, ShapeMI 2018, Held in Conjunction with MICCAI 2018, Proceedings (pp. 160-168). (Lecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics); Vol. 11167 LNCS). Springer Verlag. https://doi.org/10.1007/978-3-030-04747-4_15