用深度学习快速预测心脏无压状态,精度超传统方法百倍。
HeartUnloadNet: A Weakly-Supervised Cycle-Consistent Graph Network for Predicting Unloaded Cardiac Geometry from Diastolic States
- 基于图注意力网络,结合生物物理先验与循环一致性设计
- 亚毫米级精度,单次推理仅需0.02秒,比传统方法快10万倍
- 仅需200组数据即可保持97%精度,适合临床实时应用
无压心脏几何(即无腔内压力状态)是个性化心肌力学建模的重要零应力参考,对理解健康与疾病生理及预测心脏干预效果至关重要。然而,从临床影像中估计该几何仍具挑战性。传统方法依赖反向有限元求解器,需迭代优化且计算成本高。本文提出HeartUnloadNet,一种弱监督的循环一致图网络,可直接从舒张末期(ED)网格预测无压左室(LV)形状,显式融合生物物理先验。网络接受任意尺寸的网格及生理参数(如ED压力、心肌刚度尺度、纤维螺旋角),输出对应无压网格。采用图注意力架构并引入循环一致性策略,实现加载/卸载双向预测,支持部分自监督,提升精度并减少数据需求。在20,700个不同左室几何与生理条件的有限元仿真上训练测试,平均骰率(DSC)达0.986,豪斯多夫距离(HD)为0.083厘米,推理时间仅0.02秒/例,较传统方法快超过10⁵倍且更准确。消融实验验证架构有效性。值得注意的是,循环一致性设计使模型在仅200个训练样本下仍保持97% DSC。本工作为反向有限元求解器提供了一种可扩展、高精度的替代方案,助力未来实时临床应用。
原文摘要 · Abstract (English)
The unloaded cardiac geometry (i.e., the state of the heart devoid of luminal pressure) serves as a valuable zero-stress and zero-strain reference and is critical for personalized biomechanical modeling of cardiac function, to understand both healthy and diseased physiology and to predict the effects of cardiac interventions. However, estimating the unloaded geometry from clinical images remains a challenging task. Traditional approaches rely on inverse finite element (FE) solvers that require iterative optimization and are computationally expensive. In this work, we introduce HeartUnloadNet, a deep learning framework that predicts the unloaded left ventricular (LV) shape directly from the end diastolic (ED) mesh while explicitly incorporating biophysical priors. The network accepts a mesh of arbitrary size along with physiological parameters such as ED pressure, myocardial stiffness scale, and fiber helix orientation, and outputs the corresponding unloaded mesh. It adopts a graph attention architecture and employs a cycle-consistency strategy to enable bidirectional (loading and unloading) prediction, allowing for partial self-supervision that improves accuracy and reduces the need for large training datasets. Trained and tested on 20,700 FE simulations across diverse LV geometries and physiological conditions, HeartUnloadNet achieves sub-millimeter accuracy, with an average DSC of 0.986 and HD of 0.083 cm, while reducing inference time to just 0.02 seconds per case, over 10^5 times faster and significantly more accurate than traditional inverse FE solvers. Ablation studies confirm the effectiveness of the architecture. Notably, the cycle-consistent design enables the model to maintain a DSC of 97% even with as few as 200 training samples. This work thus presents a scalable and accurate surrogate for inverse FE solvers, supporting real-time clinical applications in the future.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。