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 2D diagnostic MRI slices with 6-DoF head pose estimation, supported by 3D MRI volumes rapidly acquired before each 2D slice. Existing 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, paving the way for clinical translation. Our implementation is available at github.com/ramyamut/E3-Pose.
翻译:我们提出了E(3)-Pose,一种新颖的快速姿态估计方法,该方法联合且显式地建模了旋转等变性与对象对称性。本研究的动机源于在诊断性磁共振成像扫描中应对胎儿头部运动的挑战性问题。我们旨在通过在每个二维切片前快速采集的三维磁共振成像体积的支持下,实现基于六自由度头部姿态估计的二维诊断性磁共振成像切片的自动自适应规划。现有方法由于固有解剖对称性引起的姿态模糊性,以及低分辨率、噪声和伪影等问题,难以泛化至临床体积数据。相比之下,E(3)-Pose通过结构设计捕捉解剖对称性和刚性姿态等变性,从而实现对胎儿头部姿态的鲁棒估计。我们在公开可获取且具有代表性的临床胎儿磁共振成像数据集上进行的实验表明,该方法在跨域场景下具有卓越的鲁棒性和泛化能力。至关重要的是,E(3)-Pose在临床磁共振成像体积数据上达到了最先进的精度,为临床转化铺平了道路。我们的实现代码可在github.com/ramyamut/E3-Pose获取。