LLM2D
深度证据学习用于放疗剂量预测
Deep Evidential Learning for Radiotherapy Dose Prediction
发布日期: 9/24/2024
arXiv ID: oai:arXiv.org:2404.17126v2

摘要

在这项工作中,我们展示了一种名为深度证据学习的不确定性量化框架在放射治疗剂量预测领域的新应用。使用开放知识计划挑战数据集的医学图像,我们发现该模型能够有效地生成不确定性估计,这些估计在网络训练完成后继承了与预测误差的相关性。只有在重新制定原始损失函数以实现稳定实施后才实现这一点。我们发现:(i) 认识不确定性与预测误差高度相关,其各种关联指数与蒙特卡罗丢弃和深度集成方法相当或更强,(ii) 在深度证据学习中,认识不确定性的中位误差随不确定性阈值的变化相对于其他两个常规框架更线性,表明对模型误差的敏感性更加均匀校准,(iii) 相对于认识不确定性,反映数据噪声的随机不确定性在响应CT强度中添加的高斯噪声时,其分布发生了更显著的变化。总体而言,我们的结果表明,深度证据学习是一种有前景的方法,可以赋予放射治疗剂量预测中的深度学习模型以统计稳健性。为了增强其临床相关性,我们展示了如何使用此类模型构建预测剂量-体积直方图的置信区间。