中文

HeartUnloadNet:用于从舒张状态预测未受载心脏几何的弱监督循环一致图网络

计算机视觉与模式识别 2025-07-28 v1 医学物理

摘要

未受载心脏几何(即无腔隙压力的心脏状态)作为零应力、零应变的参考基准,对个性化心脏生物力学模型至关重要,这对于理解健康与疾病生理以及预测心脏干预效果都有重要作用。然而,从临床影像中估计未受载几何仍是一个具有挑战性的任务。传统方法依赖逆向有限元(FE)求解器,需要迭代优化且计算成本高昂。本文引入 HeartUnloadNet,一个深度学习框架,可直接从舒张末期(ED)网格预测未受载左心室(LV)形状,同时显式融合生物物理学先验知识。该网络接受大小任意的网格输入,以及如舒张压、心肌刚性比例和纤维螺旋角等生理参数,输出对应的未受载网格。它采用图注意力架构,并采用循环一致性策略实现双向(加载与卸载)预测,实现部分自监督,提高精度并降低大规模训练数据集的需求。我们在 diverse LV 几何和生理条件下的 20,700 个 FE 模拟中进行训练和测试,HeartUnloadNet 实现亚毫米精度,平均 DSC 为 0.986,HD 为 0.083 cm,推理时间仅为 0.02 秒/例,速度远超传统逆向 FE 求解器超过 10^5 倍且显著更准确。消融实验确认了该架构的有效性。值得注意的是,循环一致设计使得即使仅使用 200 个训练样本,模型也能保持 97% 的 DSC。本工作因此为逆向 FE 求解器提供了一个可扩展且精确的替代方案,支持未来在临床中的实时应用。

关键词

引用

@article{arxiv.2507.18677,
  title  = {HeartUnloadNet: A Weakly-Supervised Cycle-Consistent Graph Network for Predicting Unloaded Cardiac Geometry from Diastolic States},
  author = {Siyu Mu and Wei Xuan Chan and Choon Hwai Yap},
  journal= {arXiv preprint arXiv:2507.18677},
  year   = {2025}
}

备注

Codes are available at https://github.com/SiyuMU/Loaded2UnNet