PreprintCited by 119Medical Image Segmentation TechniquesMedical Imaging and AnalysisArtificial Intelligence in Healthcare and Education

Real-time Deep Pose Estimation with Geodesic Loss for Image-to-Template Rigid Registration

Open access
Salehi, Seyed Sadegh Mohseni, Khan, Shadab, Erdogmus, Deniz, Gholipour, Ali·2018·arXiv

Abstract

With an aim to increase the capture range and accelerate the performance of state-of-the-art inter-subject and subject-to-template 3D registration, we propose deep learning-based methods that are trained to find the 3D position of arbitrarily oriented subjects or anatomy based on slices or volumes of medical images. For this, we propose regression CNNs that learn to predict the angle-axis representation of 3D rotations and translations using image features. We use and compare mean square error and geodesic loss to train regression CNNs for 3D pose estimation used in two different scenarios: slice-to-volume registration and volume-to-volume registration. Our results show that in such registration applications that are amendable to learning, the proposed deep learning methods with geodesic loss minimization can achieve accurate results with a wide capture range in real-time (<100ms). We also tested the generalization capability of the trained CNNs on an expanded age range and on images of newborn subjects with similar and different MR image contrasts. We trained our models on T2-weighted fetal brain MRI scans and used them to predict the 3D pose of newborn brains based on T1-weighted MRI scans. We showed that the trained models generalized well for the new domain when we performed image contrast transfer through a conditional generative adversarial network. This indicates that the domain of application of the trained deep regression CNNs can be further expanded to image modalities and contrasts other than those used in training. A combination of our proposed methods with accelerated optimization-based registration algorithms can dramatically enhance the performance of automatic imaging devices and image processing methods of the future.

Integrity signals

Registry

DataCite · 10.48550/arxiv.1803.05982

Cites

44 of 44 references verified0 citation errors · 0 not checkable

Statistics

Recomputing p-values needs the paper’s full text.

Claims the paper makes

We couldn’t retrieve this paper’s full text, so its claims and self-check weren’t run. The registry and citation checks still apply.

Claims other papers make about this one

Each is checked against this paper’s own text: does it say what the citing paper says it does?

Cites

Flagged first, then by citations

Cited by

Sorted by citations

Authors
Salehi, S. S. M., Khan, S., Erdogmus, D., & Gholipour, A.
Affiliations
Boston Children's Hospital (US)
Venue
arXiv
Publisher
arXiv
Published
2018
Volume
38
Issue
2
Pages
470-481
ISSN
0278-0062 (ISSN-L)
OpenAlex
W2805701966
Cited by
118
References
44
Citations by year
2026: 9 · 2025: 17 · 2024: 19 · 2023: 18 · 2022: 21 · 2021: 15 · 2020: 15 · 2019: 2 · 2018: 2
FWCI
4.25
Type
journal article
Language
en
Open access
Locations
  • IEEE Transactions on Medical Imaging (publishedVersion)
  • PubMed (publishedVersion)
  • PubMed Central (acceptedVersion)
  • arXiv (Cornell University) (submittedVersion)
  • arXiv (Cornell University)
Funders
  • McKnight Foundation
  • National Institute of Neurological Disorders and Stroke (R01NS106030)
  • National Institute of Biomedical Imaging and Bioengineering (R01EB018988)
Topics
  • Physical Sciences › Computer Science › Computer Vision and Pattern Recognition › Medical Image Segmentation Techniques
  • Physical Sciences › Engineering › Biomedical Engineering › Medical Imaging and Analysis
  • Health Sciences › Medicine › Health Informatics › Artificial Intelligence in Healthcare and Education
Keywords
fetal brain MRI, slice-to-volume registration, domain generalization, conditional generative adversarial networks, real-time registration
MeSH terms
Deep Learning, Algorithms, Brain, Female, Fetus, Humans, Image Processing, Computer-Assisted, Infant, Newborn, Magnetic Resonance Imaging, Pregnancy, Databases, Factual

About these checks

Registry, retraction, Cites and statistics results are computed, not judged, with one exception: a printed reference that no index lists is read once by a model before it counts as a citation error. Claims are extracted by a model and kept only when their quote appears verbatim in the text; each cited source is then checked, with a model’s help, for whether it says what the claim relies on. Those verdicts can be wrong — read the claim’s quote and the cited source before relying on one.

Statistics and arithmetic are recomputed, not judged.

Confidence bands are shown only for fields calibrated on our annotated test set. None is calibrated yet, so no band is shown.

What counts as a claim ›How we measure confidence ›How we check this ›

Are you an author, or do you see a problem with this page? Request a review