Equivariant symmetry-aware head pose estimation for fetal MRI

TL;DR

E(3)-Pose is a fast, symmetry-aware, equivariant method for fetal head pose estimation in MRI, improving robustness and accuracy over existing approaches.

Contribution

It introduces E(3)-Pose, a novel approach explicitly modeling rotation equivariance and symmetry, enhancing fetal MRI pose estimation.

Findings

Outperforms existing methods in robustness and generalization.

Achieves state-of-the-art accuracy on clinical fetal MRI datasets.

Implementation is publicly available at github.com/MedicalVisionGroup/E3-Pose.

Abstract

We present E(3)-Pose, a novel fast pose estimation method that jointly and explicitly models rotation equivariance and object symmetry. Our work is motivated by the challenging problem of accounting for fetal head motion during a diagnostic MRI scan. We aim to enable automatic adaptive prescription of diagnostic 2D MRI slices with 6-DoF head pose estimation, supported by rapid low-resolution 3D MRI volumes acquired before each 2D slice. Existing pose estimation methods struggle to generalize to clinical volumes due to pose ambiguities induced by inherent anatomical symmetries, as well as low resolution, noise, and artifacts. In contrast, E(3)-Pose captures anatomical symmetries and rigid pose equivariance by construction, and yields robust estimates of the fetal head pose. Our experiments on publicly available and representative clinical fetal MRI datasets demonstrate the superior robustness and generalization of our method across domains. Crucially, E(3)-Pose achieves state-of-the-art accuracy on clinical MRI volumes, supporting future clinical translation. Our implementation is publicly available at github.com/MedicalVisionGroup/E3-Pose.