用主动学习减少训练血流模型所需的仿真次数,提升效率与鲁棒性。
Active Learning for Deep Learning-Based Hemodynamic Parameter Estimation
- 基于几何差异、集成不确定性与物理一致性设计三种查询策略
- 最多可减少50%的仿真样本需求,且模型更抗极端情况
- 适合需快速部署血流参数估计的临床研究与新场景
血压和壁面剪切应力等血流动力学参数在心血管疾病诊疗中至关重要。虽然计算流体动力学(CFD)可精准计算这些参数,但其计算成本高昂。因此,深度学习常被用作替代方法,快速预测CFD结果。然而,这类数据驱动模型依赖大量耗时的参考CFD仿真进行训练。本文提出一种主动学习框架,通过三种查询策略——几何方差、集成不确定性及物理一致性——决定哪些未标注样本应进行CFD仿真。我们在合成冠状动脉分叉的流速场估计任务上验证,该方法显著降低标注成本。结果显示,最多可减少50%的样本需求,同时提升模型对复杂案例的鲁棒性。表明主动学习是提升深度学习代理模型实用性的可行路径。
原文摘要 · Abstract (English)
Hemodynamic parameters such as pressure and wall shear stress play an important role in diagnosis, prognosis, and treatment planning in cardiovascular diseases. These parameters can be accurately computed using computational fluid dynamics (CFD), but CFD is computationally intensive. Hence, deep learning methods have been adopted as a surrogate to rapidly estimate CFD outcomes. A drawback of such data-driven models is the need for time-consuming reference CFD simulations for training. In this work, we introduce an active learning framework to reduce the number of CFD simulations required for the training of surrogate models, lowering the barriers to their deployment in new applications. We propose three distinct querying strategies to determine for which unlabeled samples CFD simulations should be obtained. These querying strategies are based on geometrical variance, ensemble uncertainty, and adherence to the physics governing fluid dynamics. We benchmark these methods on velocity field estimation in synthetic coronary artery bifurcations and find that they allow for substantial reductions in annotation cost. Notably, we find that our strategies reduce the number of samples required by up to 50% and make the trained models more robust to difficult cases. Our results show that active learning is a feasible strategy to increase the potential of deep learning-based CFD surrogates.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。