用形状引导的贝叶斯网络预测心脏运动,更准且带置信度评估。
CardioMorphNet: Cardiac Motion Prediction Using a Shape-Guided Bayesian Recurrent Deep Network
- 基于心脏形状递归注册,避开强度相似性损失
- 在UK Biobank和M&M数据集上优于现有方法
- 输出运动场不确定性图,临床指标更精准
从短轴位心脏磁共振(SAX CMR)图像中准确估计心脏运动对评估心功能和检测异常至关重要。现有方法常因依赖强度相关的图像配准损失而忽略心脏解剖结构。为此,我们提出CardioMorphNet,一种基于3D心脏形状引导的递归贝叶斯深度学习框架,用于可变形配准。该框架采用循环变分自编码器建模心脏周期中的时空依赖性,并引入两个后验模型分别进行双心室分割与运动估计。通过贝叶斯推导出的损失函数,使模型在递归注册分割图时聚焦解剖区域,无需使用强度相似性损失,同时利用序列SAX图像和时空特征。贝叶斯建模还可计算运动场的不确定性图。在UK Biobank和M&M数据集上验证,其配准掩码形状与真值对比表现优异,显著超越当前最优方法。不确定性分析显示,其在心肌区域的预测置信度高于其他概率方法。临床指标提取结果也表明,CardioMorphNet比其他方法更准确。
原文摘要 · Abstract (English)
Accurate cardiac motion estimation from cine cardiac magnetic resonance (CMR) images is vital for assessing cardiac function and detecting its abnormalities. Existing methods often struggle to accurately capture heart motion because they rely on intensity-based image registration similarity losses that may overlook cardiac anatomical regions. To address this, we propose CardioMorphNet, a recurrent Bayesian deep learning framework for 3D cardiac shape-guided deformable registration using short-axis (SAX) CMR images. It employs a recurrent variational autoencoder to model spatio-temporal dependencies across the cardiac cycle, along with two posterior models for bi-ventricular segmentation and motion estimation. The derived loss function from the Bayesian formulation guides the framework to focus on anatomical regions by recursively registering segmentation maps without using intensity-based image registration similarity loss, while leveraging sequential SAX volumes and spatio-temporal features. The Bayesian modelling also enables the computation of uncertainty maps for the estimated motion fields. Validated on the UK Biobank and M&M datasets by comparing warped mask shapes with ground-truth masks, CardioMorphNet demonstrates superior performance in cardiac motion estimation, outperforming state-of-the-art methods. Uncertainty assessment shows that it also yields lower uncertainty values for estimated motion fields in the cardiac region compared with other probabilistic-based cardiac registration methods, indicating higher confidence in its predictions. In addition, the clinical indices extraction assessment shows that CardioMorphNet estimates the clinical indices more accurately than other approaches.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。